mirror of
https://github.com/metafy-social/python-scripts.git
synced 2024-11-24 04:21:12 +00:00
36 lines
932 B
Python
36 lines
932 B
Python
|
import json
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import yaml
|
||
|
|
|
||
|
|
# Checking there is a file name passed
|
||
|
|
if len(sys.argv) > 1:
|
||
|
|
# Opening the file
|
||
|
|
if os.path.exists(sys.argv[1]):
|
||
|
|
source_file = open(sys.argv[1], "r")
|
||
|
|
source_content = yaml.safe_load(source_file)
|
||
|
|
source_file.close()
|
||
|
|
# Failikng if the file isn't found
|
||
|
|
else:
|
||
|
|
print("ERROR: " + sys.argv[1] + " not found")
|
||
|
|
exit(1)
|
||
|
|
# No file, no usage
|
||
|
|
else:
|
||
|
|
print("Usage: yaml2json.py <source_file.yaml> [target_file.json]")
|
||
|
|
|
||
|
|
# Processing the conversion
|
||
|
|
output = json.dumps(source_content)
|
||
|
|
|
||
|
|
# If no target file send to stdout
|
||
|
|
if len(sys.argv) < 3:
|
||
|
|
print(output)
|
||
|
|
# If the target file already exists exit
|
||
|
|
elif os.path.exists(sys.argv[2]):
|
||
|
|
print("ERROR: " + sys.argv[2] + " already exists")
|
||
|
|
exit(1)
|
||
|
|
# Otherwise write to the specified file
|
||
|
|
else:
|
||
|
|
target_file = open(sys.argv[2], "w")
|
||
|
|
target_file.write(output)
|
||
|
|
target_file.close()
|