示例#1
0
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))