rename public_domain -> sl_domain if applicable

This commit is contained in:
Son NK 2020-10-15 16:52:38 +02:00
parent 4a32db5b5d
commit dbf0404aa9
2 changed files with 14 additions and 14 deletions

View File

@ -200,15 +200,15 @@ def setting():
default_domain = request.form.get("random-alias-default-domain")
if default_domain:
public_domain: SLDomain = SLDomain.get_by(domain=default_domain)
if public_domain:
if public_domain.premium_only and not current_user.is_premium():
sl_domain: SLDomain = SLDomain.get_by(domain=default_domain)
if sl_domain:
if sl_domain.premium_only and not current_user.is_premium():
flash("You cannot use this domain", "error")
return redirect(url_for("dashboard.setting"))
# make sure only default_random_alias_domain_id or default_random_alias_public_domain_id is set
current_user.default_random_alias_public_domain_id = (
public_domain.id
sl_domain.id
)
current_user.default_random_alias_domain_id = None
else:

View File

@ -516,19 +516,19 @@ class User(db.Model, ModelMixin, UserMixin):
return custom_domain.domain
if self.default_random_alias_public_domain_id:
public_domain = SLDomain.get(self.default_random_alias_public_domain_id)
sl_domain = SLDomain.get(self.default_random_alias_public_domain_id)
# sanity check
if not public_domain:
if not sl_domain:
LOG.exception("Problem with %s public random alias domain", self)
return FIRST_ALIAS_DOMAIN
if public_domain.premium_only and not self.is_premium():
if sl_domain.premium_only and not self.is_premium():
LOG.exception(
"%s is not premium and cannot use %s", self, public_domain
"%s is not premium and cannot use %s", self, sl_domain
)
return FIRST_ALIAS_DOMAIN
return public_domain.domain
return sl_domain.domain
return FIRST_ALIAS_DOMAIN
@ -567,7 +567,7 @@ class User(db.Model, ModelMixin, UserMixin):
else:
query = SLDomain.filter_by(premium_only=False)
domains = [public_domain.domain for public_domain in query]
domains = [sl_domain.domain for sl_domain in query]
return domains
@ -989,14 +989,14 @@ class Alias(db.Model, ModelMixin):
scheme=scheme, in_hex=in_hex, alias_domain=custom_domain.domain
)
elif user.default_random_alias_public_domain_id:
public_domain: SLDomain = SLDomain.get(
sl_domain: SLDomain = SLDomain.get(
user.default_random_alias_public_domain_id
)
if public_domain.premium_only and not user.is_premium():
LOG.exception("%s not premium, cannot use %s", user, public_domain)
if sl_domain.premium_only and not user.is_premium():
LOG.exception("%s not premium, cannot use %s", user, sl_domain)
else:
random_email = generate_email(
scheme=scheme, in_hex=in_hex, alias_domain=public_domain.domain
scheme=scheme, in_hex=in_hex, alias_domain=sl_domain.domain
)
if not random_email: