Add comply with pkg to kato

Author: dtmilano

.idea/runConfigurations/screenshot_box_100_300_300_600__tmp_box_png.xml

@@ -14,13 +14,20 @@ body = {'checkable': False, 'checked': False, 'clazz': 'android.widget.ImageButt
         'desc': 'add', 'index': None, 'instance': None, 'pkg': 'com.google.android.calculator',
         'res': 'com.google.android.calculator:id/op_add', 'scrollable': False, 'text': None}
 
+# This will restrict the search for candidates to the same pkg: 'com.google.android.calculator'
+helper.kato.comply_with_pkg = True
+
 try:
     obj_ref = helper.ui_device.find_object(body=body)
     print(helper.ui_object2.dump(oid=obj_ref.oid).desc)
 except Exception as e:
     print(f"I couldn't find any view matching selector={body}")
-    print(f'But found other {len(helper.kato.selectors)} potential ones.')
-    print('Choosing a closest selector from such potential list:')
-    body = next(iter(helper.kato.distances.values()))
-    obj_ref = helper.ui_device.find_object(body=body)
-    print(helper.ui_object2.dump(oid=obj_ref.oid))
+    n = len(helper.kato.selectors)
+    if n:
+        print(f'But found other {n} potential ones.')
+        print('Choosing a closest selector from such potential list:')
+        body = next(iter(helper.kato.distances.values()))
+        obj_ref = helper.ui_device.find_object(body=body)
+        print(helper.ui_object2.dump(oid=obj_ref.oid))
+    else:
+        print('There are no potential ones either.')

examples/helper/find-object

@@ -39,6 +39,8 @@ def __init__(self):
         self.enabled = False
         self.selectors = []
         self.distances = OrderedDict()
+        # If pkg has been specified in the selector, only finds other selectors with the same pkg when True
+        self.comply_with_pkg = False
 
 
 def kato(func):
@@ -90,6 +92,8 @@ def find_me_the_selectors(e: ApiException, *args, **kwargs):
             selector = Selector(**kwargs['body'])
             helper.kato.selectors = window_hierarchy_to_selector_list(
                 helper.ui_device.dump_window_hierarchy(_format='JSON'))
+            if selector.pkg and helper.kato.comply_with_pkg:
+                helper.kato.selectors = list(filter(lambda _s: _s.pkg == selector.pkg, helper.kato.selectors))
             _d = dict()
             for n, s in enumerate(helper.kato.selectors):
                 if n == 0: