66 lines
2.8 KiB
Python
66 lines
2.8 KiB
Python
"""发票抬头表迁移
|
|
|
|
创建 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')
|