def main(students_file, rooms_file, out_format):
loader = ld.LoadJSON()
students = loader.load(filename=students_file)
rooms = loader.load(filename=rooms_file)
db = sql_functions.DBops()
db.create_table()
for query in sql_queries.INDEX_QUERY:
db.select_query(query)
db.insert_queries(rooms, students)
db.commit()
for select_num, query in enumerate(sql_queries.SELECT_QUERIES):
result = db.select_query(query)
try:
if out_format.lower() == 'json':
conversion_json = cd.JSONConversion()
conversion_json.write(
result, 'select_' + query_name(select_num) + out_format)
elif out_format.lower() == "xml":
conversion_xml = cd.XMLConversion()
conversion_xml.write(
result, 'select_' + query_name(select_num) + out_format)
else:
raise ex.FormatException('Please enter format json or xml')
except ex.FormatException as fe:
print(fe)
