1

2026-08-10 16:29:26 +08:00
parent e364c46ce4
commit d1e8eb7316
18 changed files with 1335 additions and 819 deletions
@@ -0,0 +1,65 @@
+"""发票抬头表迁移
+
+创建 invoice_headers（发票抬头表），用于用户管理常用发票抬头。
+
+Revision ID: 20260811_20260811
+Revises: 20260810_20260810
+Create Date: 2026-08-11 00:00:00.000000
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+revision: str = '20260811_20260811'
+down_revision: Union[str, None] = '20260810_20260810'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def _comment_table(table_name: str, comment: str) -> None:
+    op.execute(f"COMMENT ON TABLE {table_name} IS '{comment}'")
+
+
+def _comment_column(table_name: str, column_name: str, comment: str) -> None:
+    escaped = comment.replace("'", "''")
+    op.execute(f"COMMENT ON COLUMN {table_name}.{column_name} IS '{escaped}'")
+
+
+def upgrade() -> None:
+    op.create_table(
+        'invoice_headers',
+        sa.Column('id', sa.String(32), primary_key=True),
+        sa.Column('user_id', sa.String(32), sa.ForeignKey('users.id', ondelete='CASCADE'), nullable=False),
+        sa.Column('type', sa.String(16), nullable=False),
+        sa.Column('name', sa.String(128), nullable=False),
+        sa.Column('tax_no', sa.String(32), nullable=True),
+        sa.Column('register_address', sa.String(256), nullable=True),
+        sa.Column('register_phone', sa.String(32), nullable=True),
+        sa.Column('bank_name', sa.String(128), nullable=True),
+        sa.Column('bank_account', sa.String(64), nullable=True),
+        sa.Column('email', sa.String(128), nullable=True),
+        sa.Column('is_default', sa.Boolean, nullable=False, server_default='false'),
+        sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
+        sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
+    )
+    op.create_index('idx_invoice_headers_user', 'invoice_headers', ['user_id'])
+
+    # 表注释和字段注释
+    _comment_table('invoice_headers', '发票抬头表')
+    _comment_column('invoice_headers', 'id', '主键')
+    _comment_column('invoice_headers', 'user_id', '用户ID')
+    _comment_column('invoice_headers', 'type', '抬头类型: personal/company')
+    _comment_column('invoice_headers', 'name', '抬头名称')
+    _comment_column('invoice_headers', 'tax_no', '税号')
+    _comment_column('invoice_headers', 'register_address', '注册地址')
+    _comment_column('invoice_headers', 'register_phone', '注册电话')
+    _comment_column('invoice_headers', 'bank_name', '开户行')
+    _comment_column('invoice_headers', 'bank_account', '银行账号')
+    _comment_column('invoice_headers', 'email', '接收邮箱')
+    _comment_column('invoice_headers', 'is_default', '是否默认')
+
+
+def downgrade() -> None:
+    op.drop_table('invoice_headers')