126 lines
3.8 KiB
Python
126 lines
3.8 KiB
Python
from flask import Blueprint, jsonify, request, g, current_app
|
||
from werkzeug.exceptions import HTTPException
|
||
import sqlite3
|
||
import datetime
|
||
|
||
import logging
|
||
from collections import defaultdict
|
||
|
||
# 配置日志
|
||
logging.basicConfig(level=logging.DEBUG)
|
||
logger = logging.getLogger('temperature_api')
|
||
logger.setLevel(logging.DEBUG)
|
||
|
||
# 创建蓝图
|
||
tem_bp = Blueprint('tem', __name__, url_prefix='/api')
|
||
|
||
|
||
def get_db():
|
||
"""获取数据库连接"""
|
||
if 'db' not in g:
|
||
db_path = current_app.config.get('DATABASE', 'temperature.db')
|
||
logger.info(f"连接数据库: {db_path}")
|
||
try:
|
||
g.db = sqlite3.connect(
|
||
db_path,
|
||
check_same_thread=False,
|
||
detect_types=sqlite3.PARSE_DECLTYPES
|
||
)
|
||
g.db.row_factory = sqlite3.Row
|
||
logger.info("数据库连接成功")
|
||
except Exception as e:
|
||
logger.error(f"数据库连接失败: {str(e)}")
|
||
raise HTTPException(status_code=500, detail=f"数据库连接失败: {str(e)}")
|
||
return g.db
|
||
|
||
|
||
def close_db(e=None):
|
||
"""关闭数据库连接"""
|
||
db = g.pop('db', None)
|
||
if db is not None:
|
||
db.close()
|
||
logger.info("数据库连接已关闭")
|
||
|
||
|
||
@tem_bp.teardown_app_request
|
||
def teardown_request(exception):
|
||
"""请求结束时关闭数据库连接"""
|
||
close_db()
|
||
|
||
|
||
@tem_bp.route('/devices', methods=['GET'])
|
||
def get_devices():
|
||
"""获取所有设备列表（包含设备名称）"""
|
||
try:
|
||
logger.info("获取设备列表请求")
|
||
db = get_db()
|
||
|
||
# 查询设备表，获取id和device_name
|
||
cursor = db.execute("SELECT id, device_name FROM device")
|
||
devices = cursor.fetchall()
|
||
|
||
# 转换为前端需要的格式 {id: {device_name: name}}
|
||
device_dict = {str(device['id']): {'device_name': device['device_name']} for device in devices}
|
||
|
||
logger.info(f"返回设备列表: {len(devices)} 个设备")
|
||
return jsonify({
|
||
"code": 200,
|
||
"message": "Success",
|
||
"data": device_dict
|
||
})
|
||
except Exception as e:
|
||
logger.error(f"获取设备列表失败: {str(e)}", exc_info=True)
|
||
return jsonify({
|
||
"code": 500,
|
||
"message": f"服务器错误: {str(e)}"
|
||
}), 500
|
||
|
||
|
||
@tem_bp.route('/temperature/device/<int:device_id>', methods=['GET'])
|
||
def get_device_temperature(device_id):
|
||
"""获取指定设备在2025-05-25至2025-05-29的温度数据"""
|
||
try:
|
||
logger.info(f"获取设备 {device_id} 温度数据")
|
||
db = get_db()
|
||
|
||
# 固定日期范围为2025-05-25至2025-05-29
|
||
start_date = datetime.date(2025, 5, 25)
|
||
end_date = datetime.date(2025, 5, 29)
|
||
|
||
logger.info(f"查询日期范围: {start_date} 至 {end_date}")
|
||
|
||
# 执行SQL查询
|
||
cursor = db.execute('''
|
||
SELECT
|
||
date(timestamp) AS date,
|
||
ROUND(AVG(temperature), 1) AS avg_temp
|
||
FROM temperature_data
|
||
WHERE device_id = ? AND date(timestamp) BETWEEN ? AND ?
|
||
GROUP BY date(timestamp)
|
||
ORDER BY date(timestamp) ASC
|
||
''', (device_id, start_date, end_date))
|
||
|
||
rows = cursor.fetchall()
|
||
logger.info(f"查询到 {len(rows)} 条数据")
|
||
|
||
# 处理结果
|
||
temperature_data = [
|
||
{
|
||
"date": row['date'],
|
||
"avg_temp": float(row['avg_temp'])
|
||
}
|
||
for row in rows
|
||
]
|
||
|
||
return jsonify({
|
||
"code": 200,
|
||
"message": "Success",
|
||
"data": temperature_data
|
||
})
|
||
|
||
except Exception as e:
|
||
logger.error(f"获取温度数据失败: {str(e)}", exc_info=True)
|
||
return jsonify({
|
||
"code": 500,
|
||
"message": f"服务器错误: {str(e)}"
|
||
}), 500 |