use can_be_used_as_personal_email when user registers or change emails
This commit is contained in:
parent
fc7a832969
commit
99ffd6149d
|
@ -11,6 +11,7 @@ from app.extensions import db
|
|||
from app.log import LOG
|
||||
from app.models import User
|
||||
from .login_utils import after_login
|
||||
from ...email_utils import can_be_used_as_personal_email
|
||||
|
||||
_authorization_base_url = "https://www.facebook.com/dialog/oauth"
|
||||
_token_url = "https://graph.facebook.com/oauth/access_token"
|
||||
|
@ -102,6 +103,12 @@ def facebook_callback():
|
|||
|
||||
# create user
|
||||
else:
|
||||
if not can_be_used_as_personal_email(email):
|
||||
flash(
|
||||
f"You cannot use {email} as your personal inbox.", "error",
|
||||
)
|
||||
return redirect(url_for("auth.login"))
|
||||
|
||||
LOG.d("create facebook user with %s", facebook_user_data)
|
||||
user = User.create(
|
||||
email=email.lower(), name=facebook_user_data["name"], activated=True
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
from flask import request, session, redirect, flash
|
||||
from flask import request, session, redirect, flash, url_for
|
||||
from flask_login import login_user
|
||||
from requests_oauthlib import OAuth2Session
|
||||
|
||||
|
@ -6,6 +6,7 @@ from app import email_utils
|
|||
from app.auth.base import auth_bp
|
||||
from app.auth.views.login_utils import after_login
|
||||
from app.config import GITHUB_CLIENT_ID, GITHUB_CLIENT_SECRET, URL
|
||||
from app.email_utils import can_be_used_as_personal_email
|
||||
from app.extensions import db
|
||||
from app.log import LOG
|
||||
from app.models import User
|
||||
|
@ -84,6 +85,12 @@ def github_callback():
|
|||
|
||||
# create user
|
||||
if not user:
|
||||
if not can_be_used_as_personal_email(email):
|
||||
flash(
|
||||
f"You cannot use {email} as your personal inbox.", "error",
|
||||
)
|
||||
return redirect(url_for("auth.login"))
|
||||
|
||||
LOG.d("create github user")
|
||||
user = User.create(
|
||||
email=email.lower(), name=github_user_data.get("name") or "", activated=True
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
from flask import request, session, redirect, flash
|
||||
from flask import request, session, redirect, flash, url_for
|
||||
from flask_login import login_user
|
||||
from requests_oauthlib import OAuth2Session
|
||||
|
||||
|
@ -10,6 +10,7 @@ from app.log import LOG
|
|||
from app.models import User, File
|
||||
from app.utils import random_string
|
||||
from .login_utils import after_login
|
||||
from ...email_utils import can_be_used_as_personal_email
|
||||
|
||||
_authorization_base_url = "https://accounts.google.com/o/oauth2/v2/auth"
|
||||
_token_url = "https://www.googleapis.com/oauth2/v4/token"
|
||||
|
@ -92,6 +93,12 @@ def google_callback():
|
|||
db.session.commit()
|
||||
# create user
|
||||
else:
|
||||
if not can_be_used_as_personal_email(email):
|
||||
flash(
|
||||
f"You cannot use {email} as your personal inbox.", "error",
|
||||
)
|
||||
return redirect(url_for("auth.login"))
|
||||
|
||||
LOG.d("create google user with %s", google_user_data)
|
||||
user = User.create(
|
||||
email=email.lower(), name=google_user_data["name"], activated=True
|
||||
|
|
|
@ -6,7 +6,7 @@ from wtforms import StringField, validators
|
|||
from app import email_utils
|
||||
from app.auth.base import auth_bp
|
||||
from app.config import URL
|
||||
from app.email_utils import email_belongs_to_alias_domains
|
||||
from app.email_utils import can_be_used_as_personal_email
|
||||
from app.extensions import db
|
||||
from app.log import LOG
|
||||
from app.models import User, ActivationCode
|
||||
|
@ -32,12 +32,11 @@ def register():
|
|||
|
||||
if form.validate_on_submit():
|
||||
email = form.email.data
|
||||
if email_belongs_to_alias_domains(email):
|
||||
if not can_be_used_as_personal_email(email):
|
||||
flash(
|
||||
"You cannot use alias as your personal inbox. Nice try though 😉",
|
||||
"error",
|
||||
"You cannot use this email address as your personal inbox.", "error",
|
||||
)
|
||||
|
||||
else:
|
||||
user = User.filter_by(email=email).first()
|
||||
|
||||
if user:
|
||||
|
|
|
@ -11,7 +11,10 @@ from wtforms import StringField, validators
|
|||
from app import s3, email_utils
|
||||
from app.config import URL
|
||||
from app.dashboard.base import dashboard_bp
|
||||
from app.email_utils import email_belongs_to_alias_domains
|
||||
from app.email_utils import (
|
||||
email_belongs_to_alias_domains,
|
||||
can_be_used_as_personal_email,
|
||||
)
|
||||
from app.extensions import db
|
||||
from app.log import LOG
|
||||
from app.models import (
|
||||
|
@ -93,9 +96,9 @@ def setting():
|
|||
or DeletedAlias.get_by(email=new_email)
|
||||
):
|
||||
flash(f"Email {new_email} already used", "error")
|
||||
elif email_belongs_to_alias_domains(new_email):
|
||||
elif not can_be_used_as_personal_email(new_email):
|
||||
flash(
|
||||
"You cannot use alias as your personal inbox. Nice try though 😉",
|
||||
"You cannot use this email address as your personal inbox.",
|
||||
"error",
|
||||
)
|
||||
else:
|
||||
|
|
Loading…
Reference in a new issue