Merge branch 'master' of https://www.github.com/failedxyz/easyctf
This commit is contained in:
commit
9789d69851
1 changed files with 12 additions and 3 deletions
|
@ -4,6 +4,8 @@ from flask import current_app as app
|
||||||
from models import db, Users
|
from models import db, Users
|
||||||
from utils import api_wrapper
|
from utils import api_wrapper
|
||||||
|
|
||||||
|
import utils
|
||||||
|
|
||||||
blueprint = Blueprint("user", __name__)
|
blueprint = Blueprint("user", __name__)
|
||||||
|
|
||||||
@blueprint.route("/register", methods=["POST"])
|
@blueprint.route("/register", methods=["POST"])
|
||||||
|
@ -38,13 +40,20 @@ def user_register():
|
||||||
@blueprint.route("/logout", methods=["POST"])
|
@blueprint.route("/logout", methods=["POST"])
|
||||||
@api_wrapper
|
@api_wrapper
|
||||||
def user_logout():
|
def user_logout():
|
||||||
# session.clear()
|
session.clear()
|
||||||
pass
|
|
||||||
|
|
||||||
@blueprint.route("/login", methods=["POST"])
|
@blueprint.route("/login", methods=["POST"])
|
||||||
@api_wrapper
|
@api_wrapper
|
||||||
def user_login():
|
def user_login():
|
||||||
pass
|
username = request.form["username"]
|
||||||
|
password = request.form["password"]
|
||||||
|
user = Users.query.filter_by(username=username).first()
|
||||||
|
if utils.check_password(user.password, password):
|
||||||
|
session["username"] = username
|
||||||
|
session["admin"] = user.admin
|
||||||
|
return { "success": 1, "message": "Success!" }
|
||||||
|
else:
|
||||||
|
return { "success": 0, "message": "Invalid credentials." }
|
||||||
|
|
||||||
def add_user(name, username, email, password):
|
def add_user(name, username, email, password):
|
||||||
user = Users(name, username, email, password)
|
user = Users(name, username, email, password)
|
||||||
|
|
Loading…
Reference in a new issue