def insert_time_to_recession(self): # Get Data Read hndl_Data_Read = EMF_DataSeries_Handle(self.hndl_DB, ticker='US_Rec_Ind') hndl_Data_Read.save_series_local() # Insert Time To hndl_Data_Wrte = EMF_DataSeries_Handle( self.hndl_DB, name='Months to Next Recession', ticker='US_TimeUntilRec', insertIfNot=True) data = hndl_Data_Read.get_series_values() dates = hndl_Data_Read.get_series_dates() trns_Data = transform_TimeToValue(data, {TIME_TO_VALUE: 1})[DATA_KEY] count = self.find_time_to_recession_limits(trns_Data) trns_Data = trns_Data[:-count] trns_Dates = dates[:-count] hndl_Data_Wrte.save_series_db(trns_Dates, trns_Data) hndl_Data_Wrte.periodicity = hndl_Data_Read.periodicity # Insert Time Since hndl_Data_Wrte = EMF_DataSeries_Handle( self.hndl_DB, name='Months since Last Recession', ticker='US_TimeSinceRec', insertIfNot=True) trns_Data = transform_TimeSinceValue(data, {TIME_SINCE_VALUE: 1})[DATA_KEY] count = self.find_time_since_recession_limits(trns_Data) trns_Data = trns_Data[count:] trns_Dates = dates[count:] hndl_Data_Wrte.save_series_db(trns_Dates, trns_Data) hndl_Data_Wrte.periodicity = hndl_Data_Read.periodicity
def insert_test_data(self, dates, values, tickers, periodicity=1, categorical=None): for (i, t) in enumerate(tickers): try: values = values[:, i] except IndexError: values = values hndl_Data = EMF_DataSeries_Handle(self.hndl_DB, name=t, ticker=t, insertIfNot=True) if categorical is not None: hndl_Data.is_categorical = categorical[i] hndl_Data.periodicity = periodicity hndl_Data.save_series_db(dates, values)