papirFB

import datetime
with open("zasoba.txt", 'r+') as f:
        sklad = f.read(1)
        pocetVeSkladu = int(sklad)

dovezeno = 0
odneseno = 0
def dovezeno_Kusu():
   return pocetVeSkladu + dovezeno

a = dovezeno_Kusu

def odneseno_Kusu():
    return a() - odneseno

def ve_Skladu ():
    return a()

b = ve_Skladu


def print_menu():
    print (30 * "-" , "MENU", 30 * "-")
    print ()
    print(24 * "-" ,datetime.datetime.now().strftime('%Y-%m-%d %H:%M'), 24 * "-" ,)
    print ("1. Dovezeny papir ")
    print ("2. Odneseno kusu ")
    print ("3. Ve skladu ")
    print ("4. Objednej papir ")
    print ("5. Odejit a ulozit?")
    print (67 * "-")

loop=True

while loop:
    print_menu()
    choice = input("Zadej volbu [1-5]: ")

    if choice == "1":
        dovezeno = int (input ("Kolik bylo dovezeno papiru? "))
        print ("Ve skladu mame:", dovezeno_Kusu() , "krabic.")


    elif choice == "2":
        odneseno = int (input("Kolik krabic bylo odneseno? "))
        print ("Ve skladu máme", odneseno_Kusu() , "krabic papiru.")

    elif choice == "3":
        print ("Ve skladu mame: ", ve_Skladu() , " krabic papiru. ")

    elif choice == "4":
        print ("Menu 5 has been selected")


    elif choice == "5":

        vyber = input ("Odejit a ulozit? A/N ")

        if vyber == "a":
                print ( b(),  file = open("zasoba.txt", "w"))
            #exit()

        elif vyber == "n":
                print_menu()

        else:
                input("Spatna volba. Zkus to znovu...")



        loop= False

    else:

        input("Neplatna volba. Zkus to znovu...")