basic config support

3 files changed, 52 insertions, 0 deletions

diff --git a/pokemontools/__init__.py b/pokemontools/__init__.py
index 8fb8b19..09331af 100644
--- a/pokemontools/__init__.py
+++ b/pokemontools/__init__.py
@@ -1,2 +1,3 @@
+import config
 import crystal
 import preprocessor
diff --git a/pokemontools/config.py b/pokemontools/config.py
new file mode 100644
index 0000000..656fab0
--- /dev/null
+++ b/pokemontools/config.py
@@ -0,0 +1,46 @@
+"""
+Configuration
+"""
+
+import os
+
+import exceptions
+
+class Config(object):
+    """
+    The Config class handles all configuration for pokemontools. Other classes
+    and functions use a Config object to determine where expected files can be
+    located.
+    """
+
+    def __init__(self, **kwargs):
+        """
+        Store all parameters.
+        """
+        self._config = {}
+
+        for (key, value) in kwargs.items():
+            if key not in self.__dict__:
+                self._config[key] = value
+            else:
+                raise exceptions.ConfigException(
+                    "Can't store \"{0}\" in configuration because the key conflicts with an existing property."
+                    .format(key)
+                )
+
+        if "path" not in self._config:
+            self._config["path"] = os.getcwd()
+
+    def __getattr__(self, key):
+        """
+        Grab the value from the class properties, then check the configuration,
+        and raise an exception if nothing works.
+        """
+        if key in self.__dict__:
+            return self.__dict__[key]
+        elif key in self._config:
+            return self._config[key]
+        else:
+            raise exceptions.ConfigException(
+                "no config found for \"{0}\"".format(key)
+            )
diff --git a/pokemontools/exceptions.py b/pokemontools/exceptions.py
index 71d0da2..2ccc58a 100644
--- a/pokemontools/exceptions.py
+++ b/pokemontools/exceptions.py
@@ -11,3 +11,8 @@ class TextScriptException(Exception):
     """
     TextScript encountered an inconsistency or problem.
     """
+
+class ConfigException(Exception):
+    """
+    Configuration error. Maybe a missing config variable.
+    """