This repository has been archived on 2024-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
devicehub-teal/ereuse_devicehub/migrations/versions/bf600ca861a4_adding_hid.py

66 lines
1.6 KiB
Python
Raw Normal View History

"""empty message
Revision ID: bf600ca861a4
Revises: 68a5c025ab8e
Create Date: 2020-12-15 15:58:41.545563
"""
from alembic import context
from alembic import op
import sqlalchemy as sa
import sqlalchemy_utils
import citext
import teal
# revision identifiers, used by Alembic.
revision = 'bf600ca861a4'
down_revision = '68a5c025ab8e'
branch_labels = 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 upgrade():
con = op.get_bind()
sql = f"""
select d.id, d.hid, dd.serial_number from {get_inv()}.computer as c
join {get_inv()}.device as d on c.id=d.id
inner join {get_inv()}.component as cmp on cmp.parent_id=c.id
inner join {get_inv()}.network_adapter as net on net.id=cmp.id
join {get_inv()}.device as dd on net.id=dd.id;
"""
computers = con.execute(sql)
hids = {}
macs = {}
for c in computers:
hids[c.id] = c.hid
if not c.serial_number:
continue
try:
macs[c.id].append(c.serial_number)
macs[c.id].sort()
except:
macs[c.id] = [c.serial_number]
for id_dev, hid in hids.items():
if not (id_dev and hid):
continue
if not id_dev in macs:
continue
mac = macs[id_dev][0]
new_hid = "{}-{}".format(hid, mac)
sql = f"update {get_inv()}.device set hid='{new_hid}' where id={id_dev};"
con.execute(sql)
def downgrade():
pass