step 5

gendiff/engine.py

@@ -16,4 +16,4 @@ def format_decoder(filepath):
     elif filepath.endswith('yml') or filepath.endswith('yaml'):
         return open_yaml(filepath)
     else:
-        raise ValueError("Invalid file format!")
+        raise ValueError("Invalid file format!")

gendiff/parser.py

@@ -0,0 +1,11 @@
+import argparse
+
+
+def parse_file():
+    parser = argparse.ArgumentParser(
+        description='Compares two configuration files and shows a difference.'
+    )
+    parser.add_argument('first_file')
+    parser.add_argument('second_file')
+    parser.add_argument('-f', '--format', help='set format of output')
+    args = parser.parse_args()

gendiff/scripts/gendiff.py

@@ -1,17 +1,10 @@
 #!/usr/bin/env python3
-
-import argparse
 from gendiff.define_diff import generate_diff
+from gendiff import parser
 
 
 def main():
-    parser = argparse.ArgumentParser(
-        description='Compares two configuration files and shows a difference.'
-    )
-    parser.add_argument('first_file')
-    parser.add_argument('second_file')
-    parser.add_argument('-f', '--format', help='set format of output')
-    args = parser.parse_args()
+    first_file, second_file = parser.parse_file()
     print(generate_diff(args.first_file, args.second_file))