from QueryEngine import QueryEngine from flask import flash qe = QueryEngine() qe.setup_default() def login_check(username, password): qe.connect() check_acc = qe.do_query("select count(*) from log_in where UserName = '******' and Password_hash = md5('" + password + "')") count = check_acc[0][0] qe.disconnect() if count == 1: #print("I am in") return True else: #print("wrong account") return False def account_type(username, password): login_id = get_login(username, password) qe.connect() patient_type = qe.do_query( "select count(*) from patient where Patient_ID = " + str(login_id)) doctor_type = qe.do_query( "select count(*) from doctor where Doctor_ID = " + str(login_id)) staff_type = qe.do_query("select count(*) from staff where Staff_ID = " + str(login_id))