Witajcie,
mam drobny problem z klasami i metodami.
Mógłby mi ktoś podpowiedzieć jak "sprytnie" dodać 3% z wypłaty na konto oszczędnościowe bez dublowania pytania o wynagrodzenie i przepracowane godziny?
class Employee():
def __init__(self, name, ID, departament, title, h_sal=0, h_week=0):
self.name = name
self.ID = ID
self.departament = departament
self.title = title
self.h_sal = h_sal
self.h_week = h_week
def salary(self, week_salary=0):
h_sal = int(input("\nHow much PLN do you earn per hour: "))
h_week = int(input("How many hours do you work per week: "))
if h_week > 35:
extra = (h_week - 35) * h_sal * 1.5
week_salary = h_sal * 35 + extra
return week_salary
else:
week_salary = h_sal * h_week
return week_salary
def pension(self):
print("\nyou can save 3% of your salary on your pension Account")
answer = input("Are you interested? YES/NO: ")
if answer == "YES":
return self.salary()*0.03
else:
return "OK, no problem!"
employee1 = Employee("Damian", 1, "IT", "Python Developer")
print("""
0 - exit
1 - check your salary
2 - save money on pension account""")
a = ""
while a!=0:
a = int(input("\nWhich option would you like to choose: "))
if a == 1:
print("Salary: ",employee1.salary())
elif a == 2:
print("Your salary:",employee1.salary(), "\nYou have saved: ",employee1.pension())
else:
print("I dont understand")
break