def test_find_partner_from_email(self):
email_and_expected_partner = [
('*****@*****.**', self.partner),
('*****@*****.**', self.partner),
('*****@*****.**', None),
(None, None),
]
for email, expected_partner in email_and_expected_partner:
partner = helpers.find_partner_from_email([self.partner], email)
self.assertEqual(partner, expected_partner)
def test_find_partner_from_email(self):
email_and_expected_partner = [
('*****@*****.**', self.partner),
('*****@*****.**', self.partner),
('*****@*****.**', None),
(None, None),
]
for email, expected_partner in email_and_expected_partner:
partner = helpers.find_partner_from_email([self.partner], email)
self.assertEqual(partner, expected_partner)
def process_email(request):
"""
Creates a contact record from an email received via POST.
"""
if request.method != 'POST':
return HttpResponse(status=200)
admin_email = request.REQUEST.get('from')
headers = request.REQUEST.get('headers')
key = request.REQUEST.get('key')
subject = request.REQUEST.get('subject')
email_text = request.REQUEST.get('text')
if key != settings.EMAIL_KEY:
return HttpResponse(status=200)
if headers:
parser = HeaderParser()
headers = parser.parsestr(headers)
if headers and 'Date' in headers:
try:
date_time = get_datetime_from_str(headers.get('Date'))
except Exception:
date_time = now()
else:
date_time = now()
to = request.REQUEST.get('to', '')
cc = request.REQUEST.get('cc', '')
recipient_emails_and_names = getaddresses(["%s, %s" % (to, cc)])
admin_email = getaddresses([admin_email])[0][1]
contact_emails = filter(None,
[email[1] for email in recipient_emails_and_names])
if contact_emails == [] or (len(contact_emails) == 1 and
contact_emails[0].lower() == '*****@*****.**'):
# If [email protected] is the only contact, assume it's a forward.
fwd_headers = build_email_dicts(email_text)
try:
recipient_emails_and_names = fwd_headers[0]['recipients']
contact_emails = [recipient[1] for recipient
in recipient_emails_and_names]
date_time = fwd_headers[0]['date']
except IndexError:
contact_emails = []
# Prevent duplicate contact records for an email address because
# the address was in both To and CC.
contact_emails = list(set(contact_emails))
validated_contacts = []
for element in contact_emails:
if not element.lower() == '*****@*****.**' and validate_email(element):
validated_contacts.append(element)
contact_emails = validated_contacts
try:
contact_emails.remove(admin_email)
except ValueError:
pass
admin_user = User.objects.get_email_owner(admin_email)
if admin_user is None:
return HttpResponse(status=200)
if admin_user.company_set.count() > 1:
error = "Your account is setup as the admin for multiple companies. " \
"Because of this we cannot match this email with a " \
"specific partner on a specific company with 100% certainty. " \
"You will need to login to My.jobs and go to " \
"https://secure.my.jobs/prm to create your record manually."
send_contact_record_email_response([], [], [], contact_emails,
error, admin_email)
return HttpResponse(status=200)
partners = list(chain(*[company.partner_set.all()
for company in admin_user.company_set.all()]))
possible_contacts, created_contacts, unmatched_contacts = [], [], []
for contact in contact_emails:
try:
matching_contacts = Contact.objects.filter(email=contact,
partner__in=partners)
[possible_contacts.append(x) for x in matching_contacts]
if not matching_contacts:
poss_partner = find_partner_from_email(partners, contact)
if poss_partner:
new_contact = Contact.objects.create(name=contact,
email=contact,
partner=poss_partner)
change_msg = "Contact was created from email."
log_change(new_contact, None, admin_user,
new_contact.partner, new_contact.name,
action_type=ADDITION, change_msg=change_msg)
created_contacts.append(new_contact)
else:
unmatched_contacts.append(contact)
except ValueError:
unmatched_contacts.append(contact)
num_attachments = int(request.POST.get('attachments', 0))
attachments = []
for file_number in range(1, num_attachments+1):
try:
attachment = request.FILES['attachment%s' % file_number]
except KeyError:
error = "There was an issue with the email attachments. The " \
"contact records for the email will need to be created " \
"manually."
send_contact_record_email_response([], [], [], contact_emails,
error, admin_email)
return HttpResponse(status=200)
attachments.append(attachment)
created_records = []
attachment_failures = []
date_time = now() if not date_time else date_time
all_contacts = possible_contacts + created_contacts
if not all_contacts:
error = "No contacts or contact records could be created " \
"from this email. You will need to log in and manually " \
"create contact records for this email."
send_contact_record_email_response([], [], [], contact_emails,
error, admin_email)
return HttpResponse(status=200)
for contact in all_contacts:
change_msg = "Email was sent by %s to %s" % \
(admin_user.get_full_name(), contact.name)
record = ContactRecord.objects.create(partner=contact.partner,
contact_type='email',
contact=contact,
contact_email=contact.email,
contact_phone=contact.phone,
created_by=admin_user,
date_time=date_time,
subject=subject,
notes=force_text(email_text))
try:
for attachment in attachments:
prm_attachment = PRMAttachment()
prm_attachment.attachment = attachment
prm_attachment.contact_record = record
setattr(prm_attachment, 'partner', contact.partner)
prm_attachment.save()
# The file pointer for this attachment is now at the end of the
# file; reset it to the beginning for future use.
attachment.seek(0)
except AttributeError:
attachment_failures.append(record)
log_change(record, None, admin_user, contact.partner, contact.name,
action_type=ADDITION, change_msg=change_msg)
created_records.append(record)
send_contact_record_email_response(created_records, created_contacts,
attachment_failures, unmatched_contacts,
None, admin_email)
return HttpResponse(status=200)
def test_find_partner_from_email_no_partners(self):
partner = helpers.find_partner_from_email(None, '*****@*****.**')
self.assertIsNone(partner)
def test_find_partner_from_email_no_partners(self):
partner = helpers.find_partner_from_email(None, '*****@*****.**')
self.assertIsNone(partner)
