Lib/mailbox.py

   1 #
   2 # A class to hand a unix-style or mmdf-style mailboxes
   3 #
   4 # Jack Jansen, CWI, March 1994.
   5 #
   6 import rfc822
   7
   8 class _Mailbox:
   9         def __init__(self, fp):
  10                 self.fp = fp
  11                 self.seekp = 0
  12
  13         def seek(self, pos):
  14                 self.seekp = pos
  15
  16         def next(self):
  17                 while 1:
  18                         self.fp.seek(self.seekp)
  19                         try:
  20                                 self._search_start()
  21                         except EOFError:
  22                                 self.seekp = self.fp.tell()
  23                                 return None
  24                         start = self.fp.tell()
  25                         self._search_end()
  26                         self.seekp = stop = self.fp.tell()
  27                         if start <> stop:
  28                                 break
  29                 return rfc822.Message(_Subfile(self.fp, start, stop))
  30
  31 class _Subfile:
  32         def __init__(self, fp, start, stop):
  33                 self.fp = fp
  34                 self.start = start
  35                 self.stop = stop
  36                 self.pos = self.start
  37
  38         def read(self, length = None):
  39                 if self.pos >= self.stop:
  40                         return ''
  41                 if length is None:
  42                         length = self.stop - self.pos
  43                 self.fp.seek(self.pos)
  44                 self.pos = self.pos + length
  45                 return self.fp.read(length)
  46
  47         def readline(self, length = None):
  48                 if self.pos >= self.stop:
  49                         return ''
  50                 if length is None:
  51                         length = self.stop - self.pos
  52                 self.fp.seek(self.pos)
  53                 data = self.fp.readline(length)
  54                 if len(data) < length:
  55                         length = len(data)
  56                 self.pos = self.pos + length
  57                 return data
  58
  59         def tell(self):
  60                 return self.pos - self.start
  61
  62         def seek(self, pos):
  63                 self.pos = pos + self.start
  64
  65         def close(self):
  66                 pass
  67
  68 class UnixMailbox(_Mailbox):
  69         def _search_start(self):
  70                 while 1:
  71                         line = self.fp.readline()
  72                         if not line:
  73                                 raise EOFError
  74                         if line[:5] == 'From ':
  75                                 return
  76
  77         def _search_end(self):
  78                 while 1:
  79                         pos = self.fp.tell()
  80                         line = self.fp.readline()
  81                         if not line:
  82                                 return
  83                         if line[:5] == 'From ':
  84                                 self.fp.seek(pos)
  85                                 return
  86
  87 class MmdfMailbox(_Mailbox):
  88         def _search_start(self):
  89                 while 1:
  90                         line = self.fp.readline()
  91                         if not line:
  92                                 raise EOFError
  93                         if line[:5] == '\001\001\001\001\n':
  94                                 return
  95
  96         def _search_end(self):
  97                 while 1:
  98                         pos = self.fp.tell()
  99                         line = self.fp.readline()
 100                         if not line:
 101                                 return
 102                         if line == '\001\001\001\001\n':
 103                                 self.fp.seek(pos)
 104                                 return
 105
 106 if __name__ == '__main__':
 107         import posix
 108         import time
 109         import sys
 110         import string
 111         mbox = '/usr/mail/'+posix.environ['USER']
 112         fp = open(mbox, 'r')
 113         mb = UnixMailbox(fp)
 114         msgs = []
 115         while 1:
 116                 msg = mb.next()
 117                 if not msg:
 118                         break
 119                 msgs.append(msg)
 120         if len(sys.argv) > 1:
 121                 num = string.atoi(sys.argv[1])
 122                 print 'Message %d body:'%num
 123                 msg = msgs[num-1]
 124                 msg.rewindbody()
 125                 sys.stdout.write(msg.fp.read())
 126                 sys.exit(0)
 127         print 'Mailbox',mbox,'has',len(msgs),'messages:'
 128         for msg in msgs:
 129                 f = msg.getheader('from')
 130                 s = msg.getheader('subject')
 131                 d = (msg.getheader('date'))
 132                 print '%20.20s   %18.18s   %-30.30s'%(f, d[5:], s)
 133