from alembic import op
import sqlalchemy as sa
revision = '6a126cc5b23c'
down_revision = 'add_stripe_payment_method'
branch_labels = None
depends_on = None

def upgrade() -> None:
    op.add_column('rooms', sa.Column('capacity', sa.Integer(), nullable=True))
    op.add_column('rooms', sa.Column('room_size', sa.String(length=50), nullable=True))
    op.add_column('rooms', sa.Column('view', sa.String(length=100), nullable=True))

def downgrade() -> None:
    op.drop_column('rooms', 'view')
    op.drop_column('rooms', 'room_size')
    op.drop_column('rooms', 'capacity')