chore: fix the migration

sudoblockio · Mar 22, 2024 · 1ed3e9e · 1ed3e9e
1 parent 842a0d3
commit 1ed3e9e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/...mbic/versions/2945fc0e4d84_v0_7_1_wage.py → ...mbic/versions/2945fc0e4d84_v0_8_1_wage.py b/...mbic/versions/2945fc0e4d84_v0_7_1_wage.py → ...mbic/versions/2945fc0e4d84_v0_8_1_wage.py
@@ -1,4 +1,4 @@
-"""v0.7.1-wage
+"""v0.8.1-wage
 
 Revision ID: 2945fc0e4d84
 Revises: 694432f3c689
@@ -19,11 +19,13 @@
 
 def upgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    pass
+    op.add_column('apy_time', sa.Column('total_wage', sa.Float(), nullable=True))
+    op.add_column('apy_time', sa.Column('active_preps', sa.Integer(), nullable=True))
     # ### end Alembic commands ###
 
 
 def downgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    pass
+    op.drop_column('apy_time', 'active_preps')
+    op.drop_column('apy_time', 'total_wage')
     # ### end Alembic commands ###
diff --git a/icon_governance/alembic/versions/694432f3c689_v0_8_0_commission_stats.py b/icon_governance/alembic/versions/694432f3c689_v0_8_0_commission_stats.py
@@ -46,15 +46,11 @@ def upgrade():
     sa.PrimaryKeyConstraint('timestamp')
     )
     op.create_index(op.f('ix_commission_time_height'), 'commission_time', ['height'], unique=False)
-    op.add_column('apy_time', sa.Column('total_wage', sa.Float(), nullable=False))
-    op.add_column('apy_time', sa.Column('active_preps', sa.Integer(), nullable=False))
     # ### end Alembic commands ###
 
 
 def downgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    op.drop_column('apy_time', 'active_preps')
-    op.drop_column('apy_time', 'total_wage')
     op.drop_index(op.f('ix_commission_time_height'), table_name='commission_time')
     op.drop_table('commission_time')
     # ### end Alembic commands ###