From 666a84e004b011a0023c23aaa8acd5a6f43b5f80 Mon Sep 17 00:00:00 2001 From: robcarver17 Date: Wed, 27 Jan 2021 16:36:30 +0000 Subject: [PATCH] few minor bugs --- sysdata/mongodb/mongo_generic.py | 6 ++++++ sysdata/mongodb/mongo_historic_orders.py | 2 +- 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/sysdata/mongodb/mongo_generic.py b/sysdata/mongodb/mongo_generic.py index 806b2daae1..7c83003bc8 100644 --- a/sysdata/mongodb/mongo_generic.py +++ b/sysdata/mongodb/mongo_generic.py @@ -68,6 +68,12 @@ def get_list_of_keys(self)->list: return key_list + def get_list_of_values_for_dict_key(self, dict_key): + key_list = self.get_list_of_keys() + list_of_results = [self.get_result_dict_for_key(key) for key in key_list] + list_of_values = [item_dict.get(dict_key, None) for item_dict in list_of_results] + + return list_of_values def get_result_dict_for_key(self, key) ->dict: key_name = self.key_name diff --git a/sysdata/mongodb/mongo_historic_orders.py b/sysdata/mongodb/mongo_historic_orders.py index 37bbf122e7..1e45da2165 100644 --- a/sysdata/mongodb/mongo_historic_orders.py +++ b/sysdata/mongodb/mongo_historic_orders.py @@ -176,7 +176,7 @@ def get_list_of_order_ids_for_strategy_and_contract( def get_list_of_strategies(self): - list_of_keys = self.mongo_data.get_list_of_keys() + list_of_keys = self.mongo_data.get_list_of_values_for_dict_key("key") list_of_contract_strategies = [futuresContractStrategy.from_key(key) for key in list_of_keys] list_of_strategies = [futures_contract_strategy.strategy_name for futures_contract_strategy in list_of_contract_strategies]