home *** CD-ROM | disk | FTP | other *** search
/ GameStar 2006 January / Gamestar_80_2006-01_dvd.iso / Dema / Civilization4 / data1.cab / Civ4DemoComponent / Assets / Python / System / MimeWriter.pyc (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2005-11-09  |  6.9 KB  |  194 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.4)
  3.  
  4. """Generic MIME writer.
  5.  
  6. This module defines the class MimeWriter.  The MimeWriter class implements
  7. a basic formatter for creating MIME multi-part files.  It doesn't seek around
  8. the output file nor does it use large amounts of buffer space. You must write
  9. the parts out in the order that they should occur in the final file.
  10. MimeWriter does buffer the headers you add, allowing you to rearrange their
  11. order.
  12.  
  13. """
  14. import mimetools
  15. __all__ = [
  16.     'MimeWriter']
  17.  
  18. class MimeWriter:
  19.     """Generic MIME writer.
  20.  
  21.     Methods:
  22.  
  23.     __init__()
  24.     addheader()
  25.     flushheaders()
  26.     startbody()
  27.     startmultipartbody()
  28.     nextpart()
  29.     lastpart()
  30.  
  31.     A MIME writer is much more primitive than a MIME parser.  It
  32.     doesn't seek around on the output file, and it doesn't use large
  33.     amounts of buffer space, so you have to write the parts in the
  34.     order they should occur on the output file.  It does buffer the
  35.     headers you add, allowing you to rearrange their order.
  36.  
  37.     General usage is:
  38.  
  39.     f = <open the output file>
  40.     w = MimeWriter(f)
  41.     ...call w.addheader(key, value) 0 or more times...
  42.  
  43.     followed by either:
  44.  
  45.     f = w.startbody(content_type)
  46.     ...call f.write(data) for body data...
  47.  
  48.     or:
  49.  
  50.     w.startmultipartbody(subtype)
  51.     for each part:
  52.         subwriter = w.nextpart()
  53.         ...use the subwriter's methods to create the subpart...
  54.     w.lastpart()
  55.  
  56.     The subwriter is another MimeWriter instance, and should be
  57.     treated in the same way as the toplevel MimeWriter.  This way,
  58.     writing recursive body parts is easy.
  59.  
  60.     Warning: don't forget to call lastpart()!
  61.  
  62.     XXX There should be more state so calls made in the wrong order
  63.     are detected.
  64.  
  65.     Some special cases:
  66.  
  67.     - startbody() just returns the file passed to the constructor;
  68.       but don't use this knowledge, as it may be changed.
  69.  
  70.     - startmultipartbody() actually returns a file as well;
  71.       this can be used to write the initial 'if you can read this your
  72.       mailer is not MIME-aware' message.
  73.  
  74.     - If you call flushheaders(), the headers accumulated so far are
  75.       written out (and forgotten); this is useful if you don't need a
  76.       body part at all, e.g. for a subpart of type message/rfc822
  77.       that's (mis)used to store some header-like information.
  78.  
  79.     - Passing a keyword argument 'prefix=<flag>' to addheader(),
  80.       start*body() affects where the header is inserted; 0 means
  81.       append at the end, 1 means insert at the start; default is
  82.       append for addheader(), but insert for start*body(), which use
  83.       it to determine where the Content-Type header goes.
  84.  
  85.     """
  86.     
  87.     def __init__(self, fp):
  88.         self._fp = fp
  89.         self._headers = []
  90.  
  91.     
  92.     def addheader(self, key, value, prefix = 0):
  93.         '''Add a header line to the MIME message.
  94.  
  95.         The key is the name of the header, where the value obviously provides
  96.         the value of the header. The optional argument prefix determines
  97.         where the header is inserted; 0 means append at the end, 1 means
  98.         insert at the start. The default is to append.
  99.  
  100.         '''
  101.         lines = value.split('\n')
  102.         while lines and not lines[-1]:
  103.             del lines[-1]
  104.         while lines and not lines[0]:
  105.             del lines[0]
  106.         for i in range(1, len(lines)):
  107.             lines[i] = '    ' + lines[i].strip()
  108.         
  109.         value = '\n'.join(lines) + '\n'
  110.         line = key + ': ' + value
  111.         if prefix:
  112.             self._headers.insert(0, line)
  113.         else:
  114.             self._headers.append(line)
  115.  
  116.     
  117.     def flushheaders(self):
  118.         """Writes out and forgets all headers accumulated so far.
  119.  
  120.         This is useful if you don't need a body part at all; for example,
  121.         for a subpart of type message/rfc822 that's (mis)used to store some
  122.         header-like information.
  123.  
  124.         """
  125.         self._fp.writelines(self._headers)
  126.         self._headers = []
  127.  
  128.     
  129.     def startbody(self, ctype, plist = [], prefix = 1):
  130.         '''Returns a file-like object for writing the body of the message.
  131.  
  132.         The content-type is set to the provided ctype, and the optional
  133.         parameter, plist, provides additional parameters for the
  134.         content-type declaration.  The optional argument prefix determines
  135.         where the header is inserted; 0 means append at the end, 1 means
  136.         insert at the start. The default is to insert at the start.
  137.  
  138.         '''
  139.         for name, value in plist:
  140.             ctype = ctype + ';\n %s="%s"' % (name, value)
  141.         
  142.         self.addheader('Content-Type', ctype, prefix = prefix)
  143.         self.flushheaders()
  144.         self._fp.write('\n')
  145.         return self._fp
  146.  
  147.     
  148.     def startmultipartbody(self, subtype, boundary = None, plist = [], prefix = 1):
  149.         '''Returns a file-like object for writing the body of the message.
  150.  
  151.         Additionally, this method initializes the multi-part code, where the
  152.         subtype parameter provides the multipart subtype, the boundary
  153.         parameter may provide a user-defined boundary specification, and the
  154.         plist parameter provides optional parameters for the subtype.  The
  155.         optional argument, prefix, determines where the header is inserted;
  156.         0 means append at the end, 1 means insert at the start. The default
  157.         is to insert at the start.  Subparts should be created using the
  158.         nextpart() method.
  159.  
  160.         '''
  161.         if not boundary:
  162.             pass
  163.         self._boundary = mimetools.choose_boundary()
  164.         return self.startbody('multipart/' + subtype, [
  165.             ('boundary', self._boundary)] + plist, prefix = prefix)
  166.  
  167.     
  168.     def nextpart(self):
  169.         '''Returns a new instance of MimeWriter which represents an
  170.         individual part in a multipart message.
  171.  
  172.         This may be used to write the part as well as used for creating
  173.         recursively complex multipart messages. The message must first be
  174.         initialized with the startmultipartbody() method before using the
  175.         nextpart() method.
  176.  
  177.         '''
  178.         self._fp.write('\n--' + self._boundary + '\n')
  179.         return self.__class__(self._fp)
  180.  
  181.     
  182.     def lastpart(self):
  183.         '''This is used to designate the last part of a multipart message.
  184.  
  185.         It should always be used when writing multipart messages.
  186.  
  187.         '''
  188.         self._fp.write('\n--' + self._boundary + '--\n')
  189.  
  190.  
  191. if __name__ == '__main__':
  192.     import test.test_MimeWriter as test
  193.  
  194.