Nouveau système de configuration de "base" (ROOT_DIR, ...)

Inspiré de https://whalesalad.com/blog/doing-python-configuration-right et https://12factor.net/config

Nouveau système de configuration de "base" (ROOT_DIR, ...)
Inspiré de https://whalesalad.com/blog/doing-python-configuration-right et https://12factor.net/config
Etienne Pallier
1 parent d02a5e4b
Showing 6 changed files with 95 additions and 0 deletions Show diff stats
config/__init__.py
config/config_base.py
config/config_dev.py
config/config_prod.py
config/config_test.py
src/core/pyros_django/agent/Agent.py
@@ -0,0 +1,21 @@
+# Inspired from https://whalesalad.com/blog/doing-python-configuration-right
+
+import os
+import importlib
+
+# Determine the environment we want to load, default is development.py
+PYROS_ENV = os.environ.get("PYROS_ENV", "config_dev")
+
+module = importlib.import_module(
+    #f".environments.{ENV}", package="service.config"
+    f".{PYROS_ENV}", package="config"
+)
+
+# update globals of this module (i.e. settings) with imported.
+globals().update(vars(module))
+
+
+def is_dev_env(): return PYROS_ENV == "config_dev"
+def is_test_env(): return PYROS_ENV == "config_test"
+def is_prod_env(): return PYROS_ENV == "config_prod"
+
@@ -0,0 +1,27 @@
+# Inspired from https://whalesalad.com/blog/doing-python-configuration-right
+
+import os
+
+APP_NAME = "PYROS"
+
+# Conveniences to always have a quick reference to the
+# top-level application directory.
+ROOT_DIR = os.path.join(
+    os.path.dirname(os.path.realpath(__file__)),
+    os.pardir, # '..'
+)
+ROOT_DIR = os.path.realpath(ROOT_DIR)
+
+# root directories
+DOC_DIR = os.path.join(ROOT_DIR, "doc")
+INSTALL_DIR = os.path.join(ROOT_DIR, "install")
+LOG_DIR = os.path.join(ROOT_DIR, "config", "logs")
+SOURCE_DIR = os.path.join(ROOT_DIR, "src")
+
+# source sub-directories
+CELME_DIR = os.path.join(SOURCE_DIR, "core", "celme")
+PYROS_DJANGO_DIR = os.path.join(SOURCE_DIR, "core", "pyros_django")
+DEVICES_DIR = os.path.join(SOURCE_DIR, "device_controller")
+
+
+SYSTEM_REBOOT_COMMAND = "sudo systemctl restart foo.service"
 \ No newline at end of file
@@ -0,0 +1,10 @@
+import os
+
+from .config_base import *
+
+# Now, override some configuration settings, or add new ones
+
+# Ex:
+# Override DOC_DIR
+DOC_DIR = os.path.join(DOC_DIR, "dev-doc")
+
@@ -0,0 +1,10 @@
+import os
+
+from .config_base import *
+
+# Now, override some configuration settings, or add new ones
+
+# Ex:
+# Override DOC_DIR
+DOC_DIR = os.path.join(DOC_DIR, "prod-doc")
+
@@ -0,0 +1,10 @@
+import os
+
+from .config_base import *
+
+# Now, override some configuration settings, or add new ones
+
+# Ex:
+# Override DOC_DIR
+DOC_DIR = os.path.join(DOC_DIR, "test-doc")
+
@@ -110,6 +110,11 @@ from django import db
 #from django.conf import settings as djangosettings
  
 # --- SPECIFIC IMPORT ---
+# Many ways to import configuration settings:
+import config
+#from config import PYROS_ENV, ROOT_DIR, DOC_DIR
+#from config import *
+
 from common.models import AgentSurvey, Command, AgentLogs
 from config.configpyros import ConfigPyros
 #from dashboard.views import get_sunelev
@@ -402,6 +407,18 @@ class Agent:
     #def __init__(self, name:str="Agent", config_filename:str=None, RUN_IN_THREAD=True):
     #def __init__(self, config_filename:str=None, RUN_IN_THREAD=True, DEBUG_MODE=False):
     def __init__(self, config_filename:str=None, RUN_IN_THREAD=True):
+
+        '''
+        print('PYROS_ENV', PYROS_ENV)
+        print('ROOT_DIR', ROOT_DIR)
+        print('DOC_DIR', DOC_DIR)
+        '''
+        print('PYROS_ENV', config.PYROS_ENV)
+        print('ROOT_DIR', config.ROOT_DIR)
+        print('DOC_DIR', config.DOC_DIR)
+        if config.is_dev_env(): print("DEV ENV")
+        if config.is_prod_env(): print("PROD ENV")
+
         #self.name = name
         self.name = self.__class__.__name__
         printd("*** ENVIRONMENT VARIABLE PYROS_DEBUG is:", os.environ.get('PYROS_DEBUG'), '***')