upgrade the datas in the migration

This commit is contained in:
Cayo Puigdefabregas 2022-05-31 18:33:41 +02:00
parent dbd5645c94
commit 2dd3b7c276
1 changed files with 30 additions and 1 deletions

View File

@ -5,6 +5,8 @@ Revises: 8571fb32c912
Create Date: 2022-05-27 11:07:18.245322 Create Date: 2022-05-27 11:07:18.245322
""" """
from uuid import uuid4
import citext import citext
import sqlalchemy as sa import sqlalchemy as sa
from alembic import context, op from alembic import context, op
@ -24,6 +26,31 @@ def get_inv():
return INV return INV
def upgrade_datas():
sql = f'select user_from_id, user_to_id, lot_id, code from {get_inv()}.trade where confirm=False'
con = op.get_bind()
sql_phantom = 'select id from common.user where phantom=True'
phantoms = [x[0] for x in con.execute(sql_phantom)]
for ac in con.execute(sql):
id = uuid4()
user_from = ac.user_from_id
user_to = ac.user_to_id
lot = ac.lot_id
code = ac.code
columns = '(id, user_from_id, user_to_id, lot_id, code)'
values = f'(\'{id}\', \'{user_from}\', \'{user_to}\', \'{lot}\', \'{code}\')'
if user_from not in phantoms:
columns = '(id, user_to_id, lot_id, code)'
values = f'(\'{id}\', \'{user_to}\', \'{lot}\', \'{code}\')'
if user_to not in phantoms:
columns = '(id, user_from_id, lot_id, code)'
values = f'(\'{id}\', \'{user_from}\', \'{lot}\', \'{code}\')'
new_transfer = f'insert into {get_inv()}.transfer {columns} values {values}'
op.execute(new_transfer)
def upgrade(): def upgrade():
# creating transfer table # creating transfer table
op.create_table( op.create_table(
@ -45,7 +72,7 @@ def upgrade():
sa.Column( sa.Column(
'description', 'description',
citext.CIText(), citext.CIText(),
nullable=False, nullable=True,
comment='A comment about the action.', comment='A comment about the action.',
), ),
sa.Column('date', sa.TIMESTAMP(timezone=True), nullable=True), sa.Column('date', sa.TIMESTAMP(timezone=True), nullable=True),
@ -83,6 +110,8 @@ def upgrade():
schema=f'{get_inv()}', schema=f'{get_inv()}',
) )
upgrade_datas()
def downgrade(): def downgrade():
op.drop_index( op.drop_index(