96 lines
3.2 KiB
Python
96 lines
3.2 KiB
Python
from DbKit.Db import Db
|
||
|
||
class DegreeModel:
|
||
def __init__(self, db: Db = None):
|
||
self.db = db or Db()
|
||
|
||
async def get_degree(self, station_names=None, start_time=None, end_time=None):
|
||
await self.db.init_db()
|
||
params = {}
|
||
if station_names:
|
||
params["station_names"] = station_names
|
||
if start_time:
|
||
params["start_time"] = start_time
|
||
if end_time:
|
||
params["end_time"] = end_time
|
||
|
||
# 打印 SQL 模板内容
|
||
sql_identifier = "Degree.getDegree"
|
||
sql_template = await self.db.sql_loader.get_sql(sql_identifier)
|
||
print(f"\n--- SQL Template ({sql_identifier}) ---")
|
||
print(sql_template)
|
||
|
||
# 打印填充参数后的 SQL
|
||
processed_sql = await self.db.get_sql(sql_identifier, params)
|
||
print(f"\n--- Processed SQL with Params ---")
|
||
print(processed_sql)
|
||
print("-" * 40)
|
||
|
||
return await self.db.find(sql_identifier, params)
|
||
|
||
async def getStationList(self):
|
||
await self.db.init_db()
|
||
sql_identifier = "Degree.getStationList"
|
||
return await self.db.find(sql_identifier)
|
||
|
||
async def get_trend_change(self, end_time=None):
|
||
"""查询近3个月的充电量变化"""
|
||
await self.db.init_db()
|
||
from datetime import datetime
|
||
from dateutil.relativedelta import relativedelta
|
||
|
||
if end_time:
|
||
end_dt = datetime.strptime(end_time, '%Y-%m-%d %H:%M:%S')
|
||
else:
|
||
end_dt = datetime.now()
|
||
|
||
# 计算过去三个月的月份字符串 (M, M-1, M-2)
|
||
m3_dt = end_dt
|
||
m2_dt = end_dt - relativedelta(months=1)
|
||
m1_dt = end_dt - relativedelta(months=2)
|
||
|
||
month3 = m3_dt.strftime('%Y-%m')
|
||
month2 = m2_dt.strftime('%Y-%m')
|
||
month1 = m1_dt.strftime('%Y-%m')
|
||
|
||
# 统计范围从 m1 月的第一天开始
|
||
start_time = m1_dt.replace(day=1, hour=0, minute=0, second=0).strftime('%Y-%m-%d %H:%M:%S')
|
||
end_time_str = end_dt.strftime('%Y-%m-%d %H:%M:%S')
|
||
|
||
params = {
|
||
"start_time": start_time,
|
||
"end_time": end_time_str,
|
||
"month1": month1,
|
||
"month2": month2,
|
||
"month3": month3
|
||
}
|
||
|
||
sql_identifier = "Degree.getTrendChange"
|
||
return await self.db.find(sql_identifier, params)
|
||
|
||
async def get_company_top10(self, start_time=None, end_time=None):
|
||
"""获取指定时间内，按企业充电量由大到小TOP 10"""
|
||
await self.db.init_db()
|
||
params = {}
|
||
if start_time:
|
||
params["start_time"] = start_time
|
||
if end_time:
|
||
params["end_time"] = end_time
|
||
|
||
sql_identifier = "Degree.getCompanyTop10"
|
||
|
||
# 打印调试信息
|
||
sql_template = await self.db.sql_loader.get_sql(sql_identifier)
|
||
print(f"\n--- SQL Template ({sql_identifier}) ---")
|
||
print(sql_template)
|
||
|
||
processed_sql = await self.db.get_sql(sql_identifier, params)
|
||
print(f"\n--- Processed SQL with Params ---")
|
||
print(processed_sql)
|
||
print("-" * 40)
|
||
|
||
return await self.db.find(sql_identifier, params)
|
||
|
||
async def close(self):
|
||
await self.db.shutdown()
|