From b2918852b6e4f9aaf0382e5410b8202d5e59eaa9 Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@ribaguifi.com>
Date: Tue, 27 Sep 2022 12:51:31 +0200
Subject: [PATCH] add migrations for monitors and mobiles

---
 .../e919fe0611ff_placeholder_monitors.py      | 86 +++++++++++++++++++
 1 file changed, 86 insertions(+)
 create mode 100644 ereuse_devicehub/migrations/versions/e919fe0611ff_placeholder_monitors.py

diff --git a/ereuse_devicehub/migrations/versions/e919fe0611ff_placeholder_monitors.py b/ereuse_devicehub/migrations/versions/e919fe0611ff_placeholder_monitors.py
new file mode 100644
index 00000000..cf7efdbc
--- /dev/null
+++ b/ereuse_devicehub/migrations/versions/e919fe0611ff_placeholder_monitors.py
@@ -0,0 +1,86 @@
+"""placeholder-monitors
+
+Revision ID: e919fe0611ff
+Revises: bcfda54aaf2f
+Create Date: 2022-09-27 10:55:00.859262
+
+"""
+from alembic import context
+
+from ereuse_devicehub.db import db
+from ereuse_devicehub.devicehub import Devicehub
+from ereuse_devicehub.resources.device.models import (
+    Cellphone,
+    ComputerMonitor,
+    Display,
+    Mobile,
+    Monitor,
+    Placeholder,
+    Smartphone,
+    Tablet,
+)
+
+# revision identifiers, used by Alembic.
+revision = 'e919fe0611ff'
+down_revision = 'bcfda54aaf2f'
+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 init_app():
+    app = Devicehub(inventory=f'{get_inv()}')
+    app.app_context().push()
+
+
+def clone_monitors():
+    devices = [ComputerMonitor, Monitor, Display, Smartphone, Tablet, Cellphone, Mobile]
+    for dev in devices:
+        for d in dev.query.all():
+            if d.placeholder or d.binding:
+                continue
+            clone_device(d)
+
+
+def clone_device(device):
+    if not device.owner_id:
+        return
+    placeholder = Placeholder(
+        device=device, is_abstract=False, owner_id=device.owner_id
+    )
+    db.session.add(placeholder)
+
+
+def remove_placeholders():
+    devices = [ComputerMonitor, Monitor, Display, Smartphone, Tablet, Cellphone, Mobile]
+    for dev in devices:
+        for d in dev.query.all():
+            if d.placeholder and d.binding:
+                continue
+            remove_device(d)
+
+
+def remove_device(device):
+    if not device.owner_id:
+        return
+    placeholder = device.placeholder
+    db.session.delete(placeholder)
+
+
+def upgrade():
+    init_app()
+    clone_monitors()
+    db.session.commit()
+
+
+def downgrade():
+    pass
+    # init_app()
+    # remove_placeholders()
+    # db.session.commit()