import sys
from datetime import datetime
import os
import math
from termcolor import colored, cprint

while True:
    now = datetime.now()
    menu = ("1) výpočet věku \n"
            "2) rozdíl mezi roky\n"
            "3) počet dní od data\n"
            "4) konec\n")
    print(colored(menu,color="light_green"))
    print(colored(now.date(),color="blue"))
    volba = input("Vyberte jednu možností: ")
    datum_format = "%d.%m.%Y"
    if volba == "1":
        datum_narozeni = input("Zadejte datum narozeni "
                               "ve formátu dd.mm.yyyy: ")
        try:
            date = datetime.strptime(datum_narozeni,datum_format)
            vek = now - date
            print(f"Váš věk je: {math.floor(vek.days/365.25)}")
            input("Pro pokračovní stiskněte Enter...")
        except ValueError:
            print("Byla zadáno neplatné datum.")

        pass
    elif volba == "2":
        prvni_datum = input(colored("Zadejte první datum: ",color="red"))
        druhe_datum = input("Zadejte druhé datum: ")
        try:
            prvni_datum = datetime.strptime(prvni_datum,datum_format)
            druhe_datum = datetime.strptime(druhe_datum,datum_format)

            if prvni_datum < druhe_datum:
                rozdil = druhe_datum-prvni_datum
            else:
                rozdil = prvni_datum-druhe_datum
            print(colored(f"Rozdíl mezi daty je {math.floor(rozdil.days)} dní.",color="yellow"))
            input("Pro pokračovní stiskněte Enter...")
        except ValueError:
            print(colored("Bylo zadáno neplatné datum!",color="red"))
        pass
    elif volba == "3":
        datum = input("Zadejte druhé datum: ")
        try:
            datum = datetime.strptime(datum, datum_format)
            rozdil = now - datum
            print(f"Rozdíl datumů je {rozdil.days} dní.\n Rozdíl v letech: {math.floor(rozdil.days/365.25)}")
            input("Pro pokračování stistkně Enter...")
        except ValueError:
            print(colored("Bylo zadáno neplatné datum!", color="red"))
        pass
    elif volba == "4":
        input("Opravdu chcete skončit?")
        break
        pass
    else:
        print(colored("Neplatná volba!",color="red"))