🐞 fix: flask_app 重构

flask_app/__init__.py

@@ -0,0 +1,40 @@
+# encoding=utf8
+# app 工厂函数
+from flask import Flask
+from flask_sqlalchemy import SQLAlchemy
+import sys,os
+from loguru import logger
+
+
+# 判断系统
+WIN = sys.platform.startswith('win')
+if WIN:  # 如果是 Windows 系统，使用三个斜线
+    prefix = 'sqlite:///'
+else:  # 否则使用四个斜线
+    prefix = 'sqlite:////'
+
+# 日志处理
+logger.remove()
+logger.add(
+    sink=sys.stdout,
+    format="<green>{time:YYYY-MM-DD at HH:mm:ss}</green> - <level>{level}</level> - <level>{message}</level>",
+    colorize=True,
+    backtrace=True
+)
+
+
+
+app = Flask(__name__)
+app.config['SQLALCHEMY_DATABASE_URI'] = prefix + \
+    os.path.join(app.root_path, 'data.db')
+app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False  # 关闭对模型修改的监控
+db = SQLAlchemy(app)
+
+# buleprint
+from .views import main as main_blueprint
+app.register_blueprint(main_blueprint)
+
+
+
+
+

flask_app/app.py

@@ -1,122 +0,0 @@
-# encoding=utf8
-import json
-import time
-from flask import Flask, make_response, request, jsonify, render_template
-from flask_cors import CORS
-from urllib3 import disable_warnings
-
-from utils import *
-
-disable_warnings()
-
-app = Flask(__name__)
-CORS(app, supports_credentials=True, resources="/*")  # 跨域
-
-# 解决与 vue 冲突
-app.jinja_env.variable_start_string = '[['
-app.jinja_env.variable_end_string = ']]'
-
-def request_parse(req_data: request) -> dict:
-    '''解析请求数据并以字典的形式返回'''
-    if req_data.method == 'POST':
-        data = req_data.form
-
-    elif req_data.method == 'GET':
-        data = req_data.args
-
-    return dict(data)
-
-
-class BaseResponse(BaseModel):
-    """返回的响应"""
-    status: int = 0  # 状态码 0-->成功 1-->失败
-    msg: str = "前端显示的简短信息"
-    data: Optional[str]
-
-    @property
-    def resp(self):
-        '''BaseModel类型返回json'''
-        response = make_response(
-            json.dumps(
-                self.dict(),
-                ensure_ascii=False,
-                sort_keys=False
-            ),
-        )
-        response.mimetype = 'application/json'
-        # 跨域设置
-        response.headers['Access-Control-Allow-Origin'] = '*'
-        response.headers['Access-Control-Allow-Methods'] = 'OPTIONS,HEAD,GET,POST'
-        response.headers['Access-Control-Allow-Headers'] = 'x-requested-with'
-        return response
-
-
-@app.route("/", methods=['GET'])
-def index():
-    return render_template('admin.html')
-
-@app.route("/testapi/", methods=['POST'])
-@logger.catch
-def testapi():
-    brs = BaseResponse()
-    # 需要传入 json 数据
-    try:
-        jsonData = request.get_json()
-        api = API(**jsonData)
-        phone = jsonData.get('phone')
-        if not phone:
-            raise ValueError("参数 phone 没有！")
-        try:
-            resp = test_resq(api, phone)
-            brs.status = 0
-            brs.msg = f"请求成功！{resp}"
-            brs.data = resp.text
-        except Exception as why:
-            brs.status = 1
-            brs.msg = f"请求失败：{why}"
-    except Exception as why:
-        brs.status = 1
-        brs.msg = f"参数有误：{why}"
-    return brs.resp
-
-
-@app.route("/submitapi/", methods=['POST'])
-def submitapi():
-    """提交API到json文件"""
-    # 需要传入 json 数据
-    jsonData = request.get_json()
-    api = API(**jsonData).handle_API()
-    data = json.loads(json_path.read_text(encoding='utf8'))
-    with open(json_path, mode="w", encoding="utf8") as j:
-        try:
-            data.append(api.dict())
-            json.dump(data, j, ensure_ascii=False, sort_keys=False)
-            return BaseResponse(status=0, msg="写入成功!").resp
-        except Exception as why:
-            return BaseResponse(status=1, msg=f"写入失败!{why}").resp
-
-
-@app.route("/backapi/", methods=['GET', 'POST'])
-def backjson():
-    """备份json文件"""
-    try:
-        timeStruct = time.localtime(int(time.time()))
-        strTime = time.strftime("%Y_%m_%d_%H_%M_%S", timeStruct)
-        Path(json_path.parent, 'apiback').mkdir(exist_ok=True)
-        json_back_path = Path(json_path.parent, 'apiback',
-                              f"api_back_{strTime}.json")
-        with open(json_back_path, mode="w") as j:
-            j_data = json.loads(json_path.read_text(encoding='utf8'))
-            json.dump(j_data, j, ensure_ascii=False, sort_keys=False)
-        return BaseResponse(status=0, msg="备份成功!").resp
-    except Exception as why:
-        return BaseResponse(status=1, msg=f"备份失败{why}").resp
-
-
-@app.route("/downloadapi/", methods=['GET'])
-def downloadapi():
-    """下载接口文件"""
-    return json_path.read_text(encoding='utf8')
-
-
-app.run(host="0.0.0.0", port=10981, debug=True)

flask_app/model.py

@@ -0,0 +1,13 @@
+# encoding=utf8
+# 储存数据库模型
+from . import db
+from datetime import datetime
+
+class Apis(db.Model):
+    id = db.Column(db.Integer, primary_key=True)  # 主键
+    desc = db.Column(db.String(100))  # 描述
+    url = db.Column(db.String(100))  # 链接
+    method = db.Column(db.String(10))  # 请求方法
+    header = db.Column(db.String(9999))  # 请求头
+    data = db.Column(db.String(9999))  # 请求数据
+    add_time = db.Column(db.DateTime(), default=datetime.now())  # 添加时间

flask_app/views/__init__.py

@@ -0,0 +1,6 @@
+from flask import Blueprint
+
+main = Blueprint("main",__name__)
+
+from . import views
+from .. import db

flask_app/views/views.py

@@ -0,0 +1,9 @@
+# encoding=utf8
+# flask app views
+from . import main
+from .. import db
+
+@main.route("/",methods=['GET','POST'])
+def index():
+    return "index"
+