mirror of
https://github.com/simple-login/app.git
synced 2024-11-16 00:48:32 +01:00
90 lines
3 KiB
Python
90 lines
3 KiB
Python
|
from dataclasses import dataclass
|
||
|
|
from enum import Enum
|
||
|
|
from typing import Optional
|
||
|
|
|
||
|
|
from sqlalchemy.exc import IntegrityError
|
||
|
|
|
||
|
|
from app.db import Session
|
||
|
|
from app.email_utils import generate_reply_email
|
||
|
|
from app.email_validation import is_valid_email
|
||
|
|
from app.log import LOG
|
||
|
|
from app.models import Contact, Alias
|
||
|
|
from app.utils import sanitize_email
|
||
|
|
|
||
|
|
|
||
|
|
class ContactCreateError(Enum):
|
||
|
|
InvalidEmail = "Invalid email"
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class ContactCreateResult:
|
||
|
|
contact: Optional[Contact]
|
||
|
|
error: Optional[ContactCreateError]
|
||
|
|
|
||
|
|
|
||
|
|
def __update_contact_if_needed(
|
||
|
|
contact: Contact, name: Optional[str], mail_from: Optional[str]
|
||
|
|
) -> ContactCreateResult:
|
||
|
|
if name and contact.name != name:
|
||
|
|
LOG.d(f"Setting {contact} name to {name}")
|
||
|
|
contact.name = name
|
||
|
|
Session.commit()
|
||
|
|
if mail_from and contact.mail_from is None:
|
||
|
|
LOG.d(f"Setting {contact} mail_from to {mail_from}")
|
||
|
|
contact.mail_from = mail_from
|
||
|
|
Session.commit()
|
||
|
|
return ContactCreateResult(contact, None)
|
||
|
|
|
||
|
|
|
||
|
|
def create_contact(
|
||
|
|
email: str,
|
||
|
|
name: Optional[str],
|
||
|
|
alias: Alias,
|
||
|
|
mail_from: Optional[str] = None,
|
||
|
|
allow_empty_email: bool = False,
|
||
|
|
automatic_created: bool = False,
|
||
|
|
from_partner: bool = False,
|
||
|
|
) -> ContactCreateResult:
|
||
|
|
if name is not None:
|
||
|
|
name = name[: Contact.MAX_NAME_LENGTH]
|
||
|
|
if name is not None and "\x00" in name:
|
||
|
|
LOG.w("Cannot use contact name because has \\x00")
|
||
|
|
name = ""
|
||
|
|
if not is_valid_email(email):
|
||
|
|
LOG.w(f"invalid contact email {email}")
|
||
|
|
if not allow_empty_email:
|
||
|
|
return ContactCreateResult(None, ContactCreateError.InvalidEmail)
|
||
|
|
LOG.d("Create a contact with invalid email for %s", alias)
|
||
|
|
# either reuse a contact with empty email or create a new contact with empty email
|
||
|
|
email = ""
|
||
|
|
email = sanitize_email(email, not_lower=True)
|
||
|
|
contact = Contact.get_by(alias_id=alias.id, website_email=email)
|
||
|
|
if contact is not None:
|
||
|
|
return __update_contact_if_needed(contact, name, mail_from)
|
||
|
|
reply_email = generate_reply_email(email, alias)
|
||
|
|
try:
|
||
|
|
flags = Contact.FLAG_PARTNER_CREATED if from_partner else 0
|
||
|
|
contact = Contact.create(
|
||
|
|
user_id=alias.user_id,
|
||
|
|
alias_id=alias.id,
|
||
|
|
website_email=email,
|
||
|
|
name=name,
|
||
|
|
reply_email=reply_email,
|
||
|
|
mail_from=mail_from,
|
||
|
|
automatic_created=automatic_created,
|
||
|
|
flags=flags,
|
||
|
|
invalid_email=email == "",
|
||
|
|
commit=True,
|
||
|
|
)
|
||
|
|
LOG.d(
|
||
|
|
f"Created contact {contact} for alias {alias} with email {email} invalid_email={contact.invalid_email}"
|
||
|
|
)
|
||
|
|
except IntegrityError:
|
||
|
|
Session.rollback()
|
||
|
|
LOG.info(
|
||
|
|
f"Contact with email {email} for alias_id {alias.id} already existed, fetching from DB"
|
||
|
|
)
|
||
|
|
contact = Contact.get_by(alias_id=alias.id, website_email=email)
|
||
|
|
return __update_contact_if_needed(contact, name, mail_from)
|
||
|
|
return ContactCreateResult(contact, None)
|