huanghai/aiData
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

'commit'

Browse Source
This commit is contained in:
HuangHai
2026-01-15 10:47:59 +08:00
parent 9d9b95037f
commit 3e5ddc587c
9 changed files with 136 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

118
Apps/TeLaiDian/Service.py
View File

@@ -32,7 +32,8 @@ class TeLaiDianService:
def generate_id(self):
return str(uuid.uuid4())
def get_hash(self, s: str) -> str:
def get_hash(self, station_name: str, address: str = None) -> str:
s = f"{station_name}{address or ''}"
return hashlib.md5(s.encode('utf-8')).hexdigest()
async def init_db(self):
@@ -41,14 +42,15 @@ class TeLaiDianService:
async def close_db(self):
await self.db.close()
async def save_station_profile_only(self, station_name, address):
if not station_name or not address:
async def save_station_profile_only(self, station_name, address, distance=None):
if not station_name:
return False
station_hash = self.get_hash(station_name)
station_hash = self.get_hash(station_name, address)
now = datetime.now()
async with await self.db.get_session() as session:
# 1. 保存 Profile
profile_id = self.generate_id()
await self.station_profile_model.save(
session=session,
@@ -59,9 +61,25 @@ class TeLaiDianService:
address=address,
valid_start_time=now
)
# 2. 如果有距离信息，也保存一个状态记录，确保距离被记录下来
if distance:
status_id = self.generate_id()
await self.station_status_model.save(
session=session,
id=status_id,
station_hash=station_hash,
total_piles=None,
free_piles=None,
piles_detail_json=None,
current_price=None,
distance=distance,
valid_start_time=now
)
await session.commit()
logger.info(f"仅保存场站基础信息: {station_name}")
logger.info(f"仅保存场站基础信息: {station_name} | 地址: {address} | 距离: {distance}")
return True
async def save_station_data(self, station_name, address, prices, total_piles=None, free_piles=None, piles_detail=None, parking_info=None, distance=None):
@@ -69,30 +87,59 @@ class TeLaiDianService:
保存场站全量数据：名称、地址、24小时价格计划、当前状态
"""
now = datetime.now()
station_hash = hashlib.md5(f"{station_name}{address}".encode('utf-8')).hexdigest()
station_hash = self.get_hash(station_name, address)
# 预处理价格：生成 24 小时的价格映射
hourly_schedule = [None] * 24
# 如果只有一条价格且没有时段，或者是“全天价格统一”的情况
# 我们先检查是否有这种特殊情况
for p in prices:
try:
start_parts = p['start'].split(':')
end_parts = p['end'].split(':')
start_str = p.get('start') or '00:00'
end_str = p.get('end') or '00:00'
start_parts = start_str.split(':')
end_parts = end_str.split(':')
start_hour = int(start_parts[0])
end_hour = int(end_parts[0])
# 处理 00:00 作为结束时间的情况 (表示 24:00)
if end_hour == 0 and (int(end_parts[1]) == 0 if len(end_parts) > 1 else True):
if start_hour != 0:
# 处理 00:00 或 24:00 作为结束时间的情况
if end_hour == 0 or end_hour == 24:
# 如果开始是 0，说明是跨越全天或到午夜
if start_hour == 0:
end_hour = 24
else:
# 比如 23:00 - 00:00
end_hour = 24
# 特殊处理：如果 start == end 且只有一条，视为全天
if start_hour == end_hour and len(prices) == 1:
end_hour = start_hour + 24
# 提取各项价格
price_val = p.get('price')
if price_val is None or price_val == 0:
# 尝试从电费和服务费计算
try:
e = float(p.get('elec_price', 0) or 0)
s = float(p.get('service_price', 0) or 0)
if e > 0 or s > 0:
price_val = e + s
except:
pass
# 如果还是没有价格，且有 plus_price 或 market_price，尝试回填
if price_val is None or price_val == 0:
price_val = p.get('plus_price') or p.get('market_price') or 0.0
price_data = {
"price": float(p.get('price')) if p.get('price') is not None else 0.0,
"plus_price": float(p.get('plus_price')) if p.get('plus_price') is not None else None,
"market_price": float(p.get('market_price')) if p.get('market_price') is not None else None,
"elec_price": float(p.get('elec_price')) if p.get('elec_price') is not None else None,
"service_price": float(p.get('service_price')) if p.get('service_price') is not None else None
"price": float(price_val) if price_val is not None else 0.0,
"plus_price": float(p.get('plus_price')) if p.get('plus_price') is not None and p.get('plus_price') != 0 else None,
"market_price": float(p.get('market_price')) if p.get('market_price') is not None and p.get('market_price') != 0 else None,
"elec_price": float(p.get('elec_price')) if p.get('elec_price') is not None and p.get('elec_price') != 0 else None,
"service_price": float(p.get('service_price')) if p.get('service_price') is not None and p.get('service_price') != 0 else None
}
# 填充对应的小时槽位
@@ -102,11 +149,39 @@ class TeLaiDianService:
end_hour += 24
while curr < end_hour:
hourly_schedule[curr % 24] = price_data
hour_idx = curr % 24
# 为每个小时创建独立的对象，包含明确的时段信息
hourly_schedule[hour_idx] = {
"start": f"{hour_idx:02d}:00",
"end": f"{(hour_idx + 1):02d}:00",
**price_data
}
curr += 1
except Exception as e:
logger.error(f"解析价格时段失败: {p}, error: {e}")
# 最后兜底：如果有任何价格数据但 hourly_schedule 还有空位，用最近的一个填充
# 这通常处理 VLM 识别不全或者时段没对齐的情况
last_valid = None
# 先找第一个有效的
for item in hourly_schedule:
if item:
last_valid = item
break
if last_valid:
for i in range(24):
if hourly_schedule[i] is None:
# 复制最后一份有效价格，但更新为当前小时的时段
new_item = last_valid.copy()
new_item["start"] = f"{i:02d}:00"
new_item["end"] = f"{(i+1):02d}:00"
hourly_schedule[i] = new_item
else:
last_valid = hourly_schedule[i]
else:
logger.warning(f"场站 {station_name} 未能生成有效的 24 小时价格表")
async with await self.db.get_session() as session:
# 1. 保存 Profile
profile_id = self.generate_id()
@@ -151,5 +226,12 @@ class TeLaiDianService:
)
await session.commit()
logger.info(f"成功保存场站数据: {station_name} (桩数: {total_piles}, 空闲: {free_piles}, 停车费: {parking_info})")
logger.info(f"成功保存场站数据: {station_name} (桩数: {total_piles}, 空闲: {free_piles}, 距离: {distance}, 价格计划条数: {len([x for x in hourly_schedule if x])})")
# 记录 24 小时价格概览，方便调试
summary = []
for i in range(24):
p = hourly_schedule[i]
if p:
summary.append(f"{i:02d}h:{p.get('price')}")
logger.info(f"24小时价格概览: {' | '.join(summary)}")
return True