@@ -307,9 +307,8 @@ def all_methods(self) -> Set[MethodObject]:
307307 :return: a set of MethodObjects
308308 """
309309 method_set = set ()
310- for dex_index in range (self ._number_of_dex ):
311- for method_list in self ._get_methods_classified ().values ():
312- method_set .update (method_list )
310+ for method_list in self ._get_methods_classified ().values ():
311+ method_set .update (method_list )
313312
314313 return method_set
315314
@@ -355,18 +354,16 @@ def method_filter(method):
355354 filtered_methods = list ()
356355
357356 if class_name != ".*" :
358- for dex_index in dex_list :
359- method_dict = self ._get_methods_classified ()
357+ method_dict = self ._get_methods_classified ()
358+ filtered_methods += list (
359+ filter (method_filter , method_dict [class_name ])
360+ )
361+ else :
362+ method_dict = self ._get_methods_classified ()
363+ for key_name in method_dict :
360364 filtered_methods += list (
361- filter (method_filter , method_dict [class_name ])
365+ filter (method_filter , method_dict [key_name ])
362366 )
363- else :
364- for dex_index in dex_list :
365- method_dict = self ._get_methods_classified ()
366- for key_name in method_dict :
367- filtered_methods += list (
368- filter (method_filter , method_dict [key_name ])
369- )
370367
371368 return filtered_methods
372369
0 commit comments