gmv-analytics/generate_project_gmv_csv.py

import argparse

from project_bitable_common import (
    get_default_end_date,
    get_default_start_date,
    get_base_url,
    get_store_field_name,
    load_dotenv,
    log_step,
    run_gmv_sync,
    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("--store-field-name", default=get_store_field_name(), help="飞书中的店铺字段名")
    parser.add_argument("--dry-run", action="store_true", help="只输出将要更新的内容，不调用飞书写接口")
    parser.add_argument("--skip-missing-tables", 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 同步")
    run_gmv_sync(
        start_date=start_date,
        end_date=end_date,
        base_url=args.base_url,
        store_field_name=args.store_field_name,
        dry_run=args.dry_run,
        skip_missing_tables=args.skip_missing_tables,
    )
    log_step("GMV 同步完成")


if __name__ == "__main__":
    main()