if number == '1': q1 = """SELECT company_name, replace(main_color, '#', '') as short_hex_color FROM project WHERE main_color IS NOT NULL;""" order = "q1" Imaging.create_image(Engine.engine_main(q1, order), 'pic1.png') exit() if number == '2': q2 = """SELECT name, budget_value, budget_currency, REPLACE(main_color, '#', '') AS main_color FROM project WHERE name IS NOT NULL AND main_color IS NOT NULL;""" order = "q4" normalised_incoming_datalist = Normalisation.normalize_data_q2(Engine.engine_main(q2, order)) Imaging.create_image(Engine.getsize(normalised_incoming_datalist)[:20], 'pic2.png') exit() if number == '3': q3 = """SELECT name, duedate, replace(main_color, '#', '') AS main_color FROM project WHERE company_name = 'Eadel' AND name IS NOT NULL ORDER BY duedate DESC;""" order = "" normalised_incoming_datalist = Normalisation.normalize_data_q3(Engine.engine_main(q3, order)) Imaging.create_image(Engine.getsize(normalised_incoming_datalist), 'pic3.png') exit() if number == '4': q4 = """SELECT company_hq, CAST(budget_value AS float), replace(main_color, '#', '') AS main_color FROM project