124 lines
3.9 KiB
Python
124 lines
3.9 KiB
Python
|
import logging
|
|||
|
|
from datetime import datetime
|
|||
|
|
from utils.sql_utils import SQLUtils
|
|||
|
|
|
|||
|
|
class ElectricityDAO:
|
|||
|
|
"""电力消耗数据访问对象"""
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
"""初始化数据访问对象"""
|
|||
|
|
self.db = SQLUtils('sqlite', database='db/jtDB.db')
|
|||
|
|
|
|||
|
|
def __del__(self):
|
|||
|
|
"""析构函数，确保数据库连接关闭"""
|
|||
|
|
if hasattr(self, 'db'):
|
|||
|
|
self.db.close()
|
|||
|
|
|
|||
|
|
def create_table_if_not_exists(self):
|
|||
|
|
"""创建电力消耗表（如果不存在）"""
|
|||
|
|
try:
|
|||
|
|
sql = """
|
|||
|
|
CREATE TABLE IF NOT EXISTS wsbz_electricity_consumption (
|
|||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|||
|
|
sync_time TIMESTAMP,
|
|||
|
|
electricity_number REAL
|
|||
|
|
)
|
|||
|
|
"""
|
|||
|
|
self.db.execute_query(sql)
|
|||
|
|
logging.info("电力消耗表检查/创建成功")
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
logging.error(f"创建电力消耗表失败: {str(e)}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def save_electricity_data(self, electricity_number):
|
|||
|
|
"""保存电力消耗数据
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
electricity_number: 电力消耗数值
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
bool: 保存是否成功
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
current_time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
|||
|
|
|
|||
|
|
sql = """
|
|||
|
|
INSERT INTO wsbz_electricity_consumption (
|
|||
|
|
sync_time, electricity_number
|
|||
|
|
) VALUES (?, ?)
|
|||
|
|
"""
|
|||
|
|
params = (current_time, electricity_number)
|
|||
|
|
|
|||
|
|
self.db.execute_update(sql, params)
|
|||
|
|
logging.info(f"保存电力消耗数据成功: {electricity_number}")
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
logging.error(f"保存电力消耗数据失败: {str(e)}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def get_latest_electricity_data(self):
|
|||
|
|
"""获取最新的电力消耗数据
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
dict: 最新的电力消耗数据，未找到则返回None
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
sql = """
|
|||
|
|
SELECT id, sync_time, electricity_number
|
|||
|
|
FROM wsbz_electricity_consumption
|
|||
|
|
ORDER BY id DESC
|
|||
|
|
LIMIT 1
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
self.db.cursor.execute(sql)
|
|||
|
|
row = self.db.cursor.fetchone()
|
|||
|
|
|
|||
|
|
if row:
|
|||
|
|
data = {
|
|||
|
|
'id': row[0],
|
|||
|
|
'sync_time': row[1],
|
|||
|
|
'electricity_number': row[2]
|
|||
|
|
}
|
|||
|
|
return data
|
|||
|
|
else:
|
|||
|
|
return None
|
|||
|
|
except Exception as e:
|
|||
|
|
logging.error(f"获取最新电力消耗数据失败: {str(e)}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def get_electricity_data_by_date_range(self, start_date, end_date):
|
|||
|
|
"""根据日期范围获取电力消耗数据
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
start_date: 开始日期（YYYY-MM-DD格式）
|
|||
|
|
end_date: 结束日期（YYYY-MM-DD格式）
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
list: 电力消耗数据列表
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
sql = """
|
|||
|
|
SELECT id, sync_time, electricity_number
|
|||
|
|
FROM wsbz_electricity_consumption
|
|||
|
|
WHERE sync_time BETWEEN ? AND ?
|
|||
|
|
ORDER BY sync_time
|
|||
|
|
"""
|
|||
|
|
params = (f"{start_date} 00:00:00", f"{end_date} 23:59:59")
|
|||
|
|
|
|||
|
|
self.db.cursor.execute(sql, params)
|
|||
|
|
results = self.db.cursor.fetchall()
|
|||
|
|
|
|||
|
|
data_list = []
|
|||
|
|
for row in results:
|
|||
|
|
data = {
|
|||
|
|
'id': row[0],
|
|||
|
|
'sync_time': row[1],
|
|||
|
|
'electricity_number': row[2]
|
|||
|
|
}
|
|||
|
|
data_list.append(data)
|
|||
|
|
|
|||
|
|
return data_list
|
|||
|
|
except Exception as e:
|
|||
|
|
logging.error(f"获取电力消耗数据失败: {str(e)}")
|
|||
|
|
return []
|