Registrar as entrada e saída

import csv
import datetime

escolha=True
while escolha:
    print("\n")
    print("Registrar as entrada e saída")
    print("""
    1.\tRegistrar Entrada \n
    2.\tRegistrar Saída \n
    3.\tSair do Programa
    """)
    escolha= input("Escolha uma opção:  ")
    if escolha=="1":
        nome = input("Digite o nome do funcionário: ")
        entry_time = datetime.datetime.now()
        with open("a_registar.csv", mode="a", newline="") as file:
            writer = csv.writer(file)
            writer.writerow([nome, entry_time, ""])
    elif escolha == "2":
        nome = input("Digite o nome do funcionário: ")
        exit_time = datetime.datetime.now()
        with open("a_registar.csv", mode="r") as file:
            reader = csv.reader(file)
            rows = list(reader)
            for row in rows:
                if row[0] == nome:
                    if row[2] != "":
                        print(f"{nome} já registrou sua saída.")
                    row[2] = exit_time
                    break
            else:
                print(f"{nome} não encontrado.")
        with open("a_registar.csv", mode="w", newline="") as file:
            writer = csv.writer(file)
            writer.writerows(rows)
    elif escolha=="3":
      print("\n Fim do Programa")
      escolha = None
    else:
       print("\n Escolha não válida.\n Tente outra vez.")
Mundo em Python

Registrar as entrada e saída
