classes.py

import random
import const

random.seed(0)


# All values should be in $ amounts for simplicity
class Negotiable:

    def __init__(self):
        self.salary = 0.0
        self.retirement = 0.0
        self.benefits = 0.0
        self.pto = 0.0
        self.stock_options = 0.0
        self.signing_bonus = 0.0
        self.other = 0.0

    # Get all variables as a list for comparison
    def vars_list(self):
        return [self.salary, self.retirement, self.benefits, self.pto, self.stock_options, self.signing_bonus,
                self.other]

    def total(self):
        return round(sum(self.vars_list()), 2)

    def increase_by_pct(self, pct):
        self.salary *= 1+pct
        self.retirement *= 1+pct
        self.benefits *= 1+pct
        self.pto *= 1+pct
        self.stock_options *= 1+pct
        self.signing_bonus *= 1+pct
        self.other *= 1+pct
        return self

    def __eq__(self, other):
        return self.vars_list() == other.vars_list()

    def __lt__(self, other):
        return self.total() < other.total()

    def __repr__(self):
        return '<Offer value: ${:.2f}>'.format(self.total())
        # return 'Salary: {:.2f}, Pto: {:.2f}, Retirement: {:.2f}, Benefits: {:.2f}, Stock: {:.2f}, Signing: {:.2f}, ' \
        #       'Other: {:.2f}'.format(self.salary, self.pto, self.retirement, self.benefits, self.stock_options,
        #                              self.signing_bonus, self.other)

    def __str__(self):
        return self.__repr__()


class Offer(Negotiable):

    def __init__(self):
        super(Offer, self).__init__()
        self.sender_is_company = True
        self.candidate = 0
        self.company = 0
        self.action = const.Action
        self.expires_at = -1

    @staticmethod
    def change_action(offer, new_action):
        my_offer = Offer()
        my_offer.action = new_action
        my_offer.candidate = offer.candidate
        my_offer.company = offer.company
        my_offer.salary = offer.salary
        my_offer.retirement = offer.retirement
        my_offer.benefits = offer.benefits
        my_offer.other = offer.other
        return my_offer