class Company:


    def __init__(self, company_name, company_start_budget):
        self.company_name = company_name
        self.company_start_budget = int(company_start_budget)
        self.workers_list = []
        self.budget = self.company_start_budget

    def hire_worker(self):
        if self.outgoing(Worker.worker_cost):
            print(f'\nZatrudniasz pracownika firmy {self.company_name}')
            wor = Worker()
            self.workers_list.append((wor.worker_name, wor.worker_surname))


    def fire_worker(self):

        if len(self.workers_list) != 0:
            self.show_workers_list()
            user_select = int(input('Podaj nr pracownika, którego chcesz zwolnić: '))

            while user_select - 1 not in range(len(self.workers_list)):
                user_select = int(input('Będny numer. Spróbuj ponownie: '))

            else:
                print(f'Zwolniono: {self.workers_list[user_select - 1][0]} {self.workers_list[user_select - 1][1]}')
                del self.workers_list[user_select - 1]

        else:
            print('Brak pracowników w firmie.')

    def workers_amount(self):
        print(len(self.workers_list))

    def show_workers_list(self):

        if len(self.workers_list) != 0:
            print(f'Lista pracowników firmy {self.company_name}:')
            i = 1
            for worker in self.workers_list:
                print(f'{i}. {worker[0]} {worker[1]}')
                i += 1

        else:
            print('Brak zatrudnionych pracowników')

    def income(self, income_amount):
        self.budget += int(income_amount)

    def outgoing(self, outgoing_amount):

        if int(outgoing_amount) > self.budget:
            print(f'Masz za mały budżet: {self.budget}')
            return False

        else:
            self.budget -= outgoing_amount
            return True

class Worker:

    worker_cost = 10

    def __init__(self):
        self.worker_name = input('Podaj imię pracownika: ')
        self.worker_surname = input('Podaj nazwisko pracownika: ')


    def work(self):
        print('to jest metoda czasu pracy')
        self.time = input('Podaj, jak długo pracownik ma pracować '
                          '(1s = 10 zł kosztu, 20 zł zysku) [s]: ')