refactor(移除生产部署文件)
This commit is contained in:
@@ -0,0 +1,49 @@
|
||||
from flask import render_template, make_response
|
||||
from flask import session, redirect, url_for
|
||||
from flask_login import current_user, login_user
|
||||
from flask_restful import Resource, reqparse
|
||||
|
||||
from common.gen_captcha import add_auth_session
|
||||
from common.utils.http import fail_api, success_api
|
||||
from common.utils.rights import record_logging
|
||||
from models import UserModels
|
||||
|
||||
|
||||
class LoginResource(Resource):
|
||||
|
||||
def get(self):
|
||||
if current_user.is_authenticated:
|
||||
return redirect(url_for('admin.index'))
|
||||
return make_response(render_template('index/login.html'))
|
||||
|
||||
def post(self):
|
||||
login_req = reqparse.RequestParser(bundle_errors=True)
|
||||
login_req.add_argument('username', type=str, help='请输入用户名', required=True)
|
||||
login_req.add_argument('password', type=str, help='请输入密码', required=True)
|
||||
login_req.add_argument('captcha', type=str, help='请输入验证码', required=True)
|
||||
req = login_req.parse_args()
|
||||
|
||||
s_code = session.get("code", None)
|
||||
session["code"] = None
|
||||
|
||||
if req.captcha != s_code:
|
||||
return fail_api(message="验证码错误")
|
||||
user = UserModels.query.filter_by(username=req.username).first()
|
||||
|
||||
if user is None:
|
||||
return fail_api(message="不存在的用户")
|
||||
|
||||
if user.enable == 0:
|
||||
return fail_api(message="用户被暂停使用")
|
||||
|
||||
if user.validate_password(req.password):
|
||||
# 登录
|
||||
login_user(user)
|
||||
# 记录登录日志
|
||||
record_logging()
|
||||
|
||||
# 存入权限
|
||||
add_auth_session()
|
||||
return success_api(message="登录成功")
|
||||
record_logging()
|
||||
return fail_api(message="用户名或密码错误")
|
||||
Reference in New Issue
Block a user