add_contract.py

def add_contract(self, contract):
    # Adds a new contract to the forecast
    
    self.df[contract.contract_name] = np.zeros(self.number_of_months)
    self.contracts.append(contract)
    
    for m in range(contract.start_month, contract.start_month + contract.term, contract.payment_frequency):
        try:
            increase = (contract.increase_percentage * int(m / contract.increase_period) / 100) * contract.value 
            self.df.loc[self.months[m]][contract.contract_name] = contract.value + increase
            self.df.loc[self.months[m]]['live_contracts'] += 1
        except IndexError:
            pass