54
def _parse_message(message):
55
"""Get a date and dict of an RFC822 message."""
56
parsed = email.parser.Parser().parsestr(message)
59
for key in ('from', 'to', 'subject', 'message-id'):
60
value = parsed.get(key, None)
62
message_dict[key] = value
64
date = parsed.get('date')
67
pdate = parsedate_tz(date)
68
date = datetime.datetime(
70
tzinfo=dateutil.tz.tzoffset('', pdate[9]))
73
message_dict['date'] = date.isoformat() if date is not None else None
75
return date, message_dict
54
78
class CassandraConnection(object):
56
80
def __init__(self, keyspace, host):
69
93
def add_message(self, archive_uuid, message):
70
94
message_uuid = uuid.uuid4()
71
parsed = email.parser.Parser().parsestr(message)
72
date = parsed.get('date')
75
pdate = parsedate_tz(date)
76
date = datetime.datetime(
78
tzinfo=dateutil.tz.tzoffset('', pdate[9]))
83
{'date_created': datetime.datetime.utcnow().isoformat() + 'Z',
85
'from': parsed.get('From'),
86
'subject': parsed.get('Subject'),
87
'date': date.isoformat() if date is not None else None,
88
'message_id': parsed.get('Message-ID'),
95
message_date, message_dict = _parse_message(message)
96
message_dict['content'] = message
97
message_dict['date_created'] = (
98
datetime.datetime.utcnow().isoformat() + 'Z')
99
self.messages.insert(message_uuid, message_dict)
90
100
self.archive_messages.insert(
92
{date.astimezone(dateutil.tz.tzutc()): message_uuid})
102
{message_date.astimezone(dateutil.tz.tzutc()): message_uuid})
94
'Imported %s into %s' % (parsed.get('Message-ID'), archive_uuid))
104
'Imported %s into %s'
105
% (message_dict.get('message-id', None), archive_uuid))
95
106
return message_uuid
97
108
def _format_message(self, message):
99
'date': message['date'],
100
'from': message['from'],
101
'subject': message['subject'],
110
'date': message.get('date'),
111
'from': message.get('from'),
112
'subject': message.get('subject'),
104
115
def get_messages(self, archive_uuid, order, count, start):
108
119
raise AssertionError("Unsupported order.")
109
120
pairs = self.archive_messages.get(
110
archive_uuid, column_count=count+1,
121
archive_uuid, column_count=count + 1,
111
122
column_start=start, column_reversed=reversed).items()
112
123
ids = [v for k, v in pairs]
113
124
messages = self.messages.multiget(