summaryrefslogtreecommitdiff
path: root/Omni/App.py
diff options
context:
space:
mode:
Diffstat (limited to 'Omni/App.py')
-rw-r--r--Omni/App.py26
1 files changed, 26 insertions, 0 deletions
diff --git a/Omni/App.py b/Omni/App.py
new file mode 100644
index 0000000..0c6776c
--- /dev/null
+++ b/Omni/App.py
@@ -0,0 +1,26 @@
+"""General utilities for apps."""
+
+import enum
+import os
+
+
+class Area(enum.Enum):
+ """The area we are running."""
+
+ Test = "Test"
+ Live = "Live"
+
+
+def from_env() -> Area:
+ """Load AREA from environment variable.
+
+ Raises:
+ ValueError: if AREA is not defined
+ """
+ var = os.getenv("AREA", "Test")
+ if var == "Test":
+ return Area.Test
+ if var == "Live":
+ return Area.Live
+ msg = "AREA not defined"
+ raise ValueError(msg)