def current_user():
if system.get_env() == 'testing':
if request.headers.get('REMOTE_USER', None):
user = User.objects.get(id=request.headers['REMOTE_USER'])
return user.to_json()
return 'None'
return session.get('user', None)
def decorated_function(*args, **kwargs):
if system.get_env() == 'testing': return f(*args, **kwargs)
if g.user is None:
return redirect(url_for('home.login', next=request.url))
elif json.loads(g.user).get('permission', 1) < code:
abort(403)
return f(*args, **kwargs)
def authenticate(username, password):
try:
user = User.objects.get(username=username)
encrypt_pwd = generate_encrypt_password(password)
if encrypt_pwd == user.password:
if system.get_env() != 'testing': session['user'] = user.to_json()
return True
else:
return False
except:
return False
def decorated_function(*args, **kwargs):
if g.user is None and system.get_env() != 'testing':
return redirect(url_for('home.login', next=request.url))
return f(*args, **kwargs)
