Milliways/lenticular_cloud2
4
0
Fork 0
Code Issues 1 Pull requests Packages Projects Releases Wiki Activity

fix not exist user

Browse source
This commit is contained in:
TuxCoder 2020-05-13 20:08:28 +02:00
parent 5037fe4def
commit aaf428575b
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
lenticular_cloud/views/auth.py
View file

@ -48,9 +48,13 @@ def login():
form = LoginForm() form = LoginForm()
if form.validate_on_submit(): if form.validate_on_submit():
user = User.query().by_username(form.data['name']) user = User.query().by_username(form.data['name'])
session['username'] = str(user.username) if user:
session['username'] = str(user.username)
else:
session['user'] = None
session['auth_providers'] = [] session['auth_providers'] = []
return redirect(url_for('auth.login_auth', next=flask.request.args.get('next'))) return redirect(url_for('auth.login_auth', next=flask.request.args.get('next')))
return render_template('frontend/login.html.j2', form=form) return render_template('frontend/login.html.j2', form=form)
@ -78,8 +82,8 @@ def login_auth():
_host_url = urlparse(request.url) _host_url = urlparse(request.url)
if _next_url.scheme == _host_url.scheme and _next_url.netloc == _host_url.netloc : if _next_url.scheme == _host_url.scheme and _next_url.netloc == _host_url.netloc :
_next = _next_url.geturl() _next = _next_url.geturl()
except Exception as e: except TypeError:
raise e _next = None
return redirect(_next or url_for('frontend.index')) return redirect(_next or url_for('frontend.index'))
return render_template('frontend/login_auth.html.j2', forms=auth_forms) return render_template('frontend/login_auth.html.j2', forms=auth_forms)