From 93292c88c9b0dd32043a22e1c9c280f363e3cc32 Mon Sep 17 00:00:00 2001 From: Son NK Date: Sun, 15 Mar 2020 23:18:43 +0100 Subject: [PATCH] Fix not setting custom domain when creating alias via API --- app/api/views/new_custom_alias.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/app/api/views/new_custom_alias.py b/app/api/views/new_custom_alias.py index b6f77c11..e8eb117e 100644 --- a/app/api/views/new_custom_alias.py +++ b/app/api/views/new_custom_alias.py @@ -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.extensions import db 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 @@ -61,6 +61,12 @@ def new_custom_alias(): gen_email = GenEmail.create( 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() if hostname: