'commit'
This commit is contained in:
@@ -2,7 +2,7 @@
|
||||
|
||||
# 采集配置
|
||||
SCROLL_DISTANCE_RATIO = 0.3
|
||||
MAX_STATIONS_COUNT = 30
|
||||
MAX_STATIONS_COUNT = 20
|
||||
FIRST_RUN_ONLY_ONE_STATION = False
|
||||
REDIS_STATION_EXPIRE = 120
|
||||
|
||||
|
||||
@@ -1,13 +1,13 @@
|
||||
import asyncio
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
import re
|
||||
import sys
|
||||
import zipfile
|
||||
from datetime import datetime
|
||||
from decimal import Decimal
|
||||
from typing import Any, Dict, List, Optional, Set
|
||||
import asyncio
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
import re
|
||||
import sys
|
||||
import zipfile
|
||||
from datetime import datetime
|
||||
from decimal import Decimal
|
||||
from typing import Any, Dict, List, Optional, Set
|
||||
|
||||
logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
|
||||
logger = logging.getLogger(__name__)
|
||||
@@ -516,7 +516,7 @@ async def main():
|
||||
limit = None
|
||||
|
||||
operators = ["新电途", "特来电", "驿来特", "艾特吉易充"]
|
||||
excel_paths: List[str] = []
|
||||
excel_paths: List[str] = []
|
||||
|
||||
ts = datetime.now().strftime("%Y%m%d_%H%M%S")
|
||||
export_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "Excel", ts)
|
||||
@@ -530,28 +530,16 @@ async def main():
|
||||
output_filename = f"{operator}_{ts}.xlsx"
|
||||
output_path = os.path.join(export_dir, output_filename)
|
||||
|
||||
logger.info(f"--- 正在导出 {operator} ---")
|
||||
logger.info(f"--- 正在导出 {operator} ---")
|
||||
if limit is not None:
|
||||
logger.info(f"限制导出记录数: {limit}")
|
||||
|
||||
try:
|
||||
final_path = await export_excel(operator, output_path, limit=limit)
|
||||
logger.info(f"{operator} 导出完成: {final_path}")
|
||||
excel_paths.append(final_path)
|
||||
excel_paths.append(final_path)
|
||||
except Exception as e:
|
||||
logger.error(f"{operator} 导出失败: {e}")
|
||||
|
||||
if excel_paths:
|
||||
zip_filename = f"充电场站数据_{ts}.zip"
|
||||
zip_path = os.path.join(export_dir, zip_filename)
|
||||
try:
|
||||
with zipfile.ZipFile(zip_path, "w", compression=zipfile.ZIP_DEFLATED) as zf:
|
||||
for path in excel_paths:
|
||||
if os.path.isfile(path):
|
||||
zf.write(path, arcname=os.path.basename(path))
|
||||
logger.info(f"ZIP 打包完成: {zip_path}")
|
||||
except Exception as e:
|
||||
logger.error(f"ZIP 打包失败: {e}")
|
||||
|
||||
logger.info("所有导出任务已结束。")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user