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'))