Skip to content
Snippets Groups Projects
Commit 2c4b003f authored by kaiyou's avatar kaiyou
Browse files

Simplify the profile pick function

parent 8eb09d20
No related branches found
No related tags found
1 merge request!20Add 'remember me' button
Pipeline #1068 passed
......@@ -15,18 +15,14 @@ from hiboo.profile import login, admin, forms, cli
def get_profile(service, **redirect_args):
query = models.Profile.query.filter_by(
service_uuid=service.uuid,
user_uuid=flask_login.current_user.uuid,
status=models.Profile.ACTIVE
)
profiles = models.Profile.filter(service, flask_login.current_user)
active = profiles.filter_by(status=models.Profile.ACTIVE)
form = forms.ProfilePickForm()
if form.validate_on_submit():
return query.filter_by(uuid=form.profile_uuid.data).first()
count = query.count()
if count > 1:
return active.filter_by(uuid=form.profile_uuid.data).first()
if active.count() == 1:
return active.one()
elif profile.count() > 1:
utils.force_redirect(utils.url_for("profile.pick", **redirect_args))
elif count == 1:
return query.one()
else:
utils.force_redirect(utils.url_for("profile.create_quick", **redirect_args))
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment