feat: string representation for base classes

ing-bank · Jun 28, 2022 · f618e74 · f618e74
1 parent 6891039
commit f618e74
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/popmon/base/module.py b/popmon/base/module.py
@@ -182,3 +182,11 @@ def transform(self, *args):
         :rtype: dict
         """
         raise NotImplementedError
+
+    def __repr__(self):
+        """String representation for modules when printing a pipeline/list of modules"""
+        name = self.__class__.__name__
+        input_keys = [f"{v}='{getattr(self, v)}'" for v in self._input_keys]
+        output_keys = [f"{v}='{getattr(self, v)}'" for v in self._output_keys]
+        params = ", ".join(input_keys + output_keys)
+        return f"{name}({params})"
diff --git a/popmon/base/pipeline.py b/popmon/base/pipeline.py
@@ -68,3 +68,12 @@ def transform(self, datastore):
             self.logger.debug(f"transform {module.__class__.__name__}")
             datastore = module.transform(datastore)
         return datastore
+
+    def __repr__(self):
+        """String representation for pipeline"""
+        name = self.__class__.__name__
+        ret = f"{name}: [\n"
+        for m in self.modules:
+            ret += "\t" + str(m).replace("\n", "\n\t") + "\n"
+        ret += "]"
+        return ret