@@ -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
@@ -351,22 +350,19 @@ def method_filter(method):
351350 descriptor , method .descriptor
352351 )
353352
354- dex_list = range (self ._number_of_dex )
355353 filtered_methods = list ()
356354
357355 if class_name != ".*" :
358- for dex_index in dex_list :
359- method_dict = self ._get_methods_classified ()
356+ method_dict = self ._get_methods_classified ()
357+ filtered_methods += list (
358+ filter (method_filter , method_dict [class_name ])
359+ )
360+ else :
361+ method_dict = self ._get_methods_classified ()
362+ for key_name in method_dict :
360363 filtered_methods += list (
361- filter (method_filter , method_dict [class_name ])
364+ filter (method_filter , method_dict [key_name ])
362365 )
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- )
370366
371367 return filtered_methods
372368
0 commit comments