Move Area to Omni/App.py

This matches Omni/App.hs, and I'll use it in future projects.

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)