mailbox.py нам не нужен
parent
4547c85237
commit
6812dd0b8d
|
|
@ -1,315 +0,0 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
|
||||
from twisted.mail import maildir, imap4
|
||||
from twisted.mail.smtp import rfc822date
|
||||
from twisted.internet import inotify
|
||||
from twisted.python import filepath
|
||||
|
||||
from zope.interface import implements
|
||||
from threading import Thread
|
||||
|
||||
import random
|
||||
import email
|
||||
from pickle import load, dump
|
||||
from StringIO import StringIO
|
||||
import os
|
||||
|
||||
from serpent.config import conf
|
||||
from serpent import misc
|
||||
|
||||
from sqlitedict import SqliteDict
|
||||
|
||||
class LoopingTask(Thread):
|
||||
def __init__(self, func, event, interval):
|
||||
Thread.__init__(self)
|
||||
self.func = func
|
||||
self.interval = interval
|
||||
self.stopped = event
|
||||
|
||||
def run(self):
|
||||
while not self.stopped.wait(self.interval):
|
||||
self.func()
|
||||
|
||||
class SerpentAppendMessageTask(maildir._MaildirMailboxAppendMessageTask):
|
||||
|
||||
def moveFileToNew(self):
|
||||
while True:
|
||||
newname = os.path.join(self.mbox.path, "new", maildir._generateMaildirName())
|
||||
try:
|
||||
self.osrename(self.tmpname, newname)
|
||||
break
|
||||
except OSError, (err, estr):
|
||||
import errno
|
||||
# if the newname exists, retry with a new newname.
|
||||
if err != errno.EEXIST:
|
||||
self.fail()
|
||||
newname = None
|
||||
break
|
||||
if newname is not None:
|
||||
self.mbox.lastadded = newname
|
||||
self.defer.callback(None)
|
||||
self.defer = None
|
||||
|
||||
|
||||
|
||||
class ExtendedMaildir(maildir.MaildirMailbox):
|
||||
def __iter__(self):
|
||||
return iter(self.list)
|
||||
|
||||
def __getitem__(self, i):
|
||||
return self.list[i]
|
||||
|
||||
class IMAPMailbox(ExtendedMaildir):
|
||||
implements(imap4.IMailbox, imap4.ICloseableMailbox)
|
||||
|
||||
AppendFactory = SerpentAppendMessageTask
|
||||
|
||||
def __init__(self, path):
|
||||
maildir.initializeMaildir(path)
|
||||
self.listeners = []
|
||||
self.path = path
|
||||
self.msg_info = SqliteDict(os.path.join(path, conf.imap_msg_info))
|
||||
self.mbox_info = SqliteDict(os.path.join(path, conf.imap_mbox_info))
|
||||
self.lastadded = None
|
||||
self.__check_flags_()
|
||||
|
||||
def _start_monitor(self):
|
||||
self.notifier = inotify.INotify()
|
||||
self.notifier.startReading()
|
||||
self.notifier.watch(filepath.FilePath(os.path.join(self.path, 'new')),
|
||||
callbacks=[self._new_files])
|
||||
self.notifier.watch(filepath.FilePath(os.path.join(self.path,'cur')),
|
||||
callbacks=[self._new_files])
|
||||
|
||||
def _new_files(self, wo, path, code):
|
||||
if code == inotify.IN_MOVED_TO or code == inotify.IN_DELETE:
|
||||
for l in self.listeners:
|
||||
l.newMessages(self.getMessageCount(), self.getRecentCount())
|
||||
|
||||
def __check_flags_(self):
|
||||
if 'subscribed' not in self.mbox_info.keys(): self.mbox_info['subscribed'] = False
|
||||
if 'uidvalidity' not in self.mbox_info.keys(): self.mbox_info['uidvalidity'] = random.randint(0, 2**32)
|
||||
if 'uidnext' not in self.mbox_info.keys(): self.mbox_info['uidnext'] = 1
|
||||
self.mbox_info.commit(blocking=False)
|
||||
l = [l for l in self.__msg_list_()]
|
||||
for i in l:
|
||||
fn = i.split('/')[-1]
|
||||
if fn not in msg_info.keys():
|
||||
val1 = {'uid': self.getUIDNext()}
|
||||
if i.split('/')[-2] == 'new':
|
||||
val1['flags'] = []
|
||||
else:
|
||||
val1['flags'] = [misc.IMAP_FLAGS['SEEN']]
|
||||
self.msg_info[fn] = val1
|
||||
self.msg_info.commit(blocking=False)
|
||||
|
||||
def subscribe(self):
|
||||
self.mbox_info['subscribed'] = True
|
||||
self.mbox_info.commit(blocking=False)
|
||||
|
||||
def unsubscribe(self):
|
||||
self.mbox_info['subscribed'] = False
|
||||
self.mbox_info.commit(blocking=False)
|
||||
|
||||
def is_subscribed(self):
|
||||
return self.mbox_info['subscribed']
|
||||
|
||||
def __count_flagged_msgs_(self, flag):
|
||||
val1 = [0 for fn in self.msg_info.keys() if flag in self.msg_info[fn]['flags']]
|
||||
return len(val1)
|
||||
|
||||
def getHierarchicalDelimiter(self):
|
||||
return misc.IMAP_HDELIM
|
||||
|
||||
def getFlags(self):
|
||||
return misc.IMAP_FLAGS.values()
|
||||
|
||||
def getMessageCount(self):
|
||||
val1 = [0 for fn in self.msg_info.keys() if misc.IMAP_FLAGS['DELETED'] not in self.msg_info[fn]['flags']]
|
||||
return len(val1)
|
||||
|
||||
def getRecentCount(self):
|
||||
return self.__count_flagged_msgs_(misc.IMAP_FLAGS['RECENT'])
|
||||
|
||||
def getUnseenCount(self):
|
||||
return self.getMessageCount() - self.__count_flagged_msgs_(misc.IMAP_FLAGS['SEEN'])
|
||||
|
||||
def isWriteable(self):
|
||||
return True
|
||||
|
||||
def getUIDValidity(self):
|
||||
return self.mbox_info['uidvalidity']
|
||||
|
||||
def getUIDNext(self):
|
||||
un = self.mbox_info['uidnext']
|
||||
self.mbox_info['uidnext'] += 1
|
||||
self.mbox_info.commit(blocking=False)
|
||||
return un
|
||||
|
||||
def getUID(self, num):
|
||||
return num
|
||||
|
||||
def addMessage(self, message, flags = (), date = None):
|
||||
return self.appendMessage(message).addCallback(self._cbAddMessage, flags)
|
||||
|
||||
def _cbAddMessage(self, obj, flags):
|
||||
path = self.lastadded
|
||||
self.lastadded = None
|
||||
fn = path.split('/')[-1]
|
||||
self.msg_info[fn] = {'uid': self.getUIDNext(), 'flags': flags}
|
||||
self.msg_info.commit(blocking=False)
|
||||
if misc.IMAP_FLAGS['SEEN'] in flags and path.split('/')[-2] != 'cur':
|
||||
new_path = os.path.join(self.path, 'cur', fn)
|
||||
os.rename(path, new_path)
|
||||
|
||||
def __msg_list_(self):
|
||||
a = []
|
||||
for m in os.listdir(os.path.join(self.path, 'new')):
|
||||
a.append(os.path.join(self.path, 'new', m))
|
||||
for m in os.listdir(os.path.join(self.path, 'cur')):
|
||||
a.append(os.path.join(self.path, 'cur', m))
|
||||
return a
|
||||
|
||||
def _seqMessageSetToSeqDict(self, messageSet):
|
||||
if not messageSet.last:
|
||||
messageSet.last = self.getMessageCount()
|
||||
|
||||
seqMap = {}
|
||||
msgs = self.__msg_list_()
|
||||
for messageNum in messageSet:
|
||||
if messageNum > 0 and messageNum <= self.getMessageCount():
|
||||
seqMap[messageNum] = msgs[messageNum - 1]
|
||||
return seqMap
|
||||
|
||||
def fetch(self, messages, uid):
|
||||
return [[seq, MaildirMessage(seq,
|
||||
file(filename, 'rb').read(),
|
||||
self.msg_info[filename.split('/')[-1]]['flags'],
|
||||
rfc822date())]
|
||||
for seq, filename in self.__fetch_(messages, uid).iteritems()]
|
||||
def __fetch_(self, messages, uid):
|
||||
if uid:
|
||||
messagesToFetch = {}
|
||||
if not messages.last:
|
||||
messages.last = self.mbox_info['uidnext']
|
||||
fn_uid = dict((fn, self.msg_info[fn]['uid']) for fn in self.msg_info.keys())
|
||||
for uid in messages:
|
||||
if uid in fn_uid.values():
|
||||
for name, _id in fn_uid.iteritems():
|
||||
if uid == _id:
|
||||
if os.path.exists(os.path.join(self.path,'new', name)):
|
||||
messagesToFetch[uid] = os.path.join(self.path,'new', name)
|
||||
elif os.path.exists(os.path.join(self.path,'cur', name)):
|
||||
messagesToFetch[uid] = os.path.join(self.path,'cur', name)
|
||||
else:
|
||||
messagesToFetch = self._seqMessageSetToSeqDict(messages)
|
||||
return messagesToFetch
|
||||
def store(self, messages, flags, mode, uid):
|
||||
d = {}
|
||||
for _id, path in self.__fetch_(messages, uid).iteritems():
|
||||
filename = path.split('/')[-1]
|
||||
if mode < 0:
|
||||
old_f = self.msg_info[filename]['flags']
|
||||
self.msg_info[filename]['flags'] = list(set(old_f).difference(set(flags)))
|
||||
if misc.IMAP_FLAGS['SEEN'] in flags and path.split('/')[-2] != 'new':
|
||||
new_path = os.path.join(self.path, 'new', filename)
|
||||
os.rename(path, new_path)
|
||||
elif mode == 0:
|
||||
self.msg_info[filename]['flags'] = flags
|
||||
elif mode > 0:
|
||||
old_f = self.msg_info[filename]['flags']
|
||||
self.msg_info[filename]['flags'] = list(set(old_f).union(set(flags)))
|
||||
if misc.IMAP_FLAGS['SEEN'] in flags and path.split('/')[-2] != 'cur':
|
||||
new_path = os.path.join(self.path, 'cur', filename)
|
||||
os.rename(path, new_path)
|
||||
d[_id] = self.msg_info[filename]['flags']
|
||||
msg_info.commit(blocking=False)
|
||||
return d
|
||||
|
||||
def expunge(self):
|
||||
uids = []
|
||||
for path in self.__msg_list_():
|
||||
fn = path.split('/')[-1]
|
||||
if fn not in self.msg_info.keys():
|
||||
continue
|
||||
uid = self.msg_info[fn]['uid']
|
||||
if misc.IMAP_FLAGS['DELETED'] in self.msg_info[fn]['flags']:
|
||||
os.remove(path)
|
||||
del self.msg_info[fn]
|
||||
uids.append(uid)
|
||||
self.msg_info.commit(blocking=False)
|
||||
return uids
|
||||
|
||||
def addListener(self, listener):
|
||||
self.listeners.append(listener)
|
||||
return True
|
||||
|
||||
def removeListener(self, listener):
|
||||
self.listeners.remove(listener)
|
||||
return True
|
||||
|
||||
def requestStatus(self, names):
|
||||
return imap4.statusRequestHelper(self, names)
|
||||
|
||||
def destroy(self):
|
||||
pass
|
||||
|
||||
def close(self):
|
||||
self.notifier.stopReading()
|
||||
self.notifier.loseConnection()
|
||||
if conf.imap_expunge_on_close:
|
||||
l = self.expunge()
|
||||
|
||||
class MaildirMessagePart(object):
|
||||
implements(imap4.IMessagePart)
|
||||
|
||||
def __init__(self, message):
|
||||
self.message = message
|
||||
self.data = str(message)
|
||||
|
||||
def getHeaders(self, negate, *names):
|
||||
if not names:
|
||||
names = self.message.keys()
|
||||
|
||||
headers = {}
|
||||
if negate:
|
||||
for header in self.message.keys():
|
||||
if header.upper() not in names:
|
||||
headers[header.lower()] = self.message.get(header, '')
|
||||
else:
|
||||
for name in names:
|
||||
headers[name.lower()] = self.message.get(name, '')
|
||||
|
||||
return headers
|
||||
|
||||
def getBodyFile(self):
|
||||
return StringIO(self.message.get_payload())
|
||||
|
||||
def getSize(self):
|
||||
return len(self.data)
|
||||
|
||||
def isMultipart(self):
|
||||
return self.message.is_multipart()
|
||||
|
||||
def getSubPart(self, part):
|
||||
return MaildirMessagePart(self.message.get_payload(part))
|
||||
|
||||
class MaildirMessage(MaildirMessagePart):
|
||||
implements(imap4.IMessage)
|
||||
|
||||
def __init__(self, uid, message, flags, date):
|
||||
MaildirMessagePart.__init__(self, message)
|
||||
self.uid = uid
|
||||
self.message = email.message_from_string(message)
|
||||
self.flags = flags
|
||||
self.date = date
|
||||
|
||||
|
||||
def getUID(self):
|
||||
return self.uid
|
||||
|
||||
def getFlags(self):
|
||||
return self.flags
|
||||
|
||||
def getInternalDate(self):
|
||||
return self.date
|
||||
Loading…
Reference in New Issue