开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > WEB邮件程序 > 查看源码
Message.py
资源名称：bobomail-0.3-1.tar.gz　[点击查看]
上传用户：gyjinxi
上传日期：2007-01-04
资源大小：159k
文件大小：5k
源码类别：

WEB邮件程序

开发平台：

Python

Message.py：源码内容
  1. try: import locale; locale.setlocale(locale.LC_ALL,"")
  2. except: pass
  3. import mimetools, mimify, re, multifile, base64, string, time
  4. try: from cStringIO import StringIO
  5. except: from StringIO import StringIO
  7. mime_head_quopri = re.compile('=\?iso-8859-[12]\?q\?([^? tn]+)\?=', re.I)
  8. mime_head_base64 = re.compile('=\?iso-8859-[12]\?b\?([^? tn]+)\?=', re.I)
  10. def mime_decode(line, isBase64=0):
  11.         if isBase64: return base64.decodestring(line)
  12.         else: return mimify.mime_decode(line)
  14. # mimify.mime_decode_header only decodes quoted printable
  15. # this one also handles base64 encoded headers
  16. def mime_decode_header(line):
  17.         '''Decode a header line to 8bit.'''
  18.         newline = ''
  19.         pos = 0
  20.         while 1:
  21.                 res = mime_head_quopri.search(line, pos)
  22.                 if res is None:
  23.                         res = mime_head_base64.search(line, pos)
  24.                         if res is None: break
  25.                         else: isBase64 = 1
  26.                 else: isBase64 = 0
  27.                 match = res.group(1)
  28.                 # convert underscores to spaces (before =XX conversion!)
  29.                 match = string.join(string.split(match, '_'), ' ')
  30.                 newline = newline + line[pos:res.start(0)] + mime_decode(match, isBase64)
  31.                 pos = res.end(0)
  32. return newline + line[pos:]
  36. class Message(mimetools.Message):
  37.     """A real MIME-message class :-)
  38.     Handles multipart messages and 
  39.     helps decoding parts as well as headers.
  40.     Disadvantage: slow on long mails :-(
  41.     """
  42.     
  43.     def __init__(self, fp, preview=0):
  44.             mimetools.Message.__init__(self, fp)
  45.             self.body = ""
  46.             self.parts = []
  47.             self.encoding = self.getencoding()
  48.             self.boundary = self.getparam("boundary")
  49.             disp = self.getheader("content-disposition") or ""
  50.             f = string.find(disp, 'filename="')
  51.             if f <> -1: #XXX VERY BAD HACK!!!
  52.                     self.filename = disp[f+10:string.find(disp, '"', f+10)]
  53.             else:
  54.                     self.filename = None
  55.             self.parsed = 0
  56.             self.processHeaders()
  57.             if not preview: self.parse()
  58.     
  59.     def processHeaders(self):
  60.             self.Lines = self["Lines"]
  61.             if self.Lines: self.Lines = int(self.Lines)
  62.             self.Subject = self["Subject"]
  63.             self.FromName, self.FromAddr = self.getaddr("From")
  64.             self.Sender = "%s <%s>" % (self.FromName, self.FromAddr)
  65.             self.ReplyToName, self.ReplyToAddr = self.getaddr("Reply-To")
  66.             self.ToName, self.ToAddr = self.getaddr("To")
  67.             self.CcName, self.CcAddr = self.getaddr("Cc")
  68.             self.BccName, self.BccAddr = self.getaddr("Bcc")
  69.             self.UserAgent = self["X-Mailer"] or self["User-Agent"]
  70.             self.Date = self.getdate("Date")
  71.             if self.Date:
  72.                     self.DateStr = time.strftime(
  73.                             "%a, %d. %b %Yn %H:%M", self.Date)
  74.                     self.DateShortStr = time.strftime("%D %T", self.Date)
  75.             else:
  76.                     self.DateStr = self.DateShortStr = "(no date)"
  79.     def parse(self):
  80.         if self.parsed: return
  81.         self.rewindbody()
  82.         if self.boundary:
  83.                 mf = multifile.MultiFile(self.fp)
  84.                 mf.push(self.boundary)
  85.                 while mf.next():
  86.                         self.parts.append(Message(mf))
  87.                 mf.pop()
  88.         else:
  89.                 self.body = self.fp.read()
  90.                 self.parsed = 1
  92.     def decode(self):
  93.             if self.encoding in ['7bit', '8bit']:
  94.                     return self.body
  95.             try:
  96.                     self.rewindbody()
  97.                     out = StringIO()
  98.                     mimetools.decode(StringIO(self.body), out, self.encoding)
  99.                     return out.getvalue()
  100.             except:
  101.                     raise ValueError, "MIME part decoding failed"
  102.             
  103.     def getheader(self, name, default=None):
  104.             header = mimetools.Message.getheader(self, name, default)
  105.             if header:
  106.                     return mime_decode_header(header)
  108.     def getaddr(self, name):
  109.             name, addr = mimetools.Message.getaddr(self, name)
  110.             if name: return mime_decode_header(name), addr
  111.             else: return name, addr
  112.             
  113.     def getdate(self, name):
  114.             try: return mimetools.Message.getdate(self, name)
  115.             except: pass
  116.     
  117.     def __getitem__(self, name):
  118.         return self.getheader(name)
  119.     
  120.     def __str__(self):
  121. # rebuild the message
  122.         f = StringIO()
  123.         f.write('%sn' % string.join(
  124.             map(mimify.mime_encode_header, self.headers), ''))
  125.         if self.parts: # multipart message
  126.             for p in self.parts:
  127.                 f.write('n--%sn' % self.boundary)
  128.                 f.write(str(p))
  129.             f.write('n--%s--n' % self.boundary)
  130.         else:
  131.             f.write(self.body or "")
  132.         return f.getvalue()