基本构建

2021-03-18 22:59:02 +08:00
parent 3cf62d4305
commit d2a2b35317
638 changed files with 198871 additions and 0 deletions
@@ -0,0 +1,94 @@
+from flask import Blueprint, render_template, jsonify, request, session, redirect, url_for
+from flask_login import login_user, login_required, logout_user, current_user
+from applications.models.admin import User
+from applications.service.admin.index import add_auth_session, make_menu_tree, get_captcha
+from flask_marshmallow import Marshmallow
+from applications.service.route_auth import PreAuthorize
+from applications.service.admin_log import login_log
+
+
+admin_index = Blueprint('adminIndex', __name__, url_prefix='/admin')
+
+ma = Marshmallow()
+
+
+# 首页
+@admin_index.route('/')
+@PreAuthorize('index:index')
+def index():
+    username = current_user.username
+    return render_template('admin/index.html', username=username)
+
+
+# 获取验证码
+@admin_index.route('/getCaptcha', methods=["GET"])
+def getCaptcha():
+    resp,code=get_captcha()
+    session["code"] = code
+    return resp
+
+
+# 登录
+@admin_index.route('/login', methods=['GET', 'POST'])
+def login():
+    if request.method == 'POST':
+        req = request.form
+        username = req.get('username')
+        password = req.get('password')
+        code = req.get('captcha')
+
+        if not username or not password or not code:
+            return jsonify(msg="用户名或密码没有输入", code=0,success=False)
+        s_code = session.get("code", None)
+
+        if not all([code, s_code]):
+            return jsonify(msg="参数错误", code=0,success=False)
+
+        if code != s_code:
+            return jsonify(msg="验证码错误", code=0,success=False)
+        user = User.query.filter_by(username=username).first()
+
+        if user is None:
+            return jsonify(msg="不存在的用户", code=0,success=False)
+
+        if user.enable is 0:
+            return jsonify(msg="用户被暂停使用", code=0,success=False)
+
+        if username == user.username and user.validate_password(password):
+            # 登录
+            login_user(user)
+            # 记录登录日志
+            login_log(request)
+            # 存入权限
+            add_auth_session()
+            return jsonify(msg="登录成功", code=1,success=True)
+
+        return jsonify(msg="用户名或密码错误", code=0,success=False)
+    print(current_user.is_authenticated)
+    if current_user.is_authenticated:
+        return redirect(url_for('adminIndex.index'))
+    return render_template('admin/login.html')
+
+
+# 退出登录
+@admin_index.route('/logout', methods=['GET', 'POST'])
+@login_required
+def logout():
+    logout_user()
+    session.pop('permissions')
+    return jsonify(msg="注销成功", success=True)
+
+
+# 菜单
+@admin_index.route('/menu')
+@login_required
+def menu():
+    menu_tree = make_menu_tree()
+    return jsonify(menu_tree)
+
+
+# 控制台页面
+@admin_index.route('/welcome')
+@login_required
+def welcome():
+    return render_template('admin/console/console.html')