74 lines
3.1 KiB
Python
74 lines
3.1 KiB
Python
import argparse
|
|
from pathlib import Path
|
|
|
|
from project_bitable_common import (
|
|
DailySyncConfig,
|
|
build_daily_sync_context,
|
|
get_default_end_date,
|
|
get_default_start_date,
|
|
get_base_url,
|
|
get_feishu_chat_id,
|
|
get_feishu_report_image_path,
|
|
get_group_webhook_url,
|
|
get_mapping_file,
|
|
get_store_field_name,
|
|
iter_dates,
|
|
load_dotenv,
|
|
log_step,
|
|
run_daily_sync_for_date,
|
|
send_sync_notification,
|
|
validate_date,
|
|
)
|
|
|
|
|
|
def parse_args() -> argparse.Namespace:
|
|
parser = argparse.ArgumentParser(description="按 销售 -> 退款 -> GMV 顺序执行飞书多维表格同步。")
|
|
parser.add_argument("--start-date", default=get_default_start_date(), help="开始日期，格式 YYYY-MM-DD")
|
|
parser.add_argument("--end-date", default=get_default_end_date(), help="结束日期，格式 YYYY-MM-DD，默认昨天")
|
|
parser.add_argument("--base-url", default=get_base_url(), help="飞书多维表格地址")
|
|
parser.add_argument("--mapping-file", default=str(get_mapping_file()), help="店铺映射表 xlsx")
|
|
parser.add_argument("--store-field-name", default=get_store_field_name(), help="飞书中的店铺字段名")
|
|
parser.add_argument("--chat-id", default=get_feishu_chat_id(), help="飞书群 chat_id")
|
|
parser.add_argument("--report-image-path", default=get_feishu_report_image_path(), help="飞书卡片预览图本地路径")
|
|
parser.add_argument("--group-webhook", default=get_group_webhook_url(), help="飞书群机器人 webhook")
|
|
parser.add_argument("--table-map-file", default="", help="可选，项目到表名的映射 JSON 文件路径")
|
|
parser.add_argument("--dry-run", action="store_true", help="只输出将要更新的内容，不调用飞书写接口")
|
|
parser.add_argument("--skip-missing-tables", action="store_true", help="找不到对应月份的飞书表时跳过")
|
|
parser.add_argument("--no-notify", action="store_true", help="只写入多维表格，不推送飞书群通知")
|
|
return parser.parse_args()
|
|
|
|
|
|
def main() -> None:
|
|
load_dotenv()
|
|
args = parse_args()
|
|
start_date = validate_date(args.start_date)
|
|
end_date = validate_date(args.end_date)
|
|
if end_date < start_date:
|
|
raise ValueError("end-date 不能早于 start-date")
|
|
|
|
log_step("流水线开始: 按天执行 销售 -> 退款 -> GMV")
|
|
context = build_daily_sync_context(
|
|
DailySyncConfig(
|
|
base_url=args.base_url,
|
|
mapping_file=Path(args.mapping_file),
|
|
store_field_name=args.store_field_name,
|
|
table_map_file=args.table_map_file,
|
|
dry_run=args.dry_run,
|
|
skip_missing_tables=args.skip_missing_tables,
|
|
chat_id=args.chat_id,
|
|
report_image_path=args.report_image_path,
|
|
group_webhook=args.group_webhook,
|
|
)
|
|
)
|
|
for stat_date in iter_dates(start_date, end_date):
|
|
run_daily_sync_for_date(context, stat_date)
|
|
if args.no_notify:
|
|
log_step("已跳过飞书群通知")
|
|
else:
|
|
send_sync_notification(context, start_date, end_date)
|
|
log_step("流水线完成")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|