Esempio n. 1
0
def fill_workers_db():
    workers = [
        Worker("Jan", "Kowalski"),
        Worker("Anna", "Nowak"),
        Worker("Tomasz", "Armstrong"),
        Worker("Zuzanna", "Mozart"),
        Worker("Antoni", "Skłodowski")
    ]
    connection = sql.connect(database_name)
    for i in range(len(workers)):
        cursor = connection.cursor()
        w = workers[i]
        cursor.execute("INSERT INTO workers_database VALUES (?,?,?,?)",
                       (w.number, w.name, w.surname, w.card))
        connection.commit()
    connection.close()
def get_workers_array():
    connection = sql.connect(database_name)
    cursor = connection.cursor()
    cursor.execute("SELECT * FROM workers_database")
    emp = cursor.fetchall()
    output = []
    for worker in emp:
        output.append(Worker(worker[1], worker[2], worker[0], calc_card_num(worker[3])))
    return output
Esempio n. 3
0
def print_workers():
    connection = sql.connect(used_database)
    cursor = connection.cursor()
    cursor.execute("SELECT * FROM workers_database")
    emp = cursor.fetchall()
    w_arr = []
    for worker in emp:
        w_arr.append(Worker(worker[1], worker[2], worker[0], worker[3]))
    for w in w_arr:
        print(str(w))
Esempio n. 4
0
def findWorker(workerId):
    connection = sql.connect(used_database)
    cursor = connection.cursor()
    cursor.execute("SELECT * FROM workers_database")
    emp = cursor.fetchall()
    for worker in emp:
        if int(worker[0]) == workerId:
            worker_name = worker[1]
            worker_surname = worker[2]
            worker_id = int(worker[0])
            card_id = calc_card_num(worker[3])
            return Worker(worker_name, worker_surname, worker_id, card_id)
    return None
def add_worker():
    print("__Adding worker__")

    check = True
    while check:
        name = input("Input the name: ")
        surname = input("Input the surname: ")
        inp = 'O'
        while inp != 'N' and inp != 'Y':
            inp = input(f"Is data: {name} {surname} correct?[Y/N]")
            if inp == 'Y':
                check = False
                worker = Worker(name, surname)
                add_worker_to_db(worker)
                print(f"{worker} successfully added to database")
            elif inp == 'N':
                print("Data incorrect, input again...")
            else:
                print("Please insert (Y)es or (N)o")