Simplify Converter init

pyplanner/cli.py

@@ -17,9 +17,7 @@ def main(ctx, db_url):
     ctx.ensure_object(dict)
     ctx.obj['db'] = Database(db_url)
     ctx.obj['output'] = TerminalOutput()
-    converter = ConverterYaml()
-    converter.register(ctx.obj['db'].models.values())
-    ctx.obj['converter'] = converter
+    ctx.obj['converter'] = ConverterYaml(ctx.obj['db'].models.values())
 
 
 @main.command()

pyplanner/converters.py

@@ -2,12 +2,13 @@ import ruamel.yaml
 
 
 class ConverterYaml:
-    def __init__(self):
+    def __init__(self, class_list):
         self.yaml = ruamel.yaml.YAML()
         self.yaml.default_flow_style = False
         self.yaml.allow_unicode = True
+        self._register(class_list)
 
-    def register(self, class_list):
+    def _register(self, class_list):
         for class_ in class_list:
             self.yaml.register_class(class_)