Язык python Напиши программу подсчёта зарплаты тренера

Тренер спортивного комплекса может работать на полную занятость или гибкое количество часов.

Базовая оплата тренера с полной занятостью 20 000, но есть система повышающих коэффициентов:

1) Если опыт работы от 3 лет, то зарплата увеличивается в 1,5 раза.

2) Если опыт от 5 лет, то зарплата увеличивается в 2 раза.

3) Если опыт от 7 лет, то зарплата увеличивается в 3 раза.

При почасовой оплате стоимость одного часа (800) умножается на количество отработанных часов.

Итог:

Фамилия тренера:

>>>Григорьева

Занятость(1 - полная, 2 - почасовая):

>>> 1

Стаж в годах:

>>>12

Григорьева - 60000

Объяснение:

name = str(input("Введите имя и фамилию тренера:  "))

rabota = int(input("Какой тип работы ( 1 - Полная или 2 - Почасовая, введите цифру):  "))

stag = float(input("Какой cтаж работы? Введите:   "))

baza = 20000

def rabota1():

   if stag >= 7:

      zarplata = 3 * baza

   elif stag >= 5 and stag <= 7:

      zarplata = 2 * baza

   else:

      zarplata = baza * 1.5

   print(f"Здраствуйте, {name} ,ваший стаж - {stag} лет, вы заработали - {zarplata}")

   return

   pass

def rabota2():

   time = float(input("Сколько часов отработано? Введите значение:  "))

   plata = time * 800

   print(f"Здраствуйте,{name}, вы заработали - {plata}")

if rabota == 1:

   rabota1()

else:

   rabota2()