routes.py 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454
  1. import sqlite3
  2. from flask import Blueprint, request, jsonify, current_app
  3. from .model import predict, train_and_save_model
  4. import pandas as pd
  5. from . import db # 从 app 包导入 db 实例
  6. from sqlalchemy.engine.reflection import Inspector
  7. from .database_models import Model, ModelParameters, Dataset
  8. import os
  9. from .utils import create_dynamic_table, allowed_file
  10. from sqlalchemy.orm import sessionmaker
  11. # 创建蓝图 (Blueprint),用于分离路由
  12. bp = Blueprint('routes', __name__)
  13. DATABASE = 'SoilAcidification.db'
  14. @bp.route('/upload-dataset', methods=['POST'])
  15. def upload_dataset():
  16. try:
  17. # 检查是否包含文件
  18. if 'file' not in request.files:
  19. return jsonify({'error': 'No file part'}), 400
  20. file = request.files['file']
  21. # 如果没有文件或者文件名为空
  22. if file.filename == '':
  23. return jsonify({'error': 'No selected file'}), 400
  24. # 检查文件类型是否允许
  25. if file and allowed_file(file.filename):
  26. # 获取数据集的元数据
  27. dataset_name = request.form.get('dataset_name')
  28. dataset_description = request.form.get('dataset_description', 'No description provided')
  29. dataset_type = request.form.get('dataset_type') # 新增字段:数据集类型
  30. # 校验 dataset_type 是否存在
  31. if not dataset_type:
  32. return jsonify({'error': 'Dataset type is required'}), 400
  33. # 创建 Dataset 实体并保存到数据库
  34. new_dataset = Dataset(
  35. DatasetName=dataset_name,
  36. DatasetDescription=dataset_description,
  37. RowCount=0, # 初步创建数据集时,行数先置为0
  38. Status='pending', # 状态默认为 'pending'
  39. DatasetType=dataset_type # 保存数据集类型
  40. )
  41. db.session.add(new_dataset)
  42. db.session.commit()
  43. # 获取数据集的 ID
  44. dataset_id = new_dataset.DatasetID
  45. # 保存文件时使用数据库的 DatasetID 作为文件名
  46. unique_filename = f"dataset_{dataset_id}.xlsx"
  47. upload_folder = current_app.config['UPLOAD_FOLDER']
  48. file_path = os.path.join(upload_folder, unique_filename)
  49. # 保存文件
  50. file.save(file_path)
  51. # 读取 Excel 文件内容
  52. dataset_df = pd.read_excel(file_path)
  53. # 更新数据集的行数
  54. row_count = len(dataset_df)
  55. new_dataset.RowCount = row_count
  56. new_dataset.Status = 'processed' # 状态更新为 processed
  57. db.session.commit()
  58. # 动态创建数据表
  59. columns = {}
  60. for col in dataset_df.columns:
  61. if dataset_df[col].dtype == 'int64':
  62. columns[col] = 'int'
  63. elif dataset_df[col].dtype == 'float64':
  64. columns[col] = 'float'
  65. else:
  66. columns[col] = 'str'
  67. # 创建新表格(动态表格)
  68. dynamic_table_class = create_dynamic_table(dataset_id, columns)
  69. # 创建新的数据库会话
  70. Session = sessionmaker(bind=db.engine)
  71. session = Session()
  72. # 将每一行数据插入到动态创建的表格中
  73. for _, row in dataset_df.iterrows():
  74. record_data = row.to_dict()
  75. # 将数据插入到新表格中
  76. session.execute(dynamic_table_class.__table__.insert(), [record_data])
  77. session.commit()
  78. session.close()
  79. return jsonify({
  80. 'message': f'Dataset {dataset_name} uploaded successfully!',
  81. 'dataset_id': new_dataset.DatasetID,
  82. 'filename': unique_filename
  83. }), 201
  84. else:
  85. return jsonify({'error': 'Invalid file type'}), 400
  86. except Exception as e:
  87. return jsonify({'error': str(e)}), 500
  88. @bp.route('/tables', methods=['GET'])
  89. def list_tables():
  90. engine = db.engine # 使用 db 实例的 engine
  91. inspector = Inspector.from_engine(engine) # 创建 Inspector 对象
  92. table_names = inspector.get_table_names() # 获取所有表名
  93. return jsonify(table_names) # 以 JSON 形式返回表名列表
  94. @bp.route('/models/<int:model_id>', methods=['GET'])
  95. def get_model(model_id):
  96. try:
  97. model = Model.query.filter_by(ModelID=model_id).first()
  98. if model:
  99. return jsonify({
  100. 'ModelID': model.ModelID,
  101. 'ModelName': model.ModelName,
  102. 'ModelType': model.ModelType,
  103. 'CreatedAt': model.CreatedAt.strftime('%Y-%m-%d %H:%M:%S'),
  104. 'Description': model.Description
  105. })
  106. else:
  107. return jsonify({'message': 'Model not found'}), 404
  108. except Exception as e:
  109. return jsonify({'error': 'Internal server error', 'message': str(e)}), 500
  110. @bp.route('/models', methods=['GET'])
  111. def get_all_models():
  112. try:
  113. models = Model.query.all() # 获取所有模型数据
  114. if models:
  115. result = [
  116. {
  117. 'ModelID': model.ModelID,
  118. 'ModelName': model.ModelName,
  119. 'ModelType': model.ModelType,
  120. 'CreatedAt': model.CreatedAt.strftime('%Y-%m-%d %H:%M:%S'),
  121. 'Description': model.Description
  122. }
  123. for model in models
  124. ]
  125. return jsonify(result)
  126. else:
  127. return jsonify({'message': 'No models found'}), 404
  128. except Exception as e:
  129. return jsonify({'error': 'Internal server error', 'message': str(e)}), 500
  130. @bp.route('/model-parameters', methods=['GET'])
  131. def get_all_model_parameters():
  132. try:
  133. parameters = ModelParameters.query.all() # 获取所有参数数据
  134. if parameters:
  135. result = [
  136. {
  137. 'ParamID': param.ParamID,
  138. 'ModelID': param.ModelID,
  139. 'ParamName': param.ParamName,
  140. 'ParamValue': param.ParamValue
  141. }
  142. for param in parameters
  143. ]
  144. return jsonify(result)
  145. else:
  146. return jsonify({'message': 'No parameters found'}), 404
  147. except Exception as e:
  148. return jsonify({'error': 'Internal server error', 'message': str(e)}), 500
  149. @bp.route('/models/<int:model_id>/parameters', methods=['GET'])
  150. def get_model_parameters(model_id):
  151. try:
  152. model = Model.query.filter_by(ModelID=model_id).first()
  153. if model:
  154. # 获取该模型的所有参数
  155. parameters = [
  156. {
  157. 'ParamID': param.ParamID,
  158. 'ParamName': param.ParamName,
  159. 'ParamValue': param.ParamValue
  160. }
  161. for param in model.parameters
  162. ]
  163. # 返回模型参数信息
  164. return jsonify({
  165. 'ModelID': model.ModelID,
  166. 'ModelName': model.ModelName,
  167. 'ModelType': model.ModelType,
  168. 'CreatedAt': model.CreatedAt.strftime('%Y-%m-%d %H:%M:%S'),
  169. 'Description': model.Description,
  170. 'Parameters': parameters
  171. })
  172. else:
  173. return jsonify({'message': 'Model not found'}), 404
  174. except Exception as e:
  175. return jsonify({'error': 'Internal server error', 'message': str(e)}), 500
  176. @bp.route('/predict', methods=['POST'])
  177. def predict_route():
  178. try:
  179. data = request.get_json()
  180. model_name = data.get('model_name') # 提取模型名称
  181. parameters = data.get('parameters', {}) # 提取所有参数
  182. input_data = pd.DataFrame([parameters]) # 转换参数为DataFrame
  183. predictions = predict(input_data, model_name) # 调用预测函数
  184. return jsonify({'predictions': predictions}), 200
  185. except Exception as e:
  186. return jsonify({'error': str(e)}), 400
  187. # 定义添加数据库记录的 API 接口
  188. @bp.route('/add_item', methods=['POST'])
  189. def add_item():
  190. """
  191. 接收 JSON 格式的请求体,包含表名和要插入的数据。
  192. 尝试将数据插入到指定的表中。
  193. :return:
  194. """
  195. db = get_db()
  196. try:
  197. # 确保请求体是JSON格式
  198. data = request.get_json()
  199. if not data:
  200. raise ValueError("No JSON data provided")
  201. table_name = data.get('table')
  202. item_data = data.get('item')
  203. if not table_name or not item_data:
  204. return jsonify({'error': 'Missing table name or item data'}), 400
  205. cur = db.cursor()
  206. # 动态构建 SQL 语句
  207. columns = ', '.join(item_data.keys())
  208. placeholders = ', '.join(['?'] * len(item_data))
  209. sql = f"INSERT INTO {table_name} ({columns}) VALUES ({placeholders})"
  210. cur.execute(sql, tuple(item_data.values()))
  211. db.commit()
  212. # 返回更详细的成功响应
  213. return jsonify({'success': True, 'message': 'Item added successfully'}), 201
  214. except ValueError as e:
  215. return jsonify({'error': str(e)}), 400
  216. except KeyError as e:
  217. return jsonify({'error': f'Missing data field: {e}'}), 400
  218. except sqlite3.IntegrityError as e:
  219. # 处理例如唯一性约束违反等数据库完整性错误
  220. return jsonify({'error': 'Database integrity error', 'details': str(e)}), 409
  221. except sqlite3.Error as e:
  222. # 处理其他数据库错误
  223. return jsonify({'error': 'Database error', 'details': str(e)}), 500
  224. finally:
  225. db.close()
  226. # 定义删除数据库记录的 API 接口
  227. @bp.route('/delete_item', methods=['POST'])
  228. def delete_item():
  229. data = request.get_json()
  230. table_name = data.get('table')
  231. condition = data.get('condition')
  232. # 检查表名和条件是否提供
  233. if not table_name or not condition:
  234. return jsonify({
  235. "success": False,
  236. "message": "缺少表名或条件参数"
  237. }), 400
  238. # 尝试从条件字符串中分离键和值
  239. try:
  240. key, value = condition.split('=')
  241. except ValueError:
  242. return jsonify({
  243. "success": False,
  244. "message": "条件格式错误,应为 'key=value'"
  245. }), 400
  246. db = get_db()
  247. cur = db.cursor()
  248. try:
  249. # 执行删除操作
  250. cur.execute(f"DELETE FROM {table_name} WHERE {key} = ?", (value,))
  251. db.commit()
  252. # 如果没有错误发生,返回成功响应
  253. return jsonify({
  254. "success": True,
  255. "message": "记录删除成功"
  256. }), 200
  257. except sqlite3.Error as e:
  258. # 发生错误,回滚事务
  259. db.rollback()
  260. # 返回失败响应,并包含错误信息
  261. return jsonify({
  262. "success": False,
  263. "message": f"删除失败: {e}"
  264. }), 400
  265. # 定义修改数据库记录的 API 接口
  266. @bp.route('/update_item', methods=['PUT'])
  267. def update_record():
  268. data = request.get_json()
  269. # 检查必要的数据是否提供
  270. if not data or 'table' not in data or 'item' not in data:
  271. return jsonify({
  272. "success": False,
  273. "message": "请求数据不完整"
  274. }), 400
  275. table_name = data['table']
  276. item = data['item']
  277. # 假设 item 的第一个元素是 ID
  278. if not item or next(iter(item.keys())) is None:
  279. return jsonify({
  280. "success": False,
  281. "message": "记录数据为空"
  282. }), 400
  283. # 获取 ID 和其他字段值
  284. id_key = next(iter(item.keys()))
  285. record_id = item[id_key]
  286. updates = {key: value for key, value in item.items() if key != id_key} # 排除 ID
  287. db = get_db()
  288. cur = db.cursor()
  289. try:
  290. record_id = int(record_id) # 确保 ID 是整数
  291. except ValueError:
  292. return jsonify({
  293. "success": False,
  294. "message": "ID 必须是整数"
  295. }), 400
  296. # 准备参数列表,包括更新的值和 ID
  297. parameters = list(updates.values()) + [record_id]
  298. # 执行更新操作
  299. set_clause = ','.join([f"{k} = ?" for k in updates.keys()])
  300. sql = f"UPDATE {table_name} SET {set_clause} WHERE {id_key} = ?"
  301. try:
  302. cur.execute(sql, parameters)
  303. db.commit()
  304. if cur.rowcount == 0:
  305. return jsonify({
  306. "success": False,
  307. "message": "未找到要更新的记录"
  308. }), 404
  309. return jsonify({
  310. "success": True,
  311. "message": "数据更新成功"
  312. }), 200
  313. except sqlite3.Error as e:
  314. db.rollback()
  315. return jsonify({
  316. "success": False,
  317. "message": f"更新失败: {e}"
  318. }), 400
  319. # 定义查询数据库记录的 API 接口
  320. @bp.route('/search/record', methods=['GET'])
  321. def sql_search():
  322. """
  323. 接收 JSON 格式的请求体,包含表名和要查询的 ID。
  324. 尝试查询指定 ID 的记录并返回结果。
  325. :return:
  326. """
  327. try:
  328. data = request.get_json()
  329. # 表名
  330. sql_table = data['table']
  331. # 要搜索的 ID
  332. Id = data['id']
  333. # 连接到数据库
  334. db = get_db()
  335. cur = db.cursor()
  336. # 构造查询语句
  337. sql = f"SELECT * FROM {sql_table} WHERE id = ?"
  338. # 执行查询
  339. cur.execute(sql, (Id,))
  340. # 获取查询结果
  341. rows = cur.fetchall()
  342. column_names = [desc[0] for desc in cur.description]
  343. # 检查是否有结果
  344. if not rows:
  345. return jsonify({'error': '未查找到对应数据。'}), 400
  346. # 构造响应数据
  347. results = []
  348. for row in rows:
  349. result = {column_names[i]: row[i] for i in range(len(row))}
  350. results.append(result)
  351. # 关闭游标和数据库连接
  352. cur.close()
  353. db.close()
  354. # 返回 JSON 响应
  355. return jsonify(results), 200
  356. except sqlite3.Error as e:
  357. # 如果发生数据库错误,返回错误信息
  358. return jsonify({'error': str(e)}), 400
  359. except KeyError as e:
  360. # 如果请求数据中缺少必要的键,返回错误信息
  361. return jsonify({'error': f'缺少必要的数据字段: {e}'}), 400
  362. # 定义提供数据库列表,用于展示表格的 API 接口
  363. @bp.route('/tables', methods=['POST'])
  364. def get_table():
  365. data = request.get_json()
  366. table_name = data.get('table')
  367. if not table_name:
  368. return jsonify({'error': '需要表名'}), 400
  369. db = get_db()
  370. try:
  371. cur = db.cursor()
  372. cur.execute(f"SELECT * FROM {table_name}")
  373. rows = cur.fetchall()
  374. if not rows:
  375. return jsonify({'error': '表为空或不存在'}), 400
  376. headers = [description[0] for description in cur.description]
  377. return jsonify(rows=rows, headers=headers), 200
  378. except sqlite3.Error as e:
  379. return jsonify({'error': str(e)}), 400
  380. finally:
  381. db.close()