155 lines
5.3 KiB
Python
155 lines
5.3 KiB
Python
from flask import Flask, g
|
||
from flask_cors import CORS
|
||
import sqlite3
|
||
import os
|
||
import re
|
||
|
||
|
||
def create_app():
|
||
app = Flask(__name__)
|
||
|
||
# 加载配置文件
|
||
app.config.from_pyfile('config.py')
|
||
|
||
# 数据库路径
|
||
db_path = os.path.join(os.getcwd(), 'agriculture.db')
|
||
app.config['DATABASE'] = db_path
|
||
|
||
def get_db():
|
||
"""获取数据库连接"""
|
||
if 'db' not in g:
|
||
g.db = sqlite3.connect(
|
||
app.config['DATABASE'],
|
||
check_same_thread=False
|
||
)
|
||
g.db.row_factory = sqlite3.Row
|
||
return g.db
|
||
|
||
def init_db():
|
||
"""初始化数据库（创建表），只执行一次"""
|
||
with app.app_context():
|
||
db = get_db()
|
||
cursor = db.cursor()
|
||
|
||
# 创建元数据表（用于记录初始化状态）
|
||
cursor.execute("""
|
||
CREATE TABLE IF NOT EXISTS metadata (
|
||
key TEXT PRIMARY KEY,
|
||
value TEXT
|
||
)
|
||
""")
|
||
|
||
# 检查是否已初始化
|
||
cursor.execute("SELECT value FROM metadata WHERE key = 'initialized'")
|
||
initialized = cursor.fetchone()
|
||
|
||
if initialized and initialized[0] == '1':
|
||
print("✅ 数据库已初始化，跳过初始化过程")
|
||
return
|
||
|
||
# 获取 schema.sql 文件路径
|
||
schema_path = os.path.join(app.root_path, 'schema.sql')
|
||
|
||
# 以 UTF-8 编码读取文件内容
|
||
with open(schema_path, 'r', encoding='utf-8') as f:
|
||
sql_content = f.read()
|
||
|
||
# 移除注释并分割SQL语句
|
||
sql_content = re.sub(r'--.*$', '', sql_content, flags=re.MULTILINE)
|
||
sql_statements = re.split(r';\s*', sql_content)
|
||
|
||
# 执行每个SQL语句
|
||
for i, stmt in enumerate(sql_statements, 1):
|
||
stmt = stmt.strip()
|
||
if stmt: # 跳过空语句
|
||
try:
|
||
cursor.execute(stmt)
|
||
print(f"✅ 执行SQL语句 {i} 成功")
|
||
except sqlite3.Error as e:
|
||
db.rollback()
|
||
print(f"❌ 执行SQL语句 {i} 失败: {str(e)}")
|
||
print(f" 语句内容: {stmt}")
|
||
|
||
# 标记数据库已初始化
|
||
cursor.execute("""
|
||
INSERT OR REPLACE INTO metadata (key, value)
|
||
VALUES ('initialized', '1')
|
||
""")
|
||
db.commit()
|
||
print("✅ 数据库初始化完成")
|
||
|
||
# 在应用启动时自动初始化数据库
|
||
init_db()
|
||
|
||
@app.teardown_appcontext
|
||
def close_db(exception):
|
||
"""在每次请求后关闭数据库连接"""
|
||
db = g.pop('db', None)
|
||
if db is not None:
|
||
db.close()
|
||
|
||
# 提供给蓝图使用的 db 获取方式
|
||
app.get_db = get_db
|
||
|
||
# 启用 CORS（跨域支持）
|
||
CORS(app, resources={r"/*": {"origins": ["http://203.195.163.42:7792", "http://localhost:5000"]}}, supports_credentials=True)
|
||
|
||
# 注册蓝图
|
||
from blueprints.login import bp as login_bp
|
||
from blueprints.register import bp as register_bp
|
||
from blueprints.yzm import bp as yzm_bp
|
||
from blueprints.weather import bp as weather_bp
|
||
from blueprints.shebei import bp as shebei_bp
|
||
from blueprints.device import bp as device_bp
|
||
from blueprints.personnel import bp as personnel_bp
|
||
from blueprints.aiask import bp as aiask_bp
|
||
from blueprints.temperature import bp as temperature_bp
|
||
from blueprints.ph_data import bp as ph_data_bp
|
||
from blueprints.wendu import bp as wendu_bp
|
||
from blueprints.chohai import bp as chohai_bp
|
||
from blueprints.shi1 import bp as shi1_bp
|
||
from blueprints.shi2 import bp as shi2_bp
|
||
from blueprints.tem import tem_bp
|
||
from blueprints.device_warning import bp as device_warning_bp
|
||
from blueprints.chou1 import bp as chou1_bp
|
||
from blueprints.chou2 import bp as chou2_bp
|
||
from blueprints.chou3 import bp as chou3_bp
|
||
from blueprints.liebiao import bp as liebiao_bp
|
||
from blueprints.guan import bp as guan_bp
|
||
|
||
app.register_blueprint(login_bp)
|
||
app.register_blueprint(register_bp)
|
||
app.register_blueprint(yzm_bp)
|
||
app.register_blueprint(weather_bp)
|
||
app.register_blueprint(shebei_bp)
|
||
app.register_blueprint(device_bp)
|
||
app.register_blueprint(personnel_bp)
|
||
app.register_blueprint(aiask_bp, url_prefix='/aiask')
|
||
app.register_blueprint(temperature_bp)
|
||
app.register_blueprint(ph_data_bp, url_prefix='/ph_data')
|
||
app.register_blueprint(wendu_bp)
|
||
app.register_blueprint(chohai_bp)
|
||
app.register_blueprint(shi1_bp)
|
||
app.register_blueprint(shi2_bp)
|
||
app.register_blueprint(tem_bp, url_prefix='/api')
|
||
app.register_blueprint(device_warning_bp)
|
||
app.register_blueprint(chou1_bp)
|
||
app.register_blueprint(chou2_bp)
|
||
app.register_blueprint(chou3_bp)
|
||
app.register_blueprint(liebiao_bp)
|
||
app.register_blueprint(guan_bp)
|
||
|
||
with app.app_context():
|
||
try:
|
||
from blueprints.chou1 import load_prediction_model
|
||
load_prediction_model()
|
||
print("✅ 预测模型已加载")
|
||
except Exception as e:
|
||
print(f"❌ 加载预测模型失败: {str(e)}")
|
||
|
||
return app
|
||
|
||
|
||
if __name__ == '__main__':
|
||
app = create_app()
|
||
app.run(host='0.0.0.0', port=5000, debug=True) |