From 16222f9dadce4b13d43fe61bddba9afb16390c72 Mon Sep 17 00:00:00 2001
From: Lars Wirzenius <liw@liw.fi>
Date: Sun, 31 Mar 2013 18:30:27 +0100
Subject: Initial

---
 pieni-to-mail-archive | 117 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 117 insertions(+)
 create mode 100755 pieni-to-mail-archive

(limited to 'pieni-to-mail-archive')

diff --git a/pieni-to-mail-archive b/pieni-to-mail-archive
new file mode 100755
index 0000000..97a0a4c
--- /dev/null
+++ b/pieni-to-mail-archive
@@ -0,0 +1,117 @@
+#!/usr/bin/python
+
+
+import cliapp
+import email.parser
+import glob
+import mailbox
+import os
+import re
+
+
+received_date = re.compile(
+    r'^from (|\n)*; (Mon|Tue|Wed|Thu|Fri|Sat|Sun),\s+'
+        r'(?P<day>\d+) '
+        r'(?P<mon>Jan|Feb|Mar|Apr|May|Jun|Jul|Aug|Sep|Oct|Nov|Dec) '
+        r'(?P<year>\d+)',
+    re.MULTILINE)
+
+date_date = re.compile(
+    r'^((Mon|Tue|Wed|Thu|Fri|Sat|Sun),\s+)?'
+        r'(?P<day>\d+) '
+        r'(?P<mon>Jan|Feb|Mar|Apr|May|Jun|Jul|Aug|Sep|Oct|Nov|Dec) '
+        r'(?P<year>\d+)')
+
+montab = {
+    'Jan': 1,
+    'Feb': 2,
+    'Mar': 3,
+    'Apr': 4,
+    'May': 5,
+    'Jun': 6,
+    'Jul': 7,
+    'Aug': 8,
+    'Sep': 9,
+    'Oct': 10,
+    'Nov': 11,
+    'Dec': 12,
+}
+
+
+class PieniToMailArchive(cliapp.Application):
+
+    def add_settings(self):
+        self.settings.string(
+            ['mbox-dir'],
+            'where to put archived mail mboxes',
+            default='mboxes')
+
+    def setup(self):
+        self.mbox = None
+        self.mbox_filename = None
+
+    def cleanup(self):
+        print 'closing'
+        if self.mbox != None:
+            self.mbox.close()
+
+    def process_args(self, args):
+        # FIXME: Rsync from pieni here.
+
+        for maildir in args:
+            print 'processing:', maildir
+            self.archive_mails_in_maildir(maildir)
+            self.remove_maildir_if_empty(maildir)
+    
+    def archive_mails_in_maildir(self, maildir):
+        md = mailbox.Maildir(maildir, factory=None)
+        parser = email.parser.Parser()
+        for msg in md.itervalues():
+            y, m, d = self.date_of(msg)
+            
+            filename = os.path.join(self.settings['mbox-dir'], '%04d.mbox' % y)
+            existing = glob.glob(filename + '.*')
+            if existing:
+                self.output.write(
+                    'WARNING: %s already has compressed(?) versions: %s' % 
+                        (filename, ' '.join(existing)))
+                continue
+
+            if filename != self.mbox_filename:
+                if self.mbox is not None:
+                    self.mbox.close()
+                self.mbox = mailbox.mbox(filename, create=True)
+                self.mbox_filename = filename
+            self.mbox.add(msg)
+        
+    def date_of(self, msg):
+        year = mon = day = 0
+
+        rs = msg.get_all('Received')
+        if rs:
+            r = str(rs[0])
+            m = received_date.match(r)
+            if m:
+                day = int(m.group('day'))
+                mon = montab[m.group('mon')]
+                year = int(m.group('year'))
+
+        if year is 0 and 'Date' in msg:
+            s = str(msg['Date'])
+            m = date_date.match(s)
+            if m:
+                day = int(m.group('day'))
+                mon = montab[m.group('mon')]
+                year = int(m.group('year'))
+
+        if 70 <= year < 100:
+            year += 1900
+        if year < 1988:
+            year = mon = day = 0
+        return year, mon, day
+
+    def remove_maildir_if_empty(self, maildir):
+        pass
+
+
+PieniToMailArchive().run()
-- 
cgit v1.2.1