From 7408e658a811a5431c1916d07836fbe29adeea96 Mon Sep 17 00:00:00 2001
From: Ben Sima <ben@bsima.me>
Date: Tue, 21 Jan 2025 03:45:05 -0500
Subject: Move Area to Omni/App.py

This matches Omni/App.hs, and I'll use it in future projects.
---
 Omni/App.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 Omni/App.py

(limited to 'Omni')

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)
-- 
cgit v1.2.3