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)