OS/2 Shareware BBS: 10 Tools/10-Tools.zip/pyos2bin.zip/Lib/mailbox.py

Text File | 1997-10-22 | 4KB | 208 lines

#! /usr/bin/env python

"""Classes to handle Unix style, MMDF style, and MH style mailboxes."""


import rfc822
import os

class _Mailbox:
    def __init__(self, fp):
        self.fp = fp
        self.seekp = 0

    def seek(self, pos, whence=0):
        if whence==1:        # Relative to current position
            self.pos = self.pos + pos
        if whence==2:        # Relative to file's end
            self.pos = self.stop + pos
        else:            # Default - absolute position
            self.pos = self.start + pos

    def next(self):
        while 1:
            self.fp.seek(self.seekp)
            try:
                self._search_start()
            except EOFError:
                self.seekp = self.fp.tell()
                return None
            start = self.fp.tell()
            self._search_end()
            self.seekp = stop = self.fp.tell()
            if start <> stop:
                break
        return rfc822.Message(_Subfile(self.fp, start, stop))

class _Subfile:
    def __init__(self, fp, start, stop):
        self.fp = fp
        self.start = start
        self.stop = stop
        self.pos = self.start

    def read(self, length = None):
        if self.pos >= self.stop:
            return ''
        if length is None:
            length = self.stop - self.pos
        self.fp.seek(self.pos)
        self.pos = self.pos + length
        return self.fp.read(length)

    def readline(self, length = None):
        if self.pos >= self.stop:
            return ''
        if length is None:
            length = self.stop - self.pos
        self.fp.seek(self.pos)
        data = self.fp.readline(length)
        if len(data) < length:
            length = len(data)
        self.pos = self.pos + length
        return data

    def tell(self):
        return self.pos - self.start

    def seek(self, pos, whence=0):
            if whence == 0:
            self.pos = self.start + pos
        elif whence == 1:
            self.pos = self.pos + pos
        elif whence == 2:
            self.pos = self.stop + pos

    def close(self):
        pass

class UnixMailbox(_Mailbox):
    def _search_start(self):
        while 1:
            line = self.fp.readline()
            if not line:
                raise EOFError
            if line[:5] == 'From ':
                return

    def _search_end(self):
        while 1:
            pos = self.fp.tell()
            line = self.fp.readline()
            if not line:
                return
            if line[:5] == 'From ':
                self.fp.seek(pos)
                return

class MmdfMailbox(_Mailbox):
    def _search_start(self):
        while 1:
            line = self.fp.readline()
            if not line:
                raise EOFError
            if line[:5] == '\001\001\001\001\n':
                return

    def _search_end(self):
        while 1:
            pos = self.fp.tell()
            line = self.fp.readline()
            if not line:
                return
            if line == '\001\001\001\001\n':
                self.fp.seek(pos)
                return

class MHMailbox:
    def __init__(self, dirname):
    import re
    pat = re.compile('^[0-9][0-9]*$')
    self.dirname = dirname
    files = os.listdir(self.dirname)
    self.boxes = []
    for f in files:
        if pat.match(f):
        self.boxes.append(f)

    def next(self):
    if not self.boxes:
        return None
    fn = self.boxes[0]
    del self.boxes[0]
    fp = open(os.path.join(self.dirname, fn))
    return rfc822.Message(fp)


class BabylMailbox(_Mailbox):
    def _search_start(self):
    while 1:
        line = self.fp.readline()
        if not line:
        raise EOFError
        if line == '*** EOOH ***\n':
        return

    def _search_end(self):
    while 1:
        pos = self.fp.tell()
        line = self.fp.readline()
        if not line:
        return
        if line == '\037\014\n':
        self.fp.seek(pos)
        return


def _test():
    import time
    import sys
    import string
    import os

    args = sys.argv[1:]
    if not args:
        for key in 'MAIL', 'LOGNAME', 'USER':
            if os.environ.has_key(key):
                mbox = os.environ[key]
                break
        else:
            print "$MAIL, $LOGNAME nor $USER set -- who are you?"
            return
    else:
        mbox = args[0]
    if mbox[:1] == '+':
        mbox = os.environ['HOME'] + '/Mail/' + mbox[1:]
    elif not '/' in mbox:
        mbox = '/usr/mail/' + mbox
    if os.path.isdir(mbox):
        mb = MHMailbox(mbox)
    else:
        fp = open(mbox, 'r')
        mb = UnixMailbox(fp)

    msgs = []
    while 1:
        msg = mb.next()
        if not msg:
            break
        msgs.append(msg)
        msg.fp = None
    if len(args) > 1:
        num = string.atoi(args[1])
        print 'Message %d body:'%num
        msg = msgs[num-1]
        msg.rewindbody()
        sys.stdout.write(msg.fp.read())
    else:
        print 'Mailbox',mbox,'has',len(msgs),'messages:'
        for msg in msgs:
            f = msg.getheader('from')
            s = msg.getheader('subject')
            d = (msg.getheader('date'))
            print '%20.20s   %18.18s   %-30.30s'%(f, d[5:], s)


if __name__ == '__main__':
    _test()