213 lines
6.6 KiB
Python
213 lines
6.6 KiB
Python
from fastapi import FastAPI, HTTPException
|
||
from pydantic import BaseModel, Field, validator
|
||
import sqlite3
|
||
import json,os
|
||
from datetime import datetime as dt
|
||
import uuid
|
||
from datetime import datetime
|
||
from typing import Optional, Dict, Any
|
||
|
||
app = FastAPI()
|
||
|
||
# 数据库文件路径
|
||
DATABASE = os.path.join('db', 'dtmgtDb.db') #"lms_data.db"
|
||
|
||
|
||
# 创建数据库表
|
||
def init_db():
|
||
db_path = os.path.join('db', 'dtmgtDb.db')
|
||
conn = sqlite3.connect(db_path)
|
||
c = conn.cursor()
|
||
# 创建新表结构，包含所有拆分字段
|
||
c.execute('''
|
||
CREATE TABLE IF NOT EXISTS LMIS_result (
|
||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
requestId TEXT UNIQUE NOT NULL,
|
||
module TEXT NOT NULL,
|
||
buckSn TEXT,
|
||
time TEXT NOT NULL,
|
||
deviceCode TEXT NOT NULL,
|
||
deviceName TEXT NOT NULL,
|
||
deviceStatus TEXT NOT NULL,
|
||
operatorId TEXT NOT NULL,
|
||
|
||
-- 从data字段拆分的具体字段 --
|
||
project TEXT,
|
||
phase TEXT,
|
||
channel TEXT,
|
||
startTime TEXT,
|
||
endTime TEXT,
|
||
orientation TEXT,
|
||
direction TEXT,
|
||
cycles INTEGER,
|
||
cyclesTotal INTEGER,
|
||
height REAL,
|
||
config TEXT,
|
||
|
||
-- 原始JSON数据备份 --
|
||
raw_data TEXT NOT NULL,
|
||
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||
)
|
||
''')
|
||
conn.commit()
|
||
conn.close()
|
||
|
||
|
||
# 初始化数据库
|
||
init_db()
|
||
|
||
|
||
# 请求模型 - 适配客户端格式
|
||
# 请求模型 - 适配客户端格式
|
||
class RequestData(BaseModel):
|
||
requestId: str = Field(..., min_length=12, max_length=40,
|
||
description="请求唯一编号uuid（去除横杠）")
|
||
module: str
|
||
buckSn: Optional[str] = None
|
||
time: str
|
||
deviceCode: str
|
||
deviceName: str
|
||
deviceStatus: str
|
||
operatorId: str
|
||
data: str # 客户端发送的是JSON字符串
|
||
|
||
@validator('time')
|
||
def validate_time_format(cls, v):
|
||
"""验证时间格式是否正确"""
|
||
try:
|
||
dt.strptime(v, "%Y-%m-%d %H:%M:%S")
|
||
return v
|
||
except ValueError:
|
||
raise ValueError("时间格式应为 YYYY-MM-DD HH:MM:SS")
|
||
|
||
|
||
@app.post("/gateway/api/rel/upload/commonUploadData")
|
||
async def submit_lmis_data(request_data: RequestData):
|
||
"""接收LMIS数据并写入SQLite数据库，拆分data字段"""
|
||
print(f"submit_lmis_data /gateway/api/rel/upload/commonUploadData")
|
||
# 检查requestId是否已存在（防重）
|
||
conn = sqlite3.connect(DATABASE)
|
||
c = conn.cursor()
|
||
c.execute("SELECT 1 FROM LMIS_result WHERE requestId = ?", (request_data.requestId,))
|
||
if c.fetchone():
|
||
conn.close()
|
||
# 返回客户端期望的1501状态码
|
||
return {
|
||
"code": 1501,
|
||
"message": "数据已存在，重复的requestId"
|
||
}
|
||
|
||
try:
|
||
# 解析data字段
|
||
try:
|
||
data_json = json.loads(request_data.data)
|
||
except json.JSONDecodeError as e:
|
||
error_detail = f"data字段不是有效的JSON: {str(e)}"
|
||
raise HTTPException(status_code=400, detail=error_detail)
|
||
|
||
# 提取各个字段（使用get方法避免KeyError）
|
||
project = data_json.get("project")
|
||
phase = data_json.get("phase")
|
||
channel = data_json.get("channel")
|
||
startTime = data_json.get("startTime")
|
||
endTime = data_json.get("endTime")
|
||
orientation = data_json.get("orientation")
|
||
direction = data_json.get("direction")
|
||
|
||
# 处理数值字段，确保类型正确
|
||
cycles = data_json.get("cycles")
|
||
if cycles is not None and not isinstance(cycles, int):
|
||
try:
|
||
cycles = int(cycles)
|
||
except (ValueError, TypeError):
|
||
cycles = None
|
||
|
||
cyclesTotal = data_json.get("cyclesTotal")
|
||
if cyclesTotal is not None and not isinstance(cyclesTotal, int):
|
||
try:
|
||
cyclesTotal = int(cyclesTotal)
|
||
except (ValueError, TypeError):
|
||
cyclesTotal = None
|
||
|
||
height = data_json.get("height")
|
||
if height is not None and not isinstance(height, float):
|
||
try:
|
||
height = float(height)
|
||
except (ValueError, TypeError):
|
||
height = None
|
||
|
||
config = data_json.get("config")
|
||
print(f"json={data_json}")
|
||
# 插入数据到数据库（包含所有拆分字段）
|
||
c.execute('''
|
||
INSERT INTO LMIS_result (
|
||
requestId, module, buckSn, time,
|
||
deviceCode, deviceName, deviceStatus, operatorId,
|
||
project, phase, channel, startTime, endTime,
|
||
orientation, direction, cycles, cyclesTotal, height, config,
|
||
raw_data
|
||
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||
''', (
|
||
request_data.requestId,
|
||
request_data.module,
|
||
request_data.buckSn,
|
||
request_data.time,
|
||
request_data.deviceCode,
|
||
request_data.deviceName,
|
||
request_data.deviceStatus,
|
||
request_data.operatorId,
|
||
project,
|
||
phase,
|
||
channel,
|
||
startTime,
|
||
endTime,
|
||
orientation,
|
||
direction,
|
||
cycles,
|
||
cyclesTotal,
|
||
height,
|
||
config,
|
||
request_data.data # 存储原始JSON作为备份
|
||
))
|
||
|
||
conn.commit()
|
||
row_id = c.lastrowid
|
||
|
||
# 返回客户端期望的成功响应格式
|
||
return {
|
||
"code": 200,
|
||
"message": "操作成功",
|
||
"data": {
|
||
"recordId": row_id,
|
||
"requestId": request_data.requestId
|
||
}
|
||
}
|
||
except HTTPException as e:
|
||
# 捕获已知异常
|
||
conn.rollback()
|
||
return {
|
||
"code": 400,
|
||
"message": e.detail
|
||
}
|
||
except Exception as e:
|
||
# 处理其他异常
|
||
conn.rollback()
|
||
error_detail = f"服务器内部错误: {str(e)}"
|
||
return {
|
||
"code": 500,
|
||
"message": error_detail
|
||
}
|
||
finally:
|
||
conn.close()
|
||
|
||
|
||
# 添加一个简单的根路由用于测试
|
||
@app.get("/gateway/api/rel/upload/getconnect")
|
||
async def get_connect():
|
||
return {"message": "LMIS 模拟接收数据服务已运行", "status": "ok"}
|
||
|
||
|
||
if __name__ == "__main__":
|
||
import uvicorn
|
||
|
||
uvicorn.run(app, host="0.0.0.0", port=8088) |