51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
# THIS FILE SIMPLY LOADS THE YAML FILE INTO PYYAML AND THEN SPITS IT BACK OUT
|
||
# TO CLEAN UP AND ORDER ALL THE YAML
|
||
|
||
import yaml
|
||
import glob
|
||
import os
|
||
|
||
|
||
|
||
def main():
|
||
|
||
# gets all files with a yaml extension in the directory
|
||
yfiles = []
|
||
for file in glob.glob("*.yaml"):
|
||
yfiles.append(file)
|
||
|
||
yfiles.sort()
|
||
print("Going to clean up the following files: {}".format(yfiles))
|
||
|
||
for x in yfiles:
|
||
print("Doing: {}".format(x))
|
||
with open(x, 'r') as r:
|
||
data = yaml.full_load(r)
|
||
if x == "feats.yaml":
|
||
for i in data['feat']:
|
||
# This is to clean out smart quotes that made it into the
|
||
# yaml file so it matches the requirements.yaml
|
||
if i['requirement'] != None:
|
||
# print("Before: {}".format(i['requirement']))
|
||
i['requirement'] = i['requirement'].replace('’', "'")
|
||
# print("After: {}".format(i['requirement']))
|
||
if i['trigger'] != None:
|
||
# print("Before: {}".format(i['trigger']))
|
||
i['trigger'] = i['trigger'].replace('’', "'")
|
||
# print("After: {}".format(i['trigger']))
|
||
if x == "triggers.yaml":
|
||
for i in data['trigger']:
|
||
print(i)
|
||
i = i.replace('’', "'")
|
||
|
||
|
||
|
||
final = yaml.safe_dump(data, allow_unicode=True)
|
||
with open(x, 'w') as f:
|
||
f.write(final)
|
||
print("\tDone.")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|