def update(db_session,
account,
draft_public_id,
recipients=None,
subject=None,
body=None,
attachments=None):
"""
Update the draft with public_id = draft_public_id.
To maintain our messages are immutable invariant, we create a new draft
message object.
Returns
-------
int
The public_id of the updated draft
Note: This is != draft_public_id.
"""
sender_info = SenderInfo(name=account.full_name,
email=account.email_address)
draft = SpoolMessage.get_or_copy(db_session, draft_public_id)
db_session.add(draft)
db_session.commit()
# For all these fields: if present in the update request, use the value
# provided; else, copy from original.
recipients = recipients or draft.recipients
subject = subject or draft.subject
body = body or draft.subject
attachments = attachments or draft.attachments
draftmsg = _create_gmail_draft(sender_info,
recipients,
subject,
body,
attachments,
original_draft=draft,
reply_to=draft.replyto_thread_id)
newuid = _save_gmail_draft(db_session, account.id, draftmsg)
return newuid.message
def update(db_session, account, draft_public_id, recipients=None, subject=None,
body=None, attachments=None):
"""
Update the draft with public_id = draft_public_id.
To maintain our messages are immutable invariant, we create a new draft
message object.
Returns
-------
int
The public_id of the updated draft
Note: This is != draft_public_id.
"""
sender_info = SenderInfo(name=account.full_name,
email=account.email_address)
draft = SpoolMessage.get_or_copy(db_session, draft_public_id)
db_session.add(draft)
db_session.commit()
# For all these fields: if present in the update request, use the value
# provided; else, copy from original.
recipients = recipients or draft.recipients
subject = subject or draft.subject
body = body or draft.subject
attachments = attachments or draft.attachments
draftmsg = _create_gmail_draft(sender_info, recipients, subject,
body, attachments,
original_draft=draft,
reply_to=draft.replyto_thread_id)
newuid = _save_gmail_draft(db_session, account.id, draftmsg)
return newuid.message
def create_message(db_session, log, account, mid, folder_name, received_date,
flags, body_string, created):
""" Parses message data and writes out db metadata and MIME blocks.
Returns the new Message, which links to the new Block objects through
relationships. All new objects are uncommitted.
Threads are not computed here; you gotta do that separately.
Parameters
----------
mid : int
The account backend-specific message identifier; it's only used for
logging errors.
raw_message : str
The full message including headers (encoded).
"""
# trickle-down bugs
assert account is not None and account.namespace is not None
assert not isinstance(body_string, unicode)
try:
parsed = mime.from_string(body_string)
mime_version = parsed.headers.get('Mime-Version')
# NOTE: sometimes MIME-Version is set to "1.0 (1.0)", hence the
# .startswith
if mime_version is not None and not mime_version.startswith('1.0'):
log.error('Unexpected MIME-Version: {0}'.format(mime_version))
new_msg = SpoolMessage() if created else Message()
new_msg.data_sha256 = sha256(body_string).hexdigest()
# clean_subject strips re:, fwd: etc.
new_msg.subject = parsed.clean_subject
new_msg.from_addr = parse_email_address_list(
parsed.headers.get('From'))
new_msg.sender_addr = parse_email_address_list(
parsed.headers.get('Sender'))
new_msg.reply_to = parse_email_address_list(
parsed.headers.get('Reply-To'))
new_msg.to_addr = parse_email_address_list(parsed.headers.getall('To'))
new_msg.cc_addr = parse_email_address_list(parsed.headers.getall('Cc'))
new_msg.bcc_addr = parse_email_address_list(
parsed.headers.getall('Bcc'))
new_msg.in_reply_to = parsed.headers.get('In-Reply-To')
new_msg.message_id_header = parsed.headers.get('Message-Id')
new_msg.received_date = received_date
# Optional mailing list headers
new_msg.mailing_list_headers = parse_ml_headers(parsed.headers)
# Custom Inbox header
new_msg.inbox_uid = parsed.headers.get('X-INBOX-ID')
# In accordance with JWZ (http://www.jwz.org/doc/threading.html)
new_msg.references = parse_references(
parsed.headers.get('References', ''),
parsed.headers.get('In-Reply-To', ''))
new_msg.size = len(body_string) # includes headers text
i = 0 # for walk_index
# Store all message headers as object with index 0
headers_part = Part()
headers_part.namespace_id = account.namespace.id
headers_part.message = new_msg
headers_part.walk_index = i
headers_part.data = json.dumps(parsed.headers.items())
new_msg.parts.append(headers_part)
for mimepart in parsed.walk(
with_self=parsed.content_type.is_singlepart()):
i += 1
if mimepart.content_type.is_multipart():
log.warning("multipart sub-part found! on {}"
.format(new_msg.g_msgid))
continue # TODO should we store relations?
new_part = Part()
new_part.namespace_id = account.namespace.id
new_part.message = new_msg
new_part.walk_index = i
new_part.misc_keyval = mimepart.headers.items() # everything
new_part.content_type = mimepart.content_type.value
new_part.filename = trim_filename(
mimepart.content_type.params.get('name'),
log=log)
# TODO maybe also trim other headers?
if mimepart.content_disposition[0] is not None:
value, params = mimepart.content_disposition
if value not in ['inline', 'attachment']:
errmsg = """
Unknown Content-Disposition on message {0} found in {1}.
Bad Content-Disposition was: '{2}'
Parsed Content-Disposition was: '{3}'""".format(
mid, folder_name, mimepart.content_disposition)
log.error(errmsg)
continue
else:
new_part.content_disposition = value
if value == 'attachment':
new_part.filename = trim_filename(
params.get('filename'),
log=log)
if mimepart.body is None:
data_to_write = ''
elif new_part.content_type.startswith('text'):
data_to_write = mimepart.body.encode('utf-8', 'strict')
# normalize mac/win/unix newlines
data_to_write = data_to_write \
.replace('\r\n', '\n').replace('\r', '\n')
else:
data_to_write = mimepart.body
if data_to_write is None:
data_to_write = ''
new_part.content_id = mimepart.headers.get('Content-Id')
new_part.data = data_to_write
new_msg.parts.append(new_part)
except mime.DecodingError:
# occasionally iconv will fail via maximum recursion depth
log_decode_error(account.id, folder_name, mid, body_string)
log.error('DecodeError, msg logged to {0}'.format(
get_errfilename(account.id, folder_name, mid)))
return
except RuntimeError:
log_decode_error(account.id, folder_name, mid, body_string)
log.error('RuntimeError<iconv> msg logged to {0}'.format(
get_errfilename(account.id, folder_name, mid)))
return
new_msg.calculate_sanitized_body()
return new_msg
