import sys import math def f0 (sal): return 'Bonus: 0 EUR' def f1 (sal): return 'Bonus: 1000 EUR' def f2 (sal): return "Bonus: " + str(math.floor(0.1 * sal)) + " EUR" def f3 (sal): return 'Bonus: %s EUR' % (math.floor(0.1 * sal) + 1000) func = { '0' : f0, '1' : f1, '2' : f2, '3' : f3 } category = sys.argv[1] salary = sys.argv[2] print(category, salary, "EUR") try: bonus = (func[str(category)])(int(salary)) print(bonus) except Exception as e: print("INVALID\nBonus: 0 EUR")