From ccc9abb835dbcb4d339637d3074349a674726985 Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@ribaguifi.com>
Date: Fri, 15 Jul 2022 16:49:04 +0200
Subject: [PATCH] create placeholder for the actual device

---
 scripts/create_placeholders.py | 102 +++++++++++++++++++++++++++++++++
 1 file changed, 102 insertions(+)
 create mode 100644 scripts/create_placeholders.py

diff --git a/scripts/create_placeholders.py b/scripts/create_placeholders.py
new file mode 100644
index 00000000..035ec2ed
--- /dev/null
+++ b/scripts/create_placeholders.py
@@ -0,0 +1,102 @@
+import copy
+
+from ereuse_devicehub.config import DevicehubConfig
+from ereuse_devicehub.db import db
+from ereuse_devicehub.devicehub import Devicehub
+from ereuse_devicehub.resources.action.models import (
+    ActionDevice,
+    Allocate,
+    DataWipe,
+    Deallocate,
+    Management,
+    Prepare,
+    Ready,
+    Recycling,
+    Refurbish,
+    ToPrepare,
+    ToRepair,
+    Use,
+)
+from ereuse_devicehub.resources.device.models import Computer, Placeholder
+
+app = Devicehub(inventory=DevicehubConfig.DB_SCHEMA)
+app.app_context().push()
+
+
+def main():
+    clone_computers()
+    manual_actions()
+
+
+def clone_computers():
+    for computer in Computer.query.all():
+        clone_device(computer)
+
+
+def clone_device(device):
+    if device.binding:
+        return
+
+    dict_device = copy.copy(device.__dict__)
+    dict_device.pop('_sa_instance_state')
+    dict_device.pop('id', None)
+    dict_device.pop('devicehub_id', None)
+    dict_device.pop('actions_multiple', None)
+    dict_device.pop('actions_one', None)
+    dict_device.pop('components', None)
+    dict_device.pop('tags', None)
+    dict_device.pop('system_uuid', None)
+    new_device = device.__class__(**dict_device)
+    db.session.add(new_device)
+
+    if hasattr(device, 'components'):
+        for c in device.components:
+            new_c = clone_device(c)
+            new_c.parent = new_device
+
+    placeholder = Placeholder(device=new_device, binding=device)
+    db.session.add(placeholder)
+
+    tags = [x for x in device.tags]
+    for tag in tags:
+        tag.device = new_device
+    return new_device
+
+
+def manual_actions():
+    MANUAL_ACTIONS = (
+        Recycling,
+        Use,
+        Refurbish,
+        Management,
+        Allocate,
+        Deallocate,
+        ToPrepare,
+        Prepare,
+        DataWipe,
+        ToRepair,
+        Ready,
+    )
+
+    for action in MANUAL_ACTIONS:
+        change_device(action)
+
+
+def change_device(action):
+    for ac in action.query.all():
+        # import pdb; pdb.set_trace()
+        if hasattr(ac, 'device'):
+            if not ac.device.binding:
+                continue
+            ac.device = ac.device.binding.device
+
+        if hasattr(ac, 'devices'):
+            for act in ActionDevice.query.filter_by(action_id=ac.id):
+                if not act.device.binding:
+                    continue
+                act.device = act.device.binding.device
+
+
+if __name__ == '__main__':
+    main()
+    db.session.commit()