home *** CD-ROM | disk | FTP | other *** search
/ Celestin Apprentice 7 / Apprentice-Release7.iso / Source Code / C / Applications / Python 1.4 / ic / ic.py < prev    next >
Encoding:
Python Source  |  1997-01-10  |  6.5 KB  |  259 lines  |  [TEXT/Pyth]
  1. """IC wrapper module, based on Internet Config 1.3"""
  2.  
  3. import icglue
  4. import string
  5. import sys
  6. import Res
  7. import macfs
  8. import macostools
  9.  
  10. error=icglue.error
  11.  
  12. # From ictypes.h:
  13. icPrefNotFoundErr = -666        # preference not found (duh!)
  14. icPermErr = -667                # cannot set preference
  15. icPrefDataErr = -668            # problem with preference data
  16. icInternalErr = -669            # hmm, this is not good
  17. icTruncatedErr = -670            # more data was present than was returned
  18. icNoMoreWritersErr = -671        # you cannot begin a write session because someone else is already doing it */
  19. icNothingToOverrideErr = -672    # no component for the override component to capture
  20. icNoURLErr = -673                # no URL found
  21. icConfigNotFoundErr = -674        # no configuration was found
  22. icConfigInappropriateErr = -675    # incorrect manufacturer code
  23.  
  24. ICattr_no_change = -1
  25.  
  26. icNoPerm = 0
  27. icReadOnlyPerm = 1
  28. icReadWritePerm = 2
  29. # End of ictypes.h
  30.  
  31. class ICOpaqueData:
  32.     """An unparseable IC entry"""
  33.     def __init__(self, data):
  34.         self.data = data
  35.  
  36.     def __repr__(self):
  37.         return "ICOpaqueData(%s)"%`self.data`
  38.  
  39. _ICOpaqueDataType=type(ICOpaqueData(''))
  40.         
  41. def _decode_default(data, key):
  42.     if len(data) == 0:
  43.         return data
  44.     if ord(data[0]) == len(data)-1:
  45.         # Assume Pstring
  46.         return data[1:]
  47.     return ICOpaqueData(data)
  48.     
  49.     
  50. def _decode_multistr(data, key):
  51.     numstr = ord(data[0]) << 8 | ord(data[1])
  52.     rv = []
  53.     ptr = 2
  54.     for i in range(numstr):
  55.         strlen = ord(data[ptr])
  56.         str = data[ptr+1:ptr+strlen+1]
  57.         rv.append(str)
  58.         ptr = ptr + strlen + 1
  59.     return rv
  60.     
  61. def _decode_fontrecord(data, key):
  62.     size = ord(data[0]) << 8 | ord(data[1])
  63.     face = ord(data[2])
  64.     namelen = ord(data[4])
  65.     return size, face, data[5:5+namelen]
  66.     
  67. def _decode_boolean(data, key):
  68.     print 'XXXX boolean:', `data`
  69.     return ord(data[0])
  70.     
  71. def _decode_text(data, key):
  72.     return data
  73.     
  74. def _decode_charset(data, key):
  75.     return data[:256], data[256:]
  76.     
  77. def _decode_appspec(data, key):
  78.     namelen = ord(data[4])
  79.     return data[0:4], data[5:5+namelen]
  80.  
  81. def _code_default(data, key):
  82.     return chr(len(data)) + data
  83.         
  84. def _code_multistr(data, key):
  85.     numstr = len(data)
  86.     rv = chr((numstr>>8) & 0xff) + chr(numstr & 0xff)
  87.     for i in data:
  88.         rv = rv + _code_default(i)
  89.     return rv
  90.     
  91. def _code_fontrecord(data, key):
  92.     size, face, name = data
  93.     return chr((size>>8) & 0xff) + chr(size & 0xff) + chr(face & 0xff) + \
  94.         chr(0) + _code_default(name)
  95.     
  96. def _code_boolean(data, key):
  97.     print 'XXXX boolean:', `data`
  98.     return chr(data)
  99.     
  100. def _code_text(data, key):
  101.     return data
  102.     
  103. def _code_charset(data, key):
  104.     return data[0] + data[1]
  105.     
  106. def _code_appspec(data, key):
  107.     return data[0] + _code_default(data[1])
  108.     
  109. _decoder_table = {
  110.     "ArchieAll" : (_decode_multistr , _code_multistr),
  111.     "UMichAll" : (_decode_multistr , _code_multistr),
  112.     "InfoMacAll" : (_decode_multistr , _code_multistr),
  113.     "ListFont" : (_decode_fontrecord , _code_fontrecord),
  114.     "ScreenFont" : (_decode_fontrecord , _code_fontrecord),
  115.     "PrinterFont" : (_decode_fontrecord , _code_fontrecord),
  116. #    "DownloadFolder" : (_decode_filespec , _code_filespec),
  117.     "Signature": (_decode_text , _code_text),
  118.     "Plan" : (_decode_text , _code_text),
  119.     "MailHeaders" : (_decode_text , _code_text),
  120.     "NewsHeaders" : (_decode_text , _code_text),
  121. #    "Mapping" 
  122.     "CharacterSet" : (_decode_charset , _code_charset),
  123.     "Helper\245" : (_decode_appspec , _code_appspec),
  124. #    "Services" : (_decode_services, ????),
  125.     "NewMailFlashIcon" : (_decode_boolean , _code_boolean),
  126.     "NewMailDialog" : (_decode_boolean , _code_boolean),
  127.     "NewMailPlaySound" : (_decode_boolean , _code_boolean),
  128. #    "WebBackgroundColor" : _decode_color,
  129.     "NoProxyDomains" : (_decode_multistr , _code_multistr),
  130.     "UseHTTPProxy" : (_decode_boolean , _code_boolean),
  131.     "UseGopherProxy": (_decode_boolean , _code_boolean),
  132.     "UseFTPProxy" : (_decode_boolean , _code_boolean),
  133.     "UsePassiveFTP" : (_decode_boolean , _code_boolean),
  134. }
  135.  
  136. def _decode(data, key):
  137.     if '\245' in key:
  138.         key2 = key[:string.index(key, '\245')+1]
  139.     else:
  140.         key2 = key
  141.     if _decoder_table.has_key(key2):
  142.         decoder = _decoder_table[key2][0]
  143.     else:
  144.         decoder = _decode_default
  145.     return decoder(data, key)
  146.  
  147. def _code(data, key):
  148.     if type(data) == _ICOpaqueDataType:
  149.         return data.data
  150.     if '\245' in key:
  151.         key2 = key[:string.index(key, '\245')+1]
  152.     else:
  153.         key2 = key
  154.     if _decoder_table.has_key(key2):
  155.         coder = _decoder_table[key2][1]
  156.     else:
  157.         coder = _code_default
  158.     return coder(data, key)
  159.     
  160. class IC:
  161.     def __init__(self, signature='Pyth', ic=None):
  162.         if ic:
  163.             self.ic = ic
  164.         else:
  165.             self.ic = icglue.ICStart(signature)
  166.             self.ic.ICFindConfigFile()
  167.         self.h = Res.Resource('')
  168.             
  169.     def keys(self):
  170.         rv = []
  171.         self.ic.ICBegin(icReadOnlyPerm)
  172.         num = self.ic.ICCountPref()
  173.         for i in range(num):
  174.             rv.append(self.ic.ICGetIndPref(i+1))
  175.         self.ic.ICEnd()
  176.         return rv
  177.         
  178.     def __getitem__(self, key):
  179.         attr = self.ic.ICFindPrefHandle(key, self.h)
  180.         return _decode(self.h.data, key)
  181.         
  182.     def __setitem__(self, key, value):
  183.         value = _code(value, key)
  184.         self.ic.ICSetPref(key, ICattr_no_change, value)
  185.         
  186.     def launchurl(self, url, hint=""):
  187.         self.ic.ICLaunchURL(hint, url, 0, len(url))
  188.         
  189.     def parseurl(self, data, start=None, end=None, hint=""):
  190.         if start == None:
  191.             selStart = 0
  192.             selEnd = len(data)
  193.         else:
  194.             selStart = selEnd = start
  195.         if end != None:
  196.             selEnd = end
  197.         selStart, selEnd = self.ic.ICParseURL(hint, data, selStart, selEnd, self.h)
  198.         return self.h.data, selStart, selEnd
  199.         
  200.     def mapfile(self, file):
  201.         if type(file) != type(''):
  202.             file = file.as_tuple()[2]
  203.         return self.ic.ICMapFilename(file)
  204.         
  205.     def maptypecreator(self, type, creator, filename=""):
  206.         return self.ic.ICMapTypeCreator(type, creator, filename)
  207.         
  208.     def settypecreator(self, file):
  209.         if type(file) == type(''):
  210.             fss = macfs.FSSpec(file)
  211.         else:
  212.             fss = file
  213.         name = fss.as_tuple()[2]
  214.         record = self.mapfile(name)
  215.         fss.SetCreatorType(record[2], record[1])
  216.         macostools.touched(fss)
  217.         
  218. # Convenience routines
  219. _dft_ic = None
  220.  
  221. def launchurl(url, hint=""):
  222.     global _dft_ic
  223.     if _dft_ic == None: _dft_ic = IC()
  224.     return _dft_ic.launchurl(url, hint)
  225.     
  226. def parseurl(data, start=None, end=None, hint=""):
  227.     global _dft_ic
  228.     if _dft_ic == None: _dft_ic = IC()
  229.     return _dft_ic.parseurl(data, start, end, hint)
  230.     
  231. def mapfile(filename):
  232.     global _dft_ic
  233.     if _dft_ic == None: _dft_ic = IC()
  234.     return _dft_ic.mapfile(filename)
  235.     
  236. def maptypecreator(type, creator, filename=""):
  237.     global _dft_ic
  238.     if _dft_ic == None: _dft_ic = IC()
  239.     return _dft_ic.maptypecreator(type, creator, filename)
  240.     
  241. def settypecreator(file):
  242.     global _dft_ic
  243.     if _dft_ic == None: _dft_ic = IC()
  244.     return _dft_ic.settypecreator(file)
  245.         
  246. def _test():
  247.     ic = IC()
  248.     for k in ic.keys():
  249.         try:
  250.             v = ic[k]
  251.         except error:
  252.             v = '????'
  253.         print k, '\t', v
  254.     sys.exit(1)
  255.     
  256. if __name__ == '__main__':
  257.     _test()
  258.     
  259.