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
parent ed42a16b8e
commit 93292c88c9
1 changed files with 7 additions and 1 deletions

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: