Add class for main runner

Signed-off-by: cmuraru <[email protected]>

Author: cmuraru

hierarchical_yaml/__init__.py

@@ -9,3 +9,4 @@
 # governing permissions and limitations under the License.
 
 from .config_generator import ConfigGenerator, ConfigProcessor
+from .main import ConfigRunner

hierarchical_yaml/main.py

@@ -14,35 +14,44 @@
 from .config_generator import ConfigProcessor
 
 
+class ConfigRunner(object):
+
+    def run(self, args):
+        parser = self.get_parser()
+        opts = parser.parse_args(args)
+        cwd = opts.cwd if opts.cwd else os.getcwd()
+        filters = opts.filter if opts.filter else ()
+        excluded_keys = opts.exclude if opts.exclude else ()
+        if opts.output_file is None:
+            opts.print_data = True
+
+        config_processor = ConfigProcessor()
+        config_processor.process(cwd, opts.path, filters, excluded_keys, opts.enclosing_key, opts.output_format,
+                                 opts.print_data, opts.output_file, opts.skip_interpolation_resolving,
+                                 opts.skip_interpolation_validation)
+
+    @staticmethod
+    def get_parser():
+        parser = argparse.ArgumentParser()
+        parser.add_argument('path', type=str, help='The config directory')
+        parser.add_argument('--output-file', dest='output_file', type=str,
+                            help='output file location')
+        parser.add_argument('--format', dest='output_format', type=str, default="yaml",
+                            help='output file format')
+        parser.add_argument('--filter', dest='filter', action='append',
+                            help='keep these keys from the generated data')
+        parser.add_argument('--exclude', dest='exclude', action='append',
+                            help='exclude these keys from generated data')
+        parser.add_argument('--skip-interpolation-validation', action='store_true',
+                            help='will not throw an error if interpolations can not be resolved')
+        parser.add_argument('--skip-interpolation-resolving', action='store_true',
+                            help='do not perform any AWS calls to resolve interpolations')
+        parser.add_argument('--enclosing-key', dest='enclosing_key', type=str,
+                            help='enclose the generated data under a common key')
+        parser.add_argument('--cwd', dest='cwd', type=str, default="",
+                            help='the working directory')
+        return parser
+
+
 def run(args=None):
-    """ App entry point """
-
-    parser = argparse.ArgumentParser()
-    parser.add_argument('path', type=str, help='The config directory')
-    parser.add_argument('--output-file', dest='output_file', type=str,
-                        help='output file location')
-    parser.add_argument('--format', dest='output_format', type=str, default="yaml",
-                        help='output file format')
-    parser.add_argument('--filter', dest='filter', action='append',
-                        help='keep these keys from the generated data')
-    parser.add_argument('--exclude', dest='exclude', action='append',
-                        help='exclude these keys from generated data')
-    parser.add_argument('--skip-interpolation-validation', action='store_true',
-                        help='will not throw an error if interpolations can not be resolved')
-    parser.add_argument('--skip-interpolation-resolving', action='store_true',
-                        help='do not perform any AWS calls to resolve interpolations')
-    parser.add_argument('--enclosing-key', dest='enclosing_key', type=str,
-                        help='enclose the generated data under a common key')
-    parser.add_argument('--cwd', dest='cwd', type=str, default="",
-                        help='the working directory')
-
-    opts = parser.parse_args(args)
-    cwd = opts.cwd if opts.cwd else os.getcwd()
-    filters = opts.filter if opts.filter else ()
-    excluded_keys = opts.exclude if opts.exclude else ()
-    print_data = True
-
-    config_processor = ConfigProcessor()
-    config_processor.process(cwd, opts.path, filters, excluded_keys, opts.enclosing_key, opts.output_format,
-                             print_data, opts.output_file, opts.skip_interpolation_resolving,
-                             opts.skip_interpolation_validation)
+    ConfigRunner().run(args)