Papir 0.5.1

import datetime
import sys

def print_menu():
    print (30 * "-" , "MENU", 30 * "-")
    print ()
    print(24 * "-" ,datetime.datetime.now().strftime('%Y-%m-%d %H:%M'), 24 * "-" ,)
    print ("1. Dovezený papír ")
    print ("2. Odneseno kusu ")
    print ("3. Ve skladu ")
    print ("4. Objednej papír ")
    print ("5. Konec")
    print (67 * "-")

def dovezeno_Kusu():
   try:
      f = open ("zasoba.txt", "r+")
      sklad = f.read()
      pocetVeSkladu = int (sklad)
      dovezeno = int (input ("Kolik bylo dovezeno papíru? \n"))
      pocetVeSkladu += dovezeno
      if pocetVeSkladu == 1:
          print ("Ve skladu máme:" ,pocetVeSkladu," krabici.")
      elif 1 < pocetVeSkladu <= 4:
          print ("Ve skladu máme:" ,pocetVeSkladu," krabice.")
      else:
          print ("Ve skladu máme:" ,pocetVeSkladu," krabic.")
      print ( pocetVeSkladu,  file = open("zasoba.txt", "w"))

   finally:
      f.close()

def odneseno_Kusu():
      try:
         f = open ("zasoba.txt", "r+")
         sklad = f.read()
         pocetVeSkladu = int (sklad)
         odvezeno = int (input ("Kolik bylo odvezeno krabic papíru? \n"))
         pocetVeSkladu -= odvezeno
         if pocetVeSkladu <=0:
             print ("Ve skladu nezbyla žádná krabice papíru, předej info recepční!")
             pocetVeSkladu = 0
         else:
             if pocetVeSkladu == 1:
                  print ("Ve skladu máme:" ,pocetVeSkladu," krabici.")
             elif 1 < pocetVeSkladu <= 4:
                  print ("Ve skladu máme:" ,pocetVeSkladu," krabice.")
             else:
                  print ("Ve skladu máme:" ,pocetVeSkladu," krabic.")
         print ( pocetVeSkladu,  file = open("zasoba.txt", "w"))
      finally:
         f.close()


def veSkladu():
   with open("zasoba.txt", 'r+') as f:
        sklad = f.read()
        pocetVeSkladu = int(sklad)
        if pocetVeSkladu == 1:
          print ("Ve skladu máme:" ,pocetVeSkladu," krabici.")
        elif 1 < pocetVeSkladu <= 4:
          print ("Ve skladu máme:" ,pocetVeSkladu," krabice.")
        else:
          print ("Ve skladu máme:" ,pocetVeSkladu," krabic.")
        return pocetVeSkladu

def konec():
    konec = input ("Chceš opravdu ukončit program? A/N \n")

    if ( konec == "a" ) or ( konec == "A" ):
       sys.exit()

    elif (konec == "n" )or ( konec == "N"):
         print ("Přesun zpět do menu.")

    else:
         print ("Neplatná volba --> zpátky do menu.")

    loop= False

loop=True

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

       if choice == "1":
          dovezeno_Kusu()

       elif choice == "2":
            odneseno_Kusu()

       elif choice == "3":
            veSkladu()

       elif choice == "4":
           print ("Menu 4 bylo vybráno")

       elif choice == "5":
            konec()

       else:
            print ("Neplatná volba. Zkus to znovu...")