Fix not setting custom domain when creating alias via API

This commit is contained in:
Son NK 2020-03-15 23:18:43 +01:00 committed by Son NK
parent 0c0cebf970
commit abd23cf08f

View file

@ -7,7 +7,7 @@ from app.config import MAX_NB_EMAIL_FREE_PLAN
from app.dashboard.views.custom_alias import verify_prefix_suffix from app.dashboard.views.custom_alias import verify_prefix_suffix
from app.extensions import db from app.extensions import db
from app.log import LOG from app.log import LOG
from app.models import GenEmail, AliasUsedOn, User from app.models import GenEmail, AliasUsedOn, User, CustomDomain
from app.utils import convert_to_id from app.utils import convert_to_id
@ -61,6 +61,12 @@ def new_custom_alias():
gen_email = GenEmail.create( gen_email = GenEmail.create(
user_id=user.id, email=full_alias, mailbox_id=user.default_mailbox_id, note=note user_id=user.id, email=full_alias, mailbox_id=user.default_mailbox_id, note=note
) )
if alias_suffix.startswith("@"):
alias_domain = alias_suffix[1:]
domain = CustomDomain.get_by(domain=alias_domain)
gen_email.custom_domain_id = domain.id
db.session.commit() db.session.commit()
if hostname: if hostname: