def check_user_before_request():
if "user" not in session:
g.user = BasicUser.gen_user()
session["user"] = g.user.to_dict()
else:
userid = session.get('user')["id"]
user = User.get_user_by_id(userid)
# this for when clear db usage
if user is None:
user = BasicUser.gen_user()
session["user"] = user.to_dict()
g.user = user
def decorated_function(*args, **kwargs):
r = request.path
if "user" not in session:
flash("You need to login first!", "danger")
return redirect(url_for('login', r=r))
else:
g.user = User.get_user_by_id(session["user"]["id"])
if g.user is None:
return redirect(url_for('login', r=r))
def inject_user():
if "user" not in g:
g.user = User.get_user_by_id(session.get("user", {}).get("id"))
return dict(cur_user=g.user)
app.template_context_processors[None].append(inject_user)
return f(*args, **kwargs)
def inject_user():
if "user" not in g:
g.user = User.get_user_by_id(session.get("user", {}).get("id"))
return dict(cur_user=g.user)
