fixed mismatched keys + renamed sql import file

This commit is contained in:
Djalim Simaila 2023-03-11 22:39:05 +01:00
parent a8048d8366
commit d07a61c2a0
2 changed files with 13 additions and 3 deletions

View File

@ -84,7 +84,7 @@ def bulk_import(data):
TRAIT_LIST.append(new_trait) TRAIT_LIST.append(new_trait)
# Npcs # Npcs
npcs = data["npc"] npcs = data["npcs"]
for npc in npcs.values(): for npc in npcs.values():
new_npc = Npc(0, getid()) new_npc = Npc(0, getid())
@ -92,10 +92,13 @@ def bulk_import(data):
for lang in npc["name"]: for lang in npc["name"]:
LID_LIST.append(Locale(new_npc.NAME_LID, lang, npc["name"][lang])) LID_LIST.append(Locale(new_npc.NAME_LID, lang, npc["name"][lang]))
# TODO handle reactions
"""
for reaction in npc["reactions"]: for reaction in npc["reactions"]:
if reaction in list(trait_names.keys()): if reaction in list(trait_names.keys()):
new_reaction = Reaction(0,new_npc.NPC_ID, trait_names[reaction]) new_reaction = Reaction(0,new_npc.NPC_ID, trait_names[reaction])
REACTION_LIST.append(new_reaction) REACTION_LIST.append(new_reaction) """
for question_type in npc["answers"]: for question_type in npc["answers"]:
question_type_id = 0 if question_type == "where" else 1 question_type_id = 0 if question_type == "where" else 1
@ -121,32 +124,39 @@ def bulk_import(data):
print("Locale : "+str(text)) print("Locale : "+str(text))
session.add(text) session.add(text)
session.commit() session.commit()
for question in QUESTIONS_LIST: for question in QUESTIONS_LIST:
print("Question : "+str(question)) print("Question : "+str(question))
session.add(question) session.add(question)
session.commit() session.commit()
for trait in TRAIT_LIST: for trait in TRAIT_LIST:
print("Trait : "+ str(trait)) print("Trait : "+ str(trait))
session.add(trait) session.add(trait)
session.commit() session.commit()
for npc in NPC_LIST: for npc in NPC_LIST:
print("Npc : "+ str(npc)) print("Npc : "+ str(npc))
session.add(npc) session.add(npc)
session.commit() session.commit()
for reaction in REACTION_LIST: for reaction in REACTION_LIST:
print("Reaction : " + str(reaction)) print("Reaction : " + str(reaction))
session.add(reaction) session.add(reaction)
session.commit() session.commit()
for answer in ANSWER_LIST: for answer in ANSWER_LIST:
print("Answer : "+ str(answer)) print("Answer : "+ str(answer))
session.add(answer) session.add(answer)
session.commit() session.commit()
for room in ROOMS_LIST: for room in ROOMS_LIST:
print("Room : "+str(room)) print("Room : "+str(room))
session.add(room) session.add(room)
session.commit() session.commit()
file = open("bulkData.yml", "r")
file = open("bulk_data.yml", "r")
bulk_import(yaml.load(file, yaml.Loader)) bulk_import(yaml.load(file, yaml.Loader))