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)