Add yaml extract from png

wireviz · Jun 7, 2021 · bc56063 · bc56063
1 parent 75716d3
commit bc56063
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/wireviz/wireviz.py b/src/wireviz/wireviz.py
@@ -247,6 +247,8 @@ def main():
 
  if ".png" in args.input_file:
  yaml_input = read_yaml_from_png(args.input_file.replace('.png',''))
+ with open(args.input_file.replace('.png','_out.yaml'),'w') as fh:
+ fh.write(yaml_input) # Extract yaml to separate file
  else:
  with open_file_read(args.input_file) as fh:
  yaml_input = fh.read()