def new_function(*args, **kwargs): # Assume queried user is the first argument queried_user_identifier = kwargs.get('queried_user_identifier', None) if queried_user_identifier == None: pass elif queried_user_identifier.lower() == 'me': g.queried_user_id = session.get('user_id', None) if get_object: g.queried_user = g.get('user', None) else: try: user_id = int(queried_user_identifier) g.queried_user_id = user_id if get_object: g.queried_user = user_service.get_user_by_id(user_id) except ValueError as e: return { "error" : True, "reason" : "notfound" } if self_check and g.queried_user_id == session.get('user_id', None): return { "error" : True, "reason" : "self" } return original_function(*args, **kwargs)
def new_function(*args, **kwargs): if 'user_id' in session: g.user = user_service.get_user_by_id(session['user_id']) return original_function(*args, **kwargs) else: return "Log in", 401