DjalimSimaila/SAE-A2-TruthInquiry
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add input file argument for import script

Browse Source
This commit is contained in:
Thomas Rubini 2023-03-19 09:49:10 +01:00
parent 2d0c797b84
commit 9fcd7c159a
No known key found for this signature in database
GPG Key ID: C7D287C8C1CAC373
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
bulk_import.py
View File

@ -1,8 +1,9 @@
import argparse
import yaml
from dotenv import load_dotenv from dotenv import load_dotenv
load_dotenv() load_dotenv()
import sys
import yaml
from sqlalchemy.orm import sessionmaker from sqlalchemy.orm import sessionmaker
from truthinquiry.ext.database.models import * from truthinquiry.ext.database.models import *
from truthinquiry.ext.database.sa import engine from truthinquiry.ext.database.sa import engine
@ -126,7 +127,7 @@ def bulk_import(data):
for lid in lm.get_used_lids(): for lid in lm.get_used_lids():
print("lid :"+ str(lid)) print("lid :"+ str(lid))
session.add(Locale(lid)); session.add(Locale(lid))
for text in TEXT_LIST: for text in TEXT_LIST:
print("Text : "+str(text)) print("Text : "+str(text))
@ -163,6 +164,8 @@ def bulk_import(data):
session.add(room) session.add(room)
session.commit() session.commit()
file = open("bulk_data.yml", "r") if len(sys.argv) <= 1:
print("Please enter input file")
bulk_import(yaml.load(file, yaml.Loader)) else:
file = open(sys.argv[1], "r")
bulk_import(yaml.load(file, yaml.Loader))