Add support for multiple YAML files

You can now do:

  `sshush -s ~/.ssh/config.d/*.yml`

And it will process every .yml file it finds.

Author: bencromwell

sshush/command_line.py

@@ -12,7 +12,8 @@ def main():
     arg_parser.add_argument(
         '--source', '-s',
         help='Path to source YAML file if it differs from {}'.format(default_yaml_path),
-        default=default_yaml_path
+        default=default_yaml_path,
+        nargs='+'
     )
 
     arg_parser.add_argument(
@@ -28,14 +29,19 @@ def main():
         yaml=args.source
     ))
 
-    yaml_obj = read_file(args.source)
-    config_file_contents = process_yaml(yaml_obj)
-
     try:
         with open(args.path, 'w') as fh:
-            fh.write(config_file_contents)
-            fh.write("\n")
+            fh.write('# Generated by sshush')
+
+            for file in args.source:
+                yaml_obj = read_file(file)
+                config_file_contents = process_yaml(yaml_obj)
+
+                fh.write(config_file_contents)
+                fh.write("\n")
+
             print('{} written successfully'.format(args.path))
+
     except IOError as exc:
         print('Error:', exc.strerror)
         exit(1)

sshush/sshush.py

@@ -28,7 +28,7 @@ def read_file(file):
 
 
 def process_yaml(ssh_config_yaml):
-    output = ['# Generated by sshush']
+    output = []
 
     defaults = extract_section('default', ssh_config_yaml, {})
     global_values = extract_section('global', ssh_config_yaml)