few minor bugs

anggs1996 · Jan 27, 2021 · 666a84e · 666a84e
1 parent cd68197
commit 666a84e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git 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
@@ -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]