"""companies: add exchange, country, market_cap columns

Revision ID: k2c3d4e5f6g7
Revises: j1b2c3d4e5f6
Create Date: 2026-04-20

"""
from alembic import op
import sqlalchemy as sa

revision = 'k2c3d4e5f6g7'
down_revision = 'j1b2c3d4e5f6'
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column('companies', sa.Column('exchange', sa.String(20), nullable=True))
    op.add_column('companies', sa.Column('country', sa.String(100), nullable=True))
    op.add_column('companies', sa.Column('market_cap', sa.Float(), nullable=True))


def downgrade() -> None:
    op.drop_column('companies', 'market_cap')
    op.drop_column('companies', 'country')
    op.drop_column('companies', 'exchange')