注册sql model

2 files changed, 7 insertions, 1 deletions

diff --git a/.gitignore b/.gitignore
index 000b272..89bfd5e 100644
--- a/.gitignore
+++ b/.gitignore
@@ -3,3 +3,4 @@
 *.pyc
 ./server/app/__pycache__
 *.yaml
+.vscode/
\ No newline at end of file
diff --git a/server/app.py b/server/app.py
index ee6dda0..f4253ec 100644
--- a/server/app.py
+++ b/server/app.py
@@ -1,12 +1,13 @@
 from apiflask import APIFlask
 from flask_cors import CORS
-
+from exts import db
 import settings
 from apps.agentcomm import bp as agentbp
 from apps.sysinfo import bp as sysbp
 from apps.sysmange import bp as mangbp
 from apps.target import bp as targetbp
 from apps.task import bp as taskbp
+from apps.login import bp as loginbp
 
 # 注册蓝图
 app = APIFlask(__name__, template_folder='./static/templates')
@@ -22,7 +23,11 @@ app.register_blueprint(taskbp)
 app.register_blueprint(mangbp)
 # 系统状态获取接口
 app.register_blueprint(sysbp)
+# 登录退出接口
+app.register_blueprint(loginbp)
 
+app.config["SQLALCHEMY_DATABASE_URI"] = f"mysql+pymysql://root:{settings.MYSQL_PAWD}@{settings.MYSQL_HOST}:{settings.MYSQL_PORT}/{settings.MYSQL_DATADB}"
+db.init_app(app)
 
 @app.get('/')
 @app.doc("获取测试页面")