sqxp/backend/scripts/clear_biz_orders.py

#!/usr/bin/env python3
"""清空订单及相关表（开发调试用）。用法：在 backend 目录执行 python scripts/clear_biz_orders.py"""
from __future__ import annotations

import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parent.parent))

from sqlalchemy import text

from app.core.database import engine

TABLES = [
    "biz_order_status_log_his",
    "biz_order_status_log",
    "biz_order_item",
    "biz_order_his",
    "biz_order_adjust_his",
    "biz_order_adjust",
    "biz_wechat_pay",
    "biz_refund_his",
    "biz_refund",
    "biz_settlement_his",
    "biz_settlement",
    "biz_invoice",
    "biz_deliverer_review_his",
    "biz_deliverer_review",
    "biz_delivery_warning_his",
    "biz_delivery_warning",
    "biz_delivery_task_item",
    "biz_delivery_task_his",
    "biz_delivery_task",
    "biz_order",
]


def main() -> None:
    with engine.begin() as conn:
        conn.execute(text("SET FOREIGN_KEY_CHECKS=0"))
        for t in TABLES:
            conn.execute(text(f"TRUNCATE TABLE `{t}`"))
        conn.execute(text("SET FOREIGN_KEY_CHECKS=1"))
    print("已清空订单及相关表:", ", ".join(TABLES))


if __name__ == "__main__":
    main()