import re

with open('app/core/models/order.py', 'r') as f:
    content = f.read()

# Add date import
content = content.replace('from datetime import datetime', 'from datetime import date, datetime')

# Add logistics fields before paid_at
old = """    note: Mapped[str | None] = mapped_column(Text, nullable=True, comment="买家备注")
    extra_attributes: Mapped[dict | None] = mapped_column(JSON, nullable=True, comment="ERP 编号、发票信息等行业字段")

    paid_at"""

new = """    note: Mapped[str | None] = mapped_column(Text, nullable=True, comment="买家备注")
    extra_attributes: Mapped[dict | None] = mapped_column(JSON, nullable=True, comment="ERP 编号、发票信息等行业字段")

    # 物流字段
    carrier: Mapped[str | None] = mapped_column(String(50), nullable=True, comment="快递公司名称")
    tracking_no: Mapped[str | None] = mapped_column(String(100), nullable=True, comment="运单号")
    estimated_delivery: Mapped[date | None] = mapped_column(Date, nullable=True, comment="预计送达日期")

    paid_at"""

content = content.replace(old, new)

with open('app/core/models/order.py', 'w') as f:
    f.write(content)

print("done")