"""001_add_information_fields

添加资讯表新字段：包装、评级信息、分类、行情编号

Revision ID: 001
Revises: 
Create Date: 2026-04-12
"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = '001'
down_revision = None
branch_labels = None
depends_on = None


def upgrade() -> None:
    """添加information表的新字段"""
    
    # 检查字段是否存在，如果不存在则添加
    # packaging 包装类型
    if not _column_exists('information', 'packaging'):
        op.add_column('information', sa.Column('packaging', sa.String(50), nullable=True))
    
    # is_graded 是否评级
    if not _column_exists('information', 'is_graded'):
        op.add_column('information', sa.Column('is_graded', sa.Boolean(), server_default='false', nullable=True))
    
    # grading_company 评级公司
    if not _column_exists('information', 'grading_company'):
        op.add_column('information', sa.Column('grading_company', sa.String(100), nullable=True))
    
    # grading_score 评级分数
    if not _column_exists('information', 'grading_score'):
        op.add_column('information', sa.Column('grading_score', sa.String(50), nullable=True))
    
    # category 分类
    if not _column_exists('information', 'category'):
        op.add_column('information', sa.Column('category', sa.String(100), nullable=True))
    
    # deal_no 行情编号
    if not _column_exists('information', 'deal_no'):
        op.add_column('information', sa.Column('deal_no', sa.String(50), nullable=True))
        op.create_index('ix_information_deal_no', 'information', ['deal_no'])


def downgrade() -> None:
    """删除information表的新字段"""
    if _column_exists('information', 'deal_no'):
        op.drop_index('ix_information_deal_no', 'information')
        op.drop_column('information', 'deal_no')
    
    if _column_exists('information', 'category'):
        op.drop_column('information', 'category')
    
    if _column_exists('information', 'grading_score'):
        op.drop_column('information', 'grading_score')
    
    if _column_exists('information', 'grading_company'):
        op.drop_column('information', 'grading_company')
    
    if _column_exists('information', 'is_graded'):
        op.drop_column('information', 'is_graded')
    
    if _column_exists('information', 'packaging'):
        op.drop_column('information', 'packaging')


def _column_exists(table: str, column: str) -> bool:
    """检查列是否存在"""
    from sqlalchemy import inspect
    from app.core.database import engine
    
    inspector = inspect(engine)
    columns = [col['name'] for col in inspector.get_columns(table)]
    return column in columns