WIP: Automatic script generator

pywinauto/application.py

@@ -550,32 +550,6 @@ def wait_not(self, wait_for_not, timeout=None, retry_interval=None):
         # None return value, since we are waiting for a `negative` state of the control.
         # Expect that you will have nothing to do with the window closed, disabled, etc.
 
-    def _ctrl_identifiers(self):
-
-        ctrls = self.__resolve_control(self.criteria)
-
-        if ctrls[-1].is_dialog():
-            # dialog controls are all the control on the dialog
-            dialog_controls = ctrls[-1].children()
-
-            ctrls_to_print = dialog_controls[:]
-            # filter out hidden controls
-            ctrls_to_print = [
-                ctrl for ctrl in ctrls_to_print if ctrl.is_visible()]
-        else:
-            dialog_controls = ctrls[-1].top_level_parent().children()
-            ctrls_to_print = [ctrls[-1]]
-
-        # build the list of disambiguated list of control names
-        name_control_map = findbestmatch.build_unique_dict(dialog_controls)
-
-        # swap it around so that we are mapped off the controls
-        control_name_map = {}
-        for name, ctrl in name_control_map.items():
-            control_name_map.setdefault(ctrl, []).append(name)
-
-        return control_name_map
-
     def print_control_identifiers(self, depth=None, filename=None):
         """
         Prints the 'identifiers'
@@ -597,20 +571,8 @@ def print_control_identifiers(self, depth=None, filename=None):
         # Create a list of this control and all its descendants
         all_ctrls = [this_ctrl, ] + this_ctrl.descendants()
 
-        # Create a list of all visible text controls
-        txt_ctrls = [ctrl for ctrl in all_ctrls if ctrl.can_be_label and ctrl.is_visible() and ctrl.window_text()]
-
-        # Build a dictionary of disambiguated list of control names
-        name_ctrl_id_map = findbestmatch.UniqueDict()
-        for index, ctrl in enumerate(all_ctrls):
-            ctrl_names = findbestmatch.get_control_names(ctrl, all_ctrls, txt_ctrls)
-            for name in ctrl_names:
-                name_ctrl_id_map[name] = index
-
-        # Swap it around so that we are mapped off the control indices
-        ctrl_id_name_map = {}
-        for name, index in name_ctrl_id_map.items():
-            ctrl_id_name_map.setdefault(index, []).append(name)
+        # Build unique control names map
+        ctrls_names = findbestmatch.build_names_list(all_ctrls)
 
         def print_identifiers(ctrls, current_depth=1, log_func=print):
             """Recursively print ids for ctrls and their descendants in a tree-like format"""
@@ -633,7 +595,7 @@ def print_identifiers(ctrls, current_depth=1, log_func=print):
                     "".format(class_name=ctrl.friendly_class_name(),
                               text=ctrl_text,
                               rect=ctrl.rectangle())
-                output += indent + u'{}'.format(ctrl_id_name_map[ctrl_id])
+                output += indent + u'{}'.format(ctrls_names[ctrl_id].to_list())
 
                 title = ctrl_text
                 class_name = ctrl.class_name()
@@ -1382,6 +1344,15 @@ def _process_get_modules_wmi():
     return modules
 
 
+#=========================================================================
+def get_process_command_line_wmi(process_id):
+    """Return full command line of process"""
+    from win32com.client import GetObject
+    _wmi = GetObject('winmgmts:')
+    proc = _wmi.ExecQuery("Select CommandLine from Win32_Process where ProcessId={}".format(process_id))[0]
+    return proc.CommandLine
+
+
 #=========================================================================
 def process_module(process_id):
     """Return the string module name of this process"""

pywinauto/controls/__init__.py

@@ -33,16 +33,14 @@
 
 from ..sysinfo import UIA_support
 if UIA_support:
-    from . import uiawrapper # register "uia" back-end (at the end of uiawrapper module)
+    from . import uiawrapper  # register "uia" back-end (at the end of uiawrapper module)
     from . import uia_controls
 
 from .hwndwrapper import get_dialog_props_from_handle
 from .hwndwrapper import InvalidWindowHandle
 
-# import the control classes - this will register the classes they
-# contain
+# import the control classes - this will register the classes they contain
 from . import common_controls
 from . import win32_controls
 
-
 from ..base_wrapper import InvalidElement

pywinauto/controls/uiawrapper.py

@@ -48,32 +48,6 @@
 from .. import uia_defines as uia_defs
 from ..uia_element_info import UIAElementInfo, elements_from_uia_array
 
-# region PATTERNS
-AutomationElement = IUIA().ui_automation_client.IUIAutomationElement
-DockPattern = IUIA().ui_automation_client.IUIAutomationDockPattern
-ExpandCollapsePattern = IUIA().ui_automation_client.IUIAutomationExpandCollapsePattern
-GridItemPattern = IUIA().ui_automation_client.IUIAutomationGridItemPattern
-GridPattern = IUIA().ui_automation_client.IUIAutomationGridPattern
-InvokePattern = IUIA().ui_automation_client.IUIAutomationInvokePattern
-ItemContainerPattern = IUIA().ui_automation_client.IUIAutomationItemContainerPattern
-LegacyIAccessiblePattern = IUIA().ui_automation_client.IUIAutomationLegacyIAccessiblePattern
-MultipleViewPattern = IUIA().ui_automation_client.IUIAutomationMultipleViewPattern
-RangeValuePattern = IUIA().ui_automation_client.IUIAutomationRangeValuePattern
-ScrollItemPattern = IUIA().ui_automation_client.IUIAutomationScrollItemPattern
-ScrollPattern = IUIA().ui_automation_client.IUIAutomationScrollPattern
-SelectionItemPattern = IUIA().ui_automation_client.IUIAutomationSelectionItemPattern
-SelectionPattern = IUIA().ui_automation_client.IUIAutomationSelectionPattern
-SynchronizedInputPattern = IUIA().ui_automation_client.IUIAutomationSynchronizedInputPattern
-TableItemPattern = IUIA().ui_automation_client.IUIAutomationTableItemPattern
-TablePattern = IUIA().ui_automation_client.IUIAutomationTablePattern
-TextPattern = IUIA().ui_automation_client.IUIAutomationTextPattern
-TogglePattern = IUIA().ui_automation_client.IUIAutomationTogglePattern
-TransformPattern = IUIA().ui_automation_client.IUIAutomationTransformPattern
-ValuePattern = IUIA().ui_automation_client.IUIAutomationValuePattern
-VirtualizedItemPattern = IUIA().ui_automation_client.IUIAutomationVirtualizedItemPattern
-WindowPattern = IUIA().ui_automation_client.IUIAutomationWindowPattern
-# endregion
-
 # =========================================================================
 _friendly_classes = {
     'Custom': None,
@@ -335,6 +309,13 @@ def iface_virtualized_item(self):
         elem = self.element_info.element
         return uia_defs.get_elem_interface(elem, "VirtualizedItem")
 
+    # ------------------------------------------------------------
+    @lazy_property
+    def iface_legacy_accessible(self):
+        """Get the element's LegacyIAccessible interface pattern"""
+        elem = self.element_info.element
+        return uia_defs.get_elem_interface(elem, "LegacyIAccessible")
+
     # ------------------------------------------------------------
     @property
     def writable_props(self):
@@ -349,15 +330,11 @@ def writable_props(self):
     # ------------------------------------------------------------
     def legacy_properties(self):
         """Get the element's LegacyIAccessible control pattern interface properties"""
-        elem = self.element_info.element
-        impl = uia_defs.get_elem_interface(elem, "LegacyIAccessible")
+        impl = self.iface_legacy_accessible
         property_name_identifier = 'Current'
 
-        interface_properties = [prop for prop in dir(LegacyIAccessiblePattern)
-                                if (isinstance(getattr(LegacyIAccessiblePattern, prop), property)
-                                and property_name_identifier in prop)]
-
-        return {prop.replace(property_name_identifier, '') : getattr(impl, prop) for prop in interface_properties}
+        interface_properties = [prop for prop in dir(impl) if prop.startswith(property_name_identifier)]
+        return {prop.replace(property_name_identifier, ''): getattr(impl, prop) for prop in interface_properties}
 
     # ------------------------------------------------------------
     def friendly_class_name(self):