From 9678b663a0c970e0642e4e04ee552c38a302f785 Mon Sep 17 00:00:00 2001
From: Aarni Halinen <aarni.halinen@gmail.com>
Date: Tue, 26 Jul 2022 20:52:05 +0300
Subject: [PATCH] Update kaehmy BaseRole

---
 kaehmy/migrations/0009_auto_20220726_2015.py | 29 +++++++++++
 kaehmy/migrations/0010_auto_20220726_2033.py | 51 ++++++++++++++++++++
 kaehmy/models.py                             |  4 +-
 sikweb/custom_operations.py                  | 30 ++++++++++++
 4 files changed, 112 insertions(+), 2 deletions(-)
 create mode 100644 kaehmy/migrations/0009_auto_20220726_2015.py
 create mode 100644 kaehmy/migrations/0010_auto_20220726_2033.py
 create mode 100644 sikweb/custom_operations.py

diff --git a/kaehmy/migrations/0009_auto_20220726_2015.py b/kaehmy/migrations/0009_auto_20220726_2015.py
new file mode 100644
index 0000000..d34942a
--- /dev/null
+++ b/kaehmy/migrations/0009_auto_20220726_2015.py
@@ -0,0 +1,29 @@
+# Generated by Django 2.2.28 on 2022-07-26 17:15
+
+from unicodedata import category
+from django.db import migrations
+
+
+def copyBaseRolesToNewTable(apps, schema_editor):
+    Old = apps.get_model("kaehmy", "KaehmyBaseRole")
+    New = apps.get_model("kaehmy", "BaseRole")
+    for bases in Old.objects.all():
+        New.objects.create(
+            id=bases.id,
+            name=bases.name,
+            is_board=bases.is_board,
+            category=bases.category,
+        )
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("kaehmy", "0008_baserole"),
+    ]
+
+    operations = [
+        migrations.RunPython(
+            copyBaseRolesToNewTable, reverse_code=migrations.RunPython.noop
+        ),
+    ]
diff --git a/kaehmy/migrations/0010_auto_20220726_2033.py b/kaehmy/migrations/0010_auto_20220726_2033.py
new file mode 100644
index 0000000..f881a0e
--- /dev/null
+++ b/kaehmy/migrations/0010_auto_20220726_2033.py
@@ -0,0 +1,51 @@
+# Generated by Django 2.2.28 on 2022-07-26 17:33
+
+from django.db import migrations, models
+import django.db.models.deletion
+from sikweb.custom_operations import AlterModelBases
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("kaehmy", "0009_auto_20220726_2015"),
+    ]
+
+    operations = [
+        AlterModelBases("customrole", (models.Model,)),
+        AlterModelBases("presetrole", (models.Model,)),
+        migrations.AlterField(
+            model_name="customrole",
+            name="kaehmybaserole_ptr",
+            field=models.OneToOneField(
+                auto_created=True,
+                on_delete=django.db.models.deletion.CASCADE,
+                parent_link=True,
+                primary_key=True,
+                serialize=False,
+                to="BaseRole",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="presetrole",
+            name="kaehmybaserole_ptr",
+            field=models.OneToOneField(
+                auto_created=True,
+                on_delete=django.db.models.deletion.CASCADE,
+                parent_link=True,
+                primary_key=True,
+                serialize=False,
+                to="BaseRole",
+            ),
+        ),
+        migrations.RenameField(
+            model_name="customrole",
+            old_name="kaehmybaserole_ptr",
+            new_name="baserole_ptr",
+        ),
+        migrations.RenameField(
+            model_name="presetrole",
+            old_name="kaehmybaserole_ptr",
+            new_name="baserole_ptr",
+        ),
+    ]
diff --git a/kaehmy/models.py b/kaehmy/models.py
index ba22f71..c20a865 100644
--- a/kaehmy/models.py
+++ b/kaehmy/models.py
@@ -67,7 +67,7 @@ class KaehmyBaseRole(OldBaseRole):
     )
 
 
-class PresetRole(KaehmyBaseRole):
+class PresetRole(BaseRole):
     """Model for kaehmy role."""
 
     description = models.TextField(_("Description"))
@@ -77,7 +77,7 @@ class PresetRole(KaehmyBaseRole):
         verbose_name_plural = _("Preset kaehmy roles")
 
 
-class CustomRole(KaehmyBaseRole):
+class CustomRole(BaseRole):
     """Model representing a user-specified custom occupation."""
 
     class Meta:
diff --git a/sikweb/custom_operations.py b/sikweb/custom_operations.py
new file mode 100644
index 0000000..2a32ca5
--- /dev/null
+++ b/sikweb/custom_operations.py
@@ -0,0 +1,30 @@
+# contents of yourapp/migrations/custom_operations.py
+
+from django.db.migrations.operations.models import ModelOperation
+
+
+class AlterModelBases(ModelOperation):
+    reduce_to_sql = False
+    reversible = True
+
+    def __init__(self, name, bases):
+        self.bases = bases
+        super().__init__(name)
+
+    def state_forwards(self, app_label, state):
+        """
+        Overwrite a models base classes with a custom list of
+        bases instead, then force Django to reload the model
+        with this (probably completely) different class hierarchy.
+        """
+        state.models[app_label, self.name_lower].bases = self.bases
+        state.reload_model(app_label, self.name_lower)
+
+    def database_forwards(self, app_label, schema_editor, from_state, to_state):
+        pass
+
+    def database_backwards(self, app_label, schema_editor, from_state, to_state):
+        pass
+
+    def describe(self):
+        return "Update %s bases to %s" % (self.name, self.bases)