"""添加距离、LOID、型号等扩展字段

Revision ID: add_onu_extended_fields
Revises: 32c84173ef05
Create Date: 2026-04-02
"""
from alembic import op
import sqlalchemy as sa


# revision identifiers
revision = 'add_onu_extended_fields'
down_revision = '32c84173ef05'
branch_labels = None
depends_on = None


def upgrade() -> None:
    # ONUDevice 表新增字段
    op.add_column('onu_devices', sa.Column('loid', sa.String(50), nullable=True))
    op.add_column('onu_devices', sa.Column('model', sa.String(100), nullable=True))
    op.add_column('onu_devices', sa.Column('distance_m', sa.Integer, nullable=True))

    # DeviceStatusHistory 表新增字段
    op.add_column('device_status_history', sa.Column('distance_m', sa.Integer, nullable=True))


def downgrade() -> None:
    op.drop_column('device_status_history', 'distance_m')
    op.drop_column('onu_devices', 'distance_m')
    op.drop_column('onu_devices', 'model')
    op.drop_column('onu_devices', 'loid')