def decorated_view(*args, **kwargs):
user = User.query(User.username == g.user).get()
if not user:
return redirect(url_for('login', next=request.url))
elif not user.is_admin_user():
return redirect(url_for('login', next=request.url))
return f(*args, **kwargs)
def login():
if request.method == 'GET':
return render_template('login.html')
if request.method == 'POST':
user = User.query(User.username == request.form['username']).get()
if user:
if user.check_password(request.form['password']):
session['logged_in'] = True
if user.is_superuser:
session['is_superuser'] = True
session['user'] = request.form['username']
flash(u'로그인 하였습니다.', 'success')
return redirect(url_for('home'))
else:
flash(u'잘못된 비밀번호 입니다.', 'danger')
return redirect(url_for('home'))
else:
flash(u'사용자가 존재하지 않습니다.', 'danger')
return redirect(url_for('home'))
def join():
if request.method == 'GET':
return render_template('join.html')
if request.method == 'POST':
user_exist = None
user_exist = User.query(User.username == request.form['username']).get()
print user_exist
user = User(
username=request.form['username'], password=generate_password_hash(request.form['password']), email=request.form['email']
)
exist = False
if not user_exist:
user_create_result = user.put()
logging.info(user_create_result)
flash(u'회원가입을 하였습니다.', 'success')
return redirect(url_for('home'))
else:
flash(u'중복되는 아이디가 존재합니다.', 'danger')
return redirect(url_for('join'))
