Merge branch 'standardizeyaml'
commit
8ee4c63ee3
File diff suppressed because it is too large
Load Diff
|
@ -0,0 +1,48 @@
|
|||
import yaml
|
||||
import pprint
|
||||
|
||||
|
||||
def main():
|
||||
|
||||
standardize_monsters()
|
||||
|
||||
|
||||
def standardize_monsters():
|
||||
with open('monsters.yaml', 'r') as f:
|
||||
data = yaml.full_load(f)
|
||||
goodset = set()
|
||||
for i in data['monsters']:
|
||||
deletekeys = []
|
||||
for k, v in i.items():
|
||||
# print("Key:{}\tValue:{}".format(k,v))
|
||||
if v == None:
|
||||
print("Deleting key: {} from {}".format(k, i['name']))
|
||||
deletekeys.append(k)
|
||||
else:
|
||||
goodset.add(k)
|
||||
# got to do it this way or your get a runtime error for changing the
|
||||
# index mid iteration
|
||||
for z in deletekeys:
|
||||
del (i[z])
|
||||
print(goodset)
|
||||
|
||||
for i in data['monsters']:
|
||||
for k in goodset:
|
||||
if k not in i:
|
||||
print("Adding field {} to {}".format(k, i['name']))
|
||||
i[k] = None
|
||||
|
||||
final = yaml.safe_dump(data, allow_unicode=True)
|
||||
with open('monsters.yaml', 'w') as f:
|
||||
f.write(final)
|
||||
|
||||
# pprint.pprint(data['monsters'][0])
|
||||
|
||||
# print(data['monsters'][0].keys())
|
||||
|
||||
# for key, value in data['monsters'][0]:
|
||||
# print("Key:{}\tValue:{}".format(key,value))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
Loading…
Reference in New Issue