diff options
author | rsiddharth <s@ricketyspace.net> | 2019-09-13 22:16:19 -0400 |
---|---|---|
committer | rsiddharth <s@ricketyspace.net> | 2019-09-13 22:16:19 -0400 |
commit | 879b5e3d624b64fa6fbe1d5a4bbe125ad568f567 (patch) | |
tree | 0c29682fd902c07853c072d233ab5b1afd7401fb | |
parent | 2e7e1c2aaa8ae69a88de026ba6f13c16bc835e74 (diff) |
nfsw/auth.py: Add load_logged_in_user
It's a bp.before_app_request function.
-rw-r--r-- | nfsw/auth.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/nfsw/auth.py b/nfsw/auth.py index 654f6c2..697f89e 100644 --- a/nfsw/auth.py +++ b/nfsw/auth.py @@ -165,3 +165,17 @@ def logout(): return redirect(url_for('index')) +@bp.before_app_request +def load_logged_in_user(): + user_id = session.get('user_id') + + if user_id is None: + g.user = None + else: + g.user = get_db().execute( + 'SELECT * FROM user WHERE id=?', (user_id,) + ).fetchone() + + if (request.endpoint not in ['auth.terms', 'auth.sorry'] + and g.user['terms_agreed'] != 1): + return redirect(url_for('auth.terms')) |