Better way to initialise things on setup.

1 files changed, 12 insertions, 6 deletions

diff --git a/users/apps.py b/users/apps.py
index 86a8eb3..67183ad 100644
--- a/users/apps.py
+++ b/users/apps.py
@@ -1,14 +1,20 @@
 from django.apps import AppConfig
-from django.conf import settings
+from django.db.models.signals import post_migrate
 
 
 class UsersConfig(AppConfig):
     default_auto_field = "django.db.models.BigAutoField"
     name = "users"
 
-    def ready(self) -> None:
-        if not settings.IN_TESTS:
-            # Generate the server actor keypair if needed
-            from users.models import SystemActor
+    def data_init(self, **kwargs):
+        """
+        Runs after migrations or flushes to insert anything we need for first
+        boot (or post upgrade).
+        """
+        # Generate the server actor keypair if needed
+        from users.models import SystemActor
+
+        SystemActor.generate_keys_if_needed()
 
-            SystemActor.generate_keys_if_needed()
+    def ready(self) -> None:
+        post_migrate.connect(self.data_init, sender=self)