From 9014d53399b501cee48d1b80bc5e0d59c229134a Mon Sep 17 00:00:00 2001
From: Michael Manfre
Date: Tue, 22 Nov 2022 23:06:21 -0500
Subject: StatorRunner will refresh Config.system on each schedule_interval

---
 core/models/config.py | 35 +++++++++++++++++++++++++++++++++--
 1 file changed, 33 insertions(+), 2 deletions(-)

(limited to 'core/models')

diff --git a/core/models/config.py b/core/models/config.py
index dca8a0c..0a8d624 100644
--- a/core/models/config.py
+++ b/core/models/config.py
@@ -2,6 +2,7 @@ from functools import partial
 from typing import ClassVar
 
 import pydantic
+from asgiref.sync import sync_to_async
 from django.core.files import File
 from django.db import models
 from django.templatetags.static import static
@@ -88,26 +89,56 @@ class Config(models.Model):
             {"identity__isnull": True, "user__isnull": True},
         )
 
+    @classmethod
+    async def aload_system(cls):
+        """
+        Async loads the system config options object
+        """
+        return await sync_to_async(cls.load_values)(
+            cls.SystemOptions,
+            {"identity__isnull": True, "user__isnull": True},
+        )
+
     @classmethod
     def load_user(cls, user):
         """
         Loads a user config options object
         """
         return cls.load_values(
-            cls.SystemOptions,
+            cls.UserOptions,
+            {"identity__isnull": True, "user": user},
+        )
+
+    @classmethod
+    async def aload_user(cls, user):
+        """
+        Async loads the user config options object
+        """
+        return await sync_to_async(cls.load_values)(
+            cls.UserOptions,
             {"identity__isnull": True, "user": user},
         )
 
     @classmethod
     def load_identity(cls, identity):
         """
-        Loads a user config options object
+        Loads an identity config options object
         """
         return cls.load_values(
             cls.IdentityOptions,
             {"identity": identity, "user__isnull": True},
         )
 
+    @classmethod
+    async def aload_identity(cls, identity):
+        """
+        Async loads an identity config options object
+        """
+        return await sync_to_async(cls.load_values)(
+            cls.IdentityOptions,
+            {"identity": identity, "user__isnull": True},
+        )
+
     @classmethod
     def set_value(cls, key, value, options_class, filters):
         config_field = options_class.__fields__[key]
-- 
cgit v1.2.3