alembic migration upgrade (#7)

profcomff · Sep 24, 2022 · cf3eadc · cf3eadc
1 parent 3e17c02
commit cf3eadc
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/migrations/versions/6b75dd50398f_issue2.py b/migrations/versions/6b75dd50398f_issue2.py
@@ -1,7 +1,6 @@
 from alembic import op
 import sqlalchemy as sa
 
-
 revision = '6b75dd50398f'
 down_revision = 'f2d45b6daecf'
 branch_labels = None
@@ -11,8 +10,12 @@
 def upgrade():
     op.add_column('actions_info', sa.Column('additional_data', sa.String(), nullable=True))
     op.add_column('user', sa.Column('union_number', sa.String(), nullable=True))
-    op.add_column('user', sa.Column('modify_ts', sa.DateTime(), nullable=False))
-    op.add_column('user', sa.Column('create_ts', sa.DateTime(), nullable=False))
+    op.add_column('user', sa.Column('modify_ts', sa.DateTime(), nullable=True))
+    op.execute(f'UPDATE "user" SET modify_ts = current_timestamp')
+    op.alter_column('user', 'modify_ts', nullable=False)
+    op.add_column('user', sa.Column('create_ts', sa.DateTime(), nullable=True))
+    op.execute(f'UPDATE "user" SET create_ts = current_timestamp')
+    op.alter_column('user', 'create_ts', nullable=False)
 
 
 def downgrade():