def _get_contact_map(db_session, namespace_id, all_addresses):
"""
Retrieves or creates contacts for the given address pairs, returning a dict
with the canonicalized emails mapped to Contact objects.
"""
canonicalized_addresses = [canonicalize(addr) for _, addr in all_addresses]
if not canonicalized_addresses:
return {}
existing_contacts = (
db_session.query(Contact)
.filter(
Contact._canonicalized_address.in_(canonicalized_addresses),
Contact.namespace_id == namespace_id,
)
.all()
)
contact_map = {c._canonicalized_address: c for c in existing_contacts}
for name, email_address in all_addresses:
canonicalized_address = canonicalize(email_address)
if canonicalized_address not in contact_map:
new_contact = Contact(
name=name,
email_address=email_address,
namespace_id=namespace_id,
provider_name=INBOX_PROVIDER_NAME,
uid=uuid.uuid4().hex,
)
contact_map[canonicalized_address] = new_contact
return contact_map
def update_contacts_from_message(db_session, message, namespace):
with db_session.no_autoflush:
# First create Contact objects for any email addresses that we haven't
# seen yet. We want to dedupe by canonicalized address, so this part is
# a bit finicky.
canonicalized_addresses = []
all_addresses = []
for field in (message.from_addr, message.to_addr, message.cc_addr,
message.bcc_addr, message.reply_to):
# We generally require these attributes to be non-null, but only
# set them to the default empty list at flush time. So it's better
# to be safe here.
if field is not None:
all_addresses.extend(field)
canonicalized_addresses = [
canonicalize(addr) for _, addr in all_addresses
]
existing_contacts = db_session.query(Contact).filter(
Contact._canonicalized_address.in_(canonicalized_addresses),
Contact.namespace_id == namespace.id).all()
contact_map = {c._canonicalized_address: c for c in existing_contacts}
for name, email_address in all_addresses:
canonicalized_address = canonicalize(email_address)
if canonicalized_address not in contact_map:
new_contact = Contact(name=name,
email_address=email_address,
namespace=namespace,
provider_name=INBOX_PROVIDER_NAME,
uid=uuid.uuid4().hex)
contact_map[canonicalized_address] = new_contact
# Now associate each contact to the message.
for field_name in ('from_addr', 'to_addr', 'cc_addr', 'bcc_addr',
'reply_to'):
field = getattr(message, field_name)
if field is None:
continue
for name, email_address in field:
if not valid_email(email_address):
continue
canonicalized_address = canonicalize(email_address)
contact = contact_map.get(canonicalized_address)
# Hackily address the condition that you get mail from e.g.
# "Ben Gotow (via Google Drive) <*****@*****.**"
# "Christine Spang (via Google Drive) <*****@*****.**"
# and so on: rather than creating many contacts with
# varying name, null out the name for the existing contact.
if contact.name != name and 'noreply' in canonicalized_address:
contact.name = None
message.contacts.append(
MessageContactAssociation(contact=contact,
field=field_name))
def update_contacts_from_message(db_session, message, namespace):
with db_session.no_autoflush:
# First create Contact objects for any email addresses that we haven't
# seen yet. We want to dedupe by canonicalized address, so this part is
# a bit finicky.
canonicalized_addresses = []
all_addresses = []
for field in (message.from_addr, message.to_addr, message.cc_addr,
message.bcc_addr, message.reply_to):
# We generally require these attributes to be non-null, but only
# set them to the default empty list at flush time. So it's better
# to be safe here.
if field is not None:
all_addresses.extend(field)
canonicalized_addresses = [canonicalize(addr) for _, addr in
all_addresses]
existing_contacts = db_session.query(Contact).filter(
Contact._canonicalized_address.in_(canonicalized_addresses),
Contact.namespace_id == namespace.id).all()
contact_map = {c._canonicalized_address: c for c in existing_contacts}
for name, email_address in all_addresses:
canonicalized_address = canonicalize(email_address)
if canonicalized_address not in contact_map:
new_contact = Contact(name=name, email_address=email_address,
namespace=namespace,
provider_name=INBOX_PROVIDER_NAME,
uid=uuid.uuid4().hex)
contact_map[canonicalized_address] = new_contact
# Now associate each contact to the message.
for field_name in ('from_addr', 'to_addr', 'cc_addr', 'bcc_addr',
'reply_to'):
field = getattr(message, field_name)
if field is None:
continue
for name, email_address in field:
if not valid_email(email_address):
continue
canonicalized_address = canonicalize(email_address)
contact = contact_map.get(canonicalized_address)
# Hackily address the condition that you get mail from e.g.
# "Ben Gotow (via Google Drive) <*****@*****.**" # noqa
# "Christine Spang (via Google Drive) <*****@*****.**" # noqa
# and so on: rather than creating many contacts with
# varying name, null out the name for the existing contact.
if contact.name != name and 'noreply' in canonicalized_address:
contact.name = None
message.contacts.append(MessageContactAssociation(
contact=contact, field=field_name))
def update_contacts_from_message(db_session, message, namespace):
with db_session.no_autoflush:
# First create Contact objects for any email addresses that we haven't
# seen yet. We want to dedupe by canonicalized address, so this part is
# a bit finicky.
canonicalized_addresses = []
all_addresses = []
for field in (message.from_addr, message.to_addr, message.cc_addr,
message.bcc_addr, message.reply_to):
# We generally require these attributes to be non-null, but only
# set them to the default empty list at flush time. So it's better
# to be safe here.
if field is not None:
all_addresses.extend(field)
canonicalized_addresses = [
canonicalize(addr) for _, addr in all_addresses
]
existing_contacts = db_session.query(Contact).filter(
Contact._canonicalized_address.in_(canonicalized_addresses),
Contact.namespace_id == namespace.id).all()
contact_map = {c._canonicalized_address: c for c in existing_contacts}
for name, email_address in all_addresses:
canonicalized_address = canonicalize(email_address)
if canonicalized_address not in contact_map:
new_contact = Contact(name=name,
email_address=email_address,
namespace=namespace,
provider_name=INBOX_PROVIDER_NAME,
uid=uuid.uuid4().hex)
contact_map[canonicalized_address] = new_contact
# Now associate each contact to the message.
for field_name in ('from_addr', 'to_addr', 'cc_addr', 'bcc_addr',
'reply_to'):
field = getattr(message, field_name)
if field is None:
continue
for name, email_address in field:
canonicalized_address = canonicalize(email_address)
contact = contact_map.get(canonicalized_address)
message.contacts.append(
MessageContactAssociation(contact=contact,
field=field_name))
def update_contacts_from_message(db_session, message, namespace):
with db_session.no_autoflush:
# First create Contact objects for any email addresses that we haven't
# seen yet. We want to dedupe by canonicalized address, so this part is
# a bit finicky.
canonicalized_addresses = []
all_addresses = []
for field in (message.from_addr, message.to_addr, message.cc_addr, message.bcc_addr):
# We generally require these attributes to be non-null, but only
# set them to the default empty list at flush time. So it's better
# to be safe here.
if field is not None:
all_addresses.extend(field)
canonicalized_addresses = [canonicalize(addr) for _, addr in all_addresses]
existing_contacts = (
db_session.query(Contact)
.filter(Contact._canonicalized_address.in_(canonicalized_addresses), Contact.namespace_id == namespace.id)
.all()
)
contact_map = {c._canonicalized_address: c for c in existing_contacts}
for name, email_address in all_addresses:
canonicalized_address = canonicalize(email_address)
if canonicalized_address not in contact_map:
new_contact = Contact(
name=name,
email_address=email_address,
namespace=namespace,
provider_name=INBOX_PROVIDER_NAME,
uid=uuid.uuid4().hex,
)
contact_map[canonicalized_address] = new_contact
# Now associate each contact to the message.
for field_name in ("from_addr", "to_addr", "cc_addr", "bcc_addr"):
field = getattr(message, field_name)
if field is None:
continue
for name, email_address in field:
canonicalized_address = canonicalize(email_address)
contact = contact_map.get(canonicalized_address)
message.contacts.append(MessageContactAssociation(contact=contact, field=field_name))
def _get_contact_from_map(contact_map, name, email_address):
if not valid_email(email_address):
return
canonicalized_address = canonicalize(email_address)
contact = contact_map.get(canonicalized_address)
# Hackily address the condition that you get mail from e.g.
# "Ben Gotow (via Google Drive) <*****@*****.**" # noqa
# "Christine Spang (via Google Drive) <*****@*****.**" # noqa
# and so on: rather than creating many contacts with
# varying name, null out the name for the existing contact.
if contact.name != name and "noreply" in canonicalized_address:
contact.name = None
return contact
