From 08300ef612f8240f183e3b2e200bef69202da285 Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@ribaguifi.com>
Date: Thu, 16 Jun 2022 10:38:57 +0200
Subject: [PATCH] add system_uuid in old devices

---
 ...b497b3_add_system_uuid_to_old_registers.py | 61 +++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 ereuse_devicehub/migrations/versions/8d4fe4b497b3_add_system_uuid_to_old_registers.py

diff --git a/ereuse_devicehub/migrations/versions/8d4fe4b497b3_add_system_uuid_to_old_registers.py b/ereuse_devicehub/migrations/versions/8d4fe4b497b3_add_system_uuid_to_old_registers.py
new file mode 100644
index 00000000..db363a4c
--- /dev/null
+++ b/ereuse_devicehub/migrations/versions/8d4fe4b497b3_add_system_uuid_to_old_registers.py
@@ -0,0 +1,61 @@
+"""add system uuid to old registers
+
+Revision ID: 8d4fe4b497b3
+Revises: 73348969a583
+Create Date: 2022-06-15 15:52:39.205192
+
+"""
+import os
+from uuid import UUID
+
+from alembic import context, op
+
+# revision identifiers, used by Alembic.
+revision = '8d4fe4b497b3'
+down_revision = '73348969a583'
+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 update_db(con, system_uuid, snapshot_uuid):
+    sql_snapshot = f'select id from {get_inv()}.snapshot where uuid=\'{snapshot_uuid}\''
+    sql_device_id = f'select device_id from {get_inv()}.action_with_one_device where id in ({sql_snapshot})'
+    sql = f'select id, system_uuid from {get_inv()}.computer where id in ({sql_device_id})'
+
+    for device_id, db_system_uuid in con.execute(sql):
+        if db_system_uuid:
+            return
+
+        sql = f'update {get_inv()}.computer set system_uuid=\'{system_uuid}\' where id=\'{device_id}\''
+        con.execute(sql)
+
+
+def update_to_little_endian(uuid):
+    uuid = UUID(uuid)
+    return UUID(bytes_le=uuid.bytes)
+
+
+def upgrade():
+    uuids = []
+    system_uuids_file = 'system_uuids.csv'
+    if os.path.exists(system_uuids_file):
+        with open(system_uuids_file) as f:
+            uuids = [x.split(';') for x in f.read().split('\n')]
+
+    con = op.get_bind()
+    for u in uuids[1:]:
+        if u[0] == '':
+            continue
+        u[0] = update_to_little_endian(u[0])
+        update_db(con, u[0], u[1])
+
+
+def downgrade():
+    pass