1  from alembic import op
 2  import sqlalchemy as sa
 3  
 4  revision = '022'
 5  down_revision = '021'
 6  branch_labels = None
 7  depends_on = None
 8  
 9  def upgrade():
10      try:
11          op.add_column('users', sa.Column('is_restricted', sa.Boolean(), server_default='0', nullable=True))
12      except Exception as e:
13          print(f"Error adding is_restricted: {e}")
14  
15  def downgrade():
16      try:
17          op.drop_column('users', 'is_restricted')
18      except Exception as e:
19          print(f"Error dropping is_restricted: {e}")