@@ -52,7 +52,7 @@ def __init__(self, stack_trace=False, queue_trace=False):
5252 self .queue_ref = self ._queue_init ()
5353
5454 def find_key (self , data , key ):
55- # type: (Union[dict, list], str) -> list
55+ # type: (Union[dict, list, OrderedDict ], str) -> list
5656 if not self ._valid_key_input (data , key ):
5757 raise
5858
@@ -80,7 +80,7 @@ def find_key(self, data, key):
8080 return value_list
8181
8282 def find_keys (self , data , keys , group = True ):
83- # type: (Union[dict, list], list, bool) -> list
83+ # type: (Union[dict, list, OrderedDict ], list, bool) -> list
8484 if not self ._valid_keys_input (data , keys , group ):
8585 raise
8686
@@ -110,7 +110,7 @@ def find_keys(self, data, keys, group=True):
110110 return value_list
111111
112112 def find_key_chain (self , data , keys ):
113- # type: (Union[dict, list], list) -> list
113+ # type: (Union[dict, list, OrderedDict ], list) -> list
114114 if not self ._valid_key_chain_input (data , keys ):
115115 raise
116116
@@ -148,7 +148,7 @@ def find_key_chain(self, data, keys):
148148 return self .queue_ref
149149
150150 def find_key_value (self , data , key , value ):
151- # type: (Union[dict, list], str, Union[str, int, float, bool, None]) -> list
151+ # type: (Union[dict, list, OrderedDict ], str, Union[str, int, float, bool, None]) -> list
152152 if not self ._valid_key_value_input (data , key , value ):
153153 raise
154154
@@ -174,7 +174,7 @@ def find_key_value(self, data, key, value):
174174 return value_list
175175
176176 def find_value (self , data , value ):
177- # type: (Union[dict, list], Union[str, int, float, bool, None]) -> list
177+ # type: (Union[dict, list, OrderedDict ], Union[str, int, float, bool, None]) -> list
178178 if not self ._valid_value_input (data , value ):
179179 raise
180180
@@ -208,18 +208,18 @@ def _stack_init(self):
208208 return stack
209209
210210 def _stack_push (self , elem ):
211- # type: (Union[dict, list]) -> None
211+ # type: (Union[dict, list, OrderedDict ]) -> None
212212 self .stack_ref .append (elem )
213213
214214 def _stack_pop (self ):
215- # type: () -> Union[dict, list]
215+ # type: () -> Union[dict, list, OrderedDict ]
216216 try :
217217 return self .stack_ref .pop ()
218218 except IndexError :
219219 raise
220220
221221 def _stack_peak (self ):
222- # type: () -> Union[dict, list]
222+ # type: () -> Union[dict, list, OrderedDict ]
223223 try :
224224 return self .stack_ref [- 1 :][0 ]
225225 except IndexError :
@@ -242,7 +242,7 @@ def _stack_push_list_elem(self, elem):
242242 self ._stack_trace ()
243243
244244 def _stack_all_key_values_in_dict (self , key , elem ):
245- # type: (str, dict) -> list
245+ # type: (str, Union[ dict, OrderedDict] ) -> list
246246 value_list = []
247247
248248 if not isinstance (elem , (dict , OrderedDict )):
@@ -262,7 +262,7 @@ def _stack_all_key_values_in_dict(self, key, elem):
262262 return value_list
263263
264264 def _stack_all_keys_values_in_dict (self , keys , elem ):
265- # type: (list, dict) -> list
265+ # type: (list, Union[ dict, OrderedDict] ) -> list
266266 value_list = []
267267
268268 if not isinstance (elem , (dict , OrderedDict )):
@@ -285,7 +285,7 @@ def _stack_all_keys_values_in_dict(self, keys, elem):
285285 return value_list
286286
287287 def _stack_all_key_and_value_in_dict (self , key , value , elem ):
288- # type: (str, Union[str, int, float, bool, None], dict) -> bool
288+ # type: (str, Union[str, int, float, bool, None], Union[ dict, OrderedDict] ) -> bool
289289 if not isinstance (elem , (dict , OrderedDict )):
290290 raise TypeError
291291 elif type (key ) is not str :
@@ -305,7 +305,7 @@ def _stack_all_key_and_value_in_dict(self, key, value, elem):
305305 return False
306306
307307 def _stack_all_value_in_dict (self , value , elem ):
308- # type: (Union[str, int, float, bool, None], dict) -> str
308+ # type: (Union[str, int, float, bool, None], Union[ dict, OrderedDict] ) -> str
309309 if not isinstance (elem , (dict , OrderedDict )):
310310 raise TypeError
311311 elif not isinstance (value , (str , int , float , bool , type (None ))):
@@ -339,18 +339,18 @@ def _queue_init(self):
339339 return queue
340340
341341 def _queue_push (self , elem ):
342- # type: (Union[dict, list]) -> None
342+ # type: (Union[dict, list, OrderedDict ]) -> None
343343 self .queue_ref .append (elem )
344344
345345 def _queue_pop (self ):
346- # type: () -> Union[dict, list]
346+ # type: () -> Union[dict, list, OrderedDict ]
347347 try :
348348 return self .queue_ref .pop (0 )
349349 except IndexError :
350350 raise
351351
352352 def _queue_peak (self ):
353- # type: () -> Union[dict, list]
353+ # type: () -> Union[dict, list, OrderedDict ]
354354 try :
355355 return self .queue_ref [0 ]
356356 except IndexError :
@@ -373,7 +373,7 @@ def _queue_push_list_elem(self, elem):
373373 self ._queue_trace ()
374374
375375 def _queue_all_key_values_in_dict (self , key , elem ):
376- # type: (str, dict) -> bool
376+ # type: (str, Union[ dict, OrderedDict] ) -> bool
377377 found = False
378378 if not isinstance (elem , (dict , OrderedDict )):
379379 raise TypeError
0 commit comments