Merge pull request #113 from simple-login/fix-custom-domain-not-set

Fix custom domain not set
This commit is contained in:
Son Nguyen Kim 2020-03-15 23:38:25 +01:00 committed by GitHub
commit 1e10b4d675
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 2 deletions

View File

@ -3,11 +3,11 @@ from flask import jsonify, request
from flask_cors import cross_origin from flask_cors import cross_origin
from app.api.base import api_bp, verify_api_key from app.api.base import api_bp, verify_api_key
from app.config import MAX_NB_EMAIL_FREE_PLAN from app.config import MAX_NB_EMAIL_FREE_PLAN, ALIAS_DOMAINS
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,14 @@ 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:]
if alias_domain not in ALIAS_DOMAINS:
domain = CustomDomain.get_by(domain=alias_domain)
LOG.d("set alias %s to domain %s", full_alias, domain)
gen_email.custom_domain_id = domain.id
db.session.commit() db.session.commit()
if hostname: if hostname: