import json
|
from flask import Flask, request, jsonify
|
from database import DatabaseConfig, init_database
|
from dao import ContentTypeDAO, ContentDAO
|
import logging
|
from datetime import datetime
|
from flask_cors import CORS
|
|
# 设置日志记录
|
logging.basicConfig(level=logging.INFO)
|
logger = logging.getLogger(__name__)
|
|
# 加载配置
|
with open("config.json") as f:
|
config = json.load(f)
|
|
app = Flask(__name__)
|
CORS(app)
|
|
# 初始化数据库
|
db_config = DatabaseConfig()
|
init_database() # 初始化数据库和表（如果不存在）
|
|
|
# 新的API端点：ContentType相关
|
@app.route("/api/v1/type/save", methods=["POST"])
|
def save_contenttype():
|
"""保存ContentType的API端点"""
|
connection = None
|
try:
|
# 从请求中获取JSON数据
|
data = request.get_json()
|
|
if not data:
|
return jsonify({"error": "未提供数据"}), 400
|
|
# 检查是否是单个对象还是列表
|
if isinstance(data, list):
|
# 处理多个记录 - 使用单个连接和事务
|
connection = db_config.get_connection()
|
if not connection:
|
return jsonify({"error": "数据库连接失败"}), 500
|
|
results = []
|
errors = []
|
try:
|
for i, item in enumerate(data):
|
# 为每个项目创建一个包含数据库连接的配置
|
class TempDbConfig:
|
def get_connection(self):
|
return connection
|
|
result, error = ContentTypeDAO.save_contenttype(
|
item, TempDbConfig()
|
)
|
if error:
|
errors.append({"index": i, "error": error, "data": item})
|
else:
|
results.append(result)
|
|
connection.commit() # 提交所有更改
|
|
response_data = {
|
"saved": results,
|
"errors": errors,
|
"total_saved": len(results),
|
}
|
if errors:
|
# 如果有错误，返回400状态码
|
status_code = (
|
400
|
if any(
|
err["error"] in ["代码已存在", "代码和名称是必需的"]
|
for err in errors
|
)
|
else 500
|
)
|
return jsonify(response_data), status_code
|
else:
|
# 如果没有错误，检查是否有更新的记录
|
has_updates = any("更新" in r["message"] for r in results)
|
return jsonify(response_data), 200 if has_updates else 201
|
except Exception as e:
|
if connection:
|
connection.rollback()
|
logger.error(f"批量保存ContentType时出错: {e}")
|
return jsonify({"error": "批量保存操作失败"}), 500
|
finally:
|
if connection:
|
connection.close()
|
else:
|
# 处理单个记录
|
result, error = ContentTypeDAO.save_contenttype(data, db_config)
|
|
if error:
|
if error == "代码已存在" or error == "代码和名称是必需的":
|
return jsonify({"error": error}), 400
|
else:
|
return jsonify({"error": error}), 500
|
|
return jsonify(result), 200 if "更新" in result["message"] else 201
|
|
except Exception as e:
|
if connection:
|
try:
|
connection.rollback()
|
connection.close()
|
except Exception as close_error:
|
logger.error(f"关闭连接时出错: {close_error}")
|
logger.error(f"save_contenttype中的意外错误: {e}")
|
return jsonify({"error": "无效的请求格式"}), 400
|
|
|
@app.route("/api/v1/type/get", methods=["GET"])
|
def get_contenttype():
|
"""从数据库检索ContentType的API端点"""
|
try:
|
# 获取查询参数
|
code = request.args.get("code", "").strip()
|
record_id = request.args.get("id")
|
limit = request.args.get("limit", type=int)
|
|
# 验证参数
|
if record_id and not record_id.isdigit():
|
return jsonify({"error": "无效的ID参数"}), 400
|
|
if limit and limit <= 0:
|
return jsonify({"error": "限制必须是正整数"}), 400
|
|
# 使用DAO进行数据库操作
|
result, error = ContentTypeDAO.get_contenttype(
|
contenttype=record_id, code=code, limit=limit, db_config=db_config
|
)
|
|
if error:
|
return jsonify({"error": error}), 500
|
|
result["timestamp"] = datetime.now().isoformat()
|
return jsonify(result), 200
|
|
except Exception as e:
|
logger.error(f"get_contenttype中的意外错误: {e}")
|
return jsonify({"error": "无效的请求参数"}), 400
|
|
|
# 新的API端点：Content相关
|
@app.route("/api/v1/content/save", methods=["POST"])
|
def save_content():
|
"""保存Content的API端点"""
|
connection = None
|
try:
|
# 从请求中获取JSON数据
|
data = request.get_json()
|
|
if not data:
|
return jsonify({"error": "未提供数据"}), 400
|
|
# 检查是否是单个对象还是列表
|
if isinstance(data, list):
|
# 处理多个记录 - 使用单个连接和事务
|
connection = db_config.get_connection()
|
if not connection:
|
return jsonify({"error": "数据库连接失败"}), 500
|
|
results = []
|
errors = []
|
try:
|
for i, item in enumerate(data):
|
# 为每个项目创建一个包含数据库连接的配置
|
class TempDbConfig:
|
def get_connection(self):
|
return connection
|
|
result, error = ContentDAO.save_content(item, TempDbConfig())
|
if error:
|
errors.append({"index": i, "error": error, "data": item})
|
else:
|
results.append(result)
|
|
connection.commit() # 提交所有更改
|
|
response_data = {
|
"saved": results,
|
"errors": errors,
|
"total_saved": len(results),
|
}
|
if errors:
|
# 如果有错误，返回400状态码
|
status_code = (
|
400
|
if any(
|
err["error"]
|
in ["类型、问题和答案是必需的", "指定的类型ID不存在"]
|
for err in errors
|
)
|
else 500
|
)
|
return jsonify(response_data), status_code
|
else:
|
return jsonify(response_data), 201
|
except Exception as e:
|
if connection:
|
connection.rollback()
|
logger.error(f"批量保存Content时出错: {e}")
|
return jsonify({"error": "批量保存操作失败"}), 500
|
finally:
|
if connection:
|
connection.close()
|
else:
|
# 处理单个记录
|
result, error = ContentDAO.save_content(data, db_config)
|
|
if error:
|
if error in ["类型、问题和答案是必需的", "指定的类型ID不存在"]:
|
return jsonify({"error": error}), 400
|
else:
|
return jsonify({"error": error}), 500
|
|
return jsonify(result), 201 if "更新" not in result["message"] else 200
|
|
except Exception as e:
|
if connection:
|
try:
|
connection.rollback()
|
connection.close()
|
except Exception as close_error:
|
logger.error(f"关闭连接时出错: {close_error}")
|
logger.error(f"save_content中的意外错误: {e}")
|
return jsonify({"error": "无效的请求格式"}), 400
|
|
|
@app.route("/api/v1/content/get", methods=["GET"])
|
def get_content():
|
"""从数据库检索Content的API端点"""
|
try:
|
# 获取查询参数
|
content_type = request.args.get("type", type=int)
|
record_id = request.args.get("id", type=int)
|
limit = request.args.get("limit", type=int)
|
|
# 验证参数
|
if record_id and record_id <= 0:
|
return jsonify({"error": "无效的ID参数"}), 400
|
|
if limit and limit <= 0:
|
return jsonify({"error": "限制必须是正整数"}), 400
|
|
# 使用DAO进行数据库操作
|
result, error = ContentDAO.get_content(
|
content_id=record_id,
|
content_type=content_type,
|
limit=limit,
|
db_config=db_config,
|
)
|
|
if error:
|
return jsonify({"error": error}), 500
|
|
result["timestamp"] = datetime.now().isoformat()
|
return jsonify(result), 200
|
|
except Exception as e:
|
logger.error(f"get_content中的意外错误: {e}")
|
return jsonify({"error": "无效的请求参数"}), 400
|
|
|
@app.route("/api/v1/content/delete", methods=["DELETE"])
|
def delete_content():
|
"""根据ID删除Content的API端点"""
|
try:
|
# 从请求中获取JSON数据
|
data = request.get_json()
|
if not data:
|
# 也可以从URL参数获取ID
|
content_id = request.args.get("id", type=int)
|
if content_id is None:
|
return jsonify({"error": "未提供数据或ID参数"}), 400
|
else:
|
# 从JSON数据中获取ID
|
content_id = data.get("id")
|
if content_id is None or not isinstance(content_id, int) or content_id <= 0:
|
return jsonify({"error": "ID是必需的，且必须是正整数"}), 400
|
|
# 使用DAO进行删除操作
|
result, error = ContentDAO.delete_content(content_id, db_config)
|
|
if error:
|
if error == "指定的ID不存在":
|
return jsonify({"error": error}), 404
|
else:
|
return jsonify({"error": error}), 500
|
|
return jsonify(result), 200
|
except Exception as e:
|
logger.error(f"delete_content中的意外错误: {e}")
|
return jsonify({"error": "无效的请求格式"}), 400
|
|
|
# 健康检查端点
|
@app.route("/health", methods=["GET"])
|
def health_check():
|
"""健康检查端点，验证服务是否正在运行"""
|
return (
|
jsonify(
|
{
|
"status": "OK",
|
"message": "服务正在运行",
|
"timestamp": datetime.now().isoformat(),
|
}
|
),
|
200,
|
)
|
|
|
if __name__ == "__main__":
|
# 获取服务器配置
|
server_config = config["server"]
|
logger.info(f"在 {server_config['host']}:{server_config['port']} 上启动服务器")
|
app.run(
|
host=server_config["host"],
|
port=server_config["port"],
|
debug=server_config["debug"],
|
)
|
