mimetypes.py

   1 from xml.dom import minidom
   2 from zeroinstall.injector import namespaces
   3
   4 def add_types(data):
   5         doc = minidom.parseString(data)
   6
   7         changed = False
   8         for archive in doc.documentElement.getElementsByTagNameNS(namespaces.XMLNS_IFACE, 'archive'):
   9                 href = archive.getAttribute('href')
  10                 type = archive.getAttribute('type')
  11
  12                 if not type:
  13                         if href.endswith('.tar.bz2'):
  14                                 type = "application/x-bzip-compressed-tar"
  15                         elif href.endswith('.tgz') or href.endswith('.tar.gz'):
  16                                 type = "application/x-compressed-tar"
  17                         else:
  18                                 raise Exception("Can't guess type for " + href)
  19                         archive.setAttribute('type', type)
  20                         changed = True
  21
  22         if changed:
  23                 return doc.toxml()
  24         else:
  25                 return data