trustchain-oc1-orchestral/devicehub-teal
Archived
10
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

adding migration of datas in migrate file

Browse source
This commit is contained in:
Cayo Puigdefabregas 2020-11-04 20:22:31 +01:00
parent 8145e606db
commit dac0d1d32f
1 changed files with 25 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
ereuse_devicehub/migrations/versions/68a5c025ab8e_adding_owner_id_in_device.py
View file

@ -23,31 +23,43 @@ def get_inv():
raise ValueError("Inventory value is not specified")
return INV
def upgrade_data():
con = op.get_bind()
computers = con.execute(f"select id, owner_id from {get_inv()}.computer")
for c in computers:
id_dev = c.id
id_owner = c.owner_id
sql = f"update {get_inv()}.device set owner_id='{id_owner}' where id={id_dev};"
con.execute(sql)
values = f"{get_inv()}.component.id, {get_inv()}.computer.owner_id"
table = f"{get_inv()}.component"
joins = f"inner join {get_inv()}.computer"
on = f"on {table}.parent_id={get_inv()}.computer.id"
sql = f"select {values} from {table} {joins} {on}"
components = con.execute(sql)
for c in components:
id = c.id
id_owner = c.owner_id
sql = f"update {get_inv()}.device set owner_id='{id_owner}' where id={id};"
con.execute(sql)
def upgrade():
# We need get the actual computers with owner_id
# because when add a column in device this reset the values of the owner_id
# in the computer tables
con = op.get_bind()
# computers = con.execute(f"select id, owner_id from {get_inv()}.computer")
op.add_column('device', sa.Column('owner_id', postgresql.UUID(),
sa.ForeignKeyConstraint(['owner_id'], ['common.user.id'], ),
nullable=True), schema=f'{get_inv()}')
op.create_foreign_key("fk_device_owner_id_user_id",
"device", "user",
["owner_id"], ["id"],
ondelete="SET NULL",
source_schema=f'{get_inv()}', referent_schema='common')
sql = f"select {get_inv()}.component.id, {get_inv()}.computer.owner_id from \
{get_inv()}.component \
inner join {get_inv()}.computer on \
{get_inv()}.component.parent_id={get_inv()}.computer.id"
components = con.execute(sql)
for id_component, id_owner in components:
_sql = f"update {get_inv()}.component set owner_id={id_owner} where id={id_component}"
con.execute(_sql)
upgrade_data()
def downgrade():