feat: configure root dir using config file

Breakthrough-Energy · May 13, 2021 · 756b028 · 756b028
1 parent 01895bd
commit 756b028
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,6 @@
+# This is specific to this package
+config.ini
+
 # This file taken from github/gitignore
 # https://github.com/github/gitignore/blob/master/Python.gitignore
 

diff --git a/pyreisejl/utility/const.py b/pyreisejl/utility/const.py
@@ -1,6 +1,15 @@
+import configparser
 import os
+from pathlib import Path
 
-DATA_ROOT_DIR = "/mnt/bes/pcm"
+INI_FILE = "config.ini"
+if Path(INI_FILE).exists():
+    conf = configparser.ConfigParser()
+    conf.read(INI_FILE)
+    for k, v in conf["REISE.jl"].items():
+        os.environ[k.upper()] = v
+
+DATA_ROOT_DIR = os.getenv("DATA_ROOT_DIR", "/mnt/bes/pcm")
 
 SCENARIO_LIST = os.path.join(DATA_ROOT_DIR, "ScenarioList.csv")
 EXECUTE_LIST = os.path.join(DATA_ROOT_DIR, "ExecuteList.csv")