step9

Author: Nurzhan2023

gendiff/formaters/stylish.py

@@ -18,28 +18,29 @@ def format_value(value, depth):
     return str(value)
 
 
-
 def format_stylish(diff, depth=0):
     indent = " " * (depth * 4)
     lines = ["{"]
 
     for key, info in sorted(diff.items()):
-        status = info["status"]
-        value = format_value(info.get("value"), depth + 1)
-        old_value = format_value(info.get("old_value"), depth + 1)
-        new_value = format_value(info.get("new_value"), depth + 1)
+        status = info.get("status")
 
         if status == "nested":
-            children = format_stylish(info["children"], depth + 1)
+            children = format_stylish(info.get("children", {}), depth + 1)
             lines.append(f"{indent}    {key}: {children}")
         elif status == "added":
+            value = format_value(info.get("value"), depth + 1)
             lines.append(f"{indent}  + {key}: {value}")
         elif status == "removed":
+            value = format_value(info.get("value"), depth + 1)
             lines.append(f"{indent}  - {key}: {value}")
         elif status == "modified":
+            old_value = format_value(info.get("old_value"), depth + 1)
+            new_value = format_value(info.get("new_value"), depth + 1)
             lines.append(f"{indent}  - {key}: {old_value}")
             lines.append(f"{indent}  + {key}: {new_value}")
         else:  # unchanged
+            value = format_value(info.get("value"), depth + 1)
             lines.append(f"{indent}    {key}: {value}")
 
     lines.append(f"{indent}}}")

gendiff/parser.py

@@ -15,6 +15,5 @@ def args_parser():
     )
     args = parser.parse_args()
 
-    diff = generate_diff(args.first_file, args.second_file, 
+    return generate_diff(args.first_file, args.second_file, 
                          format_name=args.format)
-    print(diff)