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

fix 2b90b41a556a_create_placeholders by null enviroment

This commit is contained in:
Cayo Puigdefabregas 2022-07-19 19:02:21 +02:00
parent 1167b3cb1e
commit e2870f2b3e
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
ereuse_devicehub/migrations/versions/2b90b41a556a_create_placeholders.py
View File

@ -7,6 +7,8 @@ Create Date: 2022-07-19 12:17:16.690865
""" """
import copy import copy
from alembic import context, op
from ereuse_devicehub.config import DevicehubConfig from ereuse_devicehub.config import DevicehubConfig
from ereuse_devicehub.db import db from ereuse_devicehub.db import db
from ereuse_devicehub.devicehub import Devicehub from ereuse_devicehub.devicehub import Devicehub
@ -36,6 +38,13 @@ branch_labels = None
depends_on = None depends_on = None
def get_inv():
INV = context.get_x_argument(as_dictionary=True).get('inventory')
if not INV:
raise ValueError("Inventory value is not specified")
return INV
def init_app(): def init_app():
app = Devicehub(inventory=DevicehubConfig.DB_SCHEMA) app = Devicehub(inventory=DevicehubConfig.DB_SCHEMA)
app.app_context().push() app.app_context().push()
@ -207,6 +216,11 @@ def remove_placeholders():
def upgrade(): def upgrade():
con = op.get_bind()
devices = con.execute(f'select * from {get_inv()}.device')
if not list(devices):
return
init_app() init_app()
clone_computers() clone_computers()
manual_actions() manual_actions()
@ -214,6 +228,11 @@ def upgrade():
def downgrade(): def downgrade():
con = op.get_bind()
devices = con.execute(f'select * from {get_inv()}.device')
if not list(devices):
return
init_app() init_app()
remove_manual_actions() remove_manual_actions()
change_lot() change_lot()