diff options
Diffstat (limited to 'Omni/App.py')
-rw-r--r-- | Omni/App.py | 26 |
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) |