#! /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 ''

remaining = self.stop - self.pos

if length is None or length < 0:

length = remaining

elif length > remaining:

length = remaining

self.fp.seek(self.pos)

data = self.fp.read(length)

self.pos = self.fp.tell()

return data

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)

self.pos = self.fp.tell()

return data

def readlines(self, sizehint = -1):

lines = []

while 1:

line = self.readline()

if not line:

break

lines.append(line)

if sizehint >= 0:

sizehint = sizehint - len(line)

if sizehint <= 0:

break

return lines

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):

del self.fp

class UnixMailbox(_Mailbox):

def _search_start(self):

while 1:

pos = self.fp.tell()

line = self.fp.readline()

if not line:

raise EOFError

if line[:5] == 'From ' and self._isrealfromline(line):

self.fp.seek(pos)

return

def _search_end(self):

self.fp.readline() # Throw away header line

while 1:

pos = self.fp.tell()

line = self.fp.readline()

if not line:

return

if line[:5] == 'From ' and self._isrealfromline(line):

self.fp.seek(pos)

return

# An overridable mechanism to test for From-line-ness.

# You can either specify a different regular expression

# or define a whole new _isrealfromline() method.

# Note that this only gets called for lines starting with

# the 5 characters "From ".

_fromlinepattern = r"From \s*[^\s]+\s+\w\w\w\s+\w\w\w\s+\d?\d\s+" \

r"\d?\d:\d\d(:\d\d)?(\s+[^\s]+)?\s+\d\d\d\d\s*$"

_regexp = None

def _isrealfromline(self, line):

if not self._regexp:

import re

self._regexp = re.compile(self._fromlinepattern)

return self._regexp.match(line)

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('^[1-9][0-9]*$')

self.dirname = dirname

# the three following lines could be combined into:

# list = map(long, filter(pat.match, os.listdir(self.dirname)))

list = os.listdir(self.dirname)

list = filter(pat.match, list)

list = map(long, list)

list.sort()

# This only works in Python 1.6 or later;

# before that str() added 'L':

self.boxes = map(str, list)

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 Maildir:

# Qmail directory mailbox

def __init__(self, dirname):

import string

self.dirname = dirname

self.boxes = []

# check for new mail

newdir = os.path.join(self.dirname, 'new')

for file in os.listdir(newdir):

if len(string.split(file, '.')) > 2:

self.boxes.append(os.path.join(newdir, file))

# Now check for current mail in this maildir

curdir = os.path.join(self.dirname, 'cur')

for file in os.listdir(curdir):

if len(string.split(file, '.')) > 2:

self.boxes.append(os.path.join(curdir, file))

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 'MAILDIR', '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):

if os.path.isdir(os.path.join(mbox, 'cur')):

mb = Maildir(mbox)

else:

mb = MHMailbox(mbox)

else:

fp = open(mbox, 'r')

mb = UnixMailbox(fp)

msgs = []

while 1:

msg = mb.next()

if msg is None:

break

msgs.append(msg)

if len(args) <= 1:

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') or ""

s = msg.getheader('subject') or ""

d = msg.getheader('date') or ""

print '%20.20s %18.18s %-30.30s'%(f, d[5:], s)

if __name__ == '__main__':

_test()