1  from alembic import op
 2  import sqlalchemy as sa
 3  
 4  # revision identifiers, used by Alembic.
 5  revision = '006'
 6  down_revision = '005'
 7  branch_labels = None
 8  depends_on = None
 9  
10  def upgrade():
11      try:
12          op.add_column('teams', sa.Column('budget', sa.Float(), default=-1.0))
13          op.execute("UPDATE teams SET budget = -1.0 WHERE budget IS NULL")
14      except Exception as e:
15          print(f"Error adding budget column: {e}")
16  
17  def downgrade():
18      try:
19          op.drop_column('teams', 'budget')
20      except Exception as e:
21          print(f"Error dropping budget column: {e}")