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
