def _get_config_data_object(self): database_name = self._database_name host = self._host port = self._port data_object = mongoFuturesInstrumentData( mongoDb(database_name=database_name, host=host)) return data_object
def get_instrument_spec_from_mongo(instrument_code): mongo_instruments = mongoFuturesInstrumentData() mongo_roll_data = mongoRollParametersData() instrument_object = mongo_instruments.get_instrument_data(instrument_code) if instrument_object.empty(): raise Exception("Instrument %s missing from %s" % (instrument_code, mongo_instruments)) roll_parameters = mongo_roll_data.get_roll_parameters(instrument_code) if roll_parameters.empty(): raise Exception("Instrument %s missing from %s" % (instrument_code, mongo_roll_data)) return (instrument_object, roll_parameters)
def test_futures_instruments(self): data = mongoFuturesInstrumentData(database_name="test") # test db so okay to do this data._mongo.db.drop_collection(data._mongo.collection_name) codes = data.get_list_of_instruments() self.assertEqual(codes, []) instrument_object = data.get_instrument_data("EDOLLAR") self.assertTrue(instrument_object.empty()) instrument_object = futuresInstrument("EDOLLAR", some_data="test") data.add_instrument_data(instrument_object) self.assertEqual(data.get_list_of_instruments(), ["EDOLLAR"]) found_object = data.get_instrument_data("EDOLLAR") self.assertEqual(found_object.instrument_code, "EDOLLAR") found_object = data["EDOLLAR"] self.assertEqual(found_object.instrument_code, "EDOLLAR") self.assertEqual(found_object.meta_data["some_data"], "test") codes = data.get_list_of_instruments() self.assertEqual(codes, ["EDOLLAR"]) data.delete_instrument_data("EDOLLAR", are_you_sure=True) instrument_object = data.get_instrument_data("EDOLLAR") self.assertTrue(instrument_object.empty()) codes = data.get_list_of_instruments() self.assertEqual(codes, [])
""" Populate a mongo DB collection with instrument data from a csv """ from sysdata.mongodb.mongo_futures_instruments import mongoFuturesInstrumentData from sysdata.csv.csv_instrument_config import csvFuturesInstrumentData INSTRUMENT_CONFIG_PATH = "data.futures.csvconfig" data_out = mongoFuturesInstrumentData() data_in = csvFuturesInstrumentData(datapath=INSTRUMENT_CONFIG_PATH) print(data_in) instrument_list = data_in.get_list_of_instruments() for instrument_code in instrument_list: instrument_object = data_in.get_instrument_data(instrument_code) data_out.delete_instrument_data(instrument_code, are_you_sure=True) data_out.add_instrument_data(instrument_object) # check instrument_added = data_out.get_instrument_data(instrument_code) print("Added %s to %s" % (instrument_added.instrument_code, data_out))
def _get_config_data_object(self): data_object = mongoFuturesInstrumentData(self.mongo_db) return data_object
def _get_config_data_object(self): database_name = self._database_name data_object = mongoFuturesInstrumentData(database_name) return data_object