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