1 files changed, 23 insertions, 0 deletions
diff --git a/src/ee/doit.py b/src/ee/doit.py
index 87a6601..014af05 100644
--- a/src/ee/doit.py
+++ b/src/ee/doit.py
@@ -2,6 +2,8 @@ import logging
 
 from doit import get_var
 
+from ee.ds import DataSetManager
+
 
 def configure_logging():
     log_level = get_var("log-level", None)
@@ -14,3 +16,24 @@ def configure_logging():
         ee_logger.addHandler(ch)
 
         ee_logger.setLevel(log_level)
+
+
+class DoItConfig(object):
+    def __init__(self):
+        self._dsm = None  # type: DataSetManager
+        self._data_sets = {}
+
+    def configure(self, *, data_set_manager: DataSetManager):
+        self._dsm = data_set_manager
+
+    @property
+    def dsm(self):
+        if not self._dsm:
+            raise Exception("The DataSetManager has not been set")
+        return self._dsm
+
+    def data_sets_for(self, task):
+        return self._data_sets[task]
+
+    def set_data_sets_for(self, task, out_dataset: str, *in_datasets: str):
+        self._data_sets[task] = [out_dataset, in_datasets]