Implement native backend for WPF applications

.gitmodules

@@ -0,0 +1,4 @@
+[submodule "pywinauto/windows/injected"]
+	path = pywinauto/windows/injected
+	url = https://github.com/pywinauto/injected.git
+	branch = master

appveyor.yml

@@ -6,7 +6,7 @@
 #
 
 # fetch repository as a zip archive
-shallow_clone: true                 # default is "false"
+shallow_clone: false                 # default is "false"
 
 environment:
 
@@ -67,6 +67,10 @@ install:
   # as well as pywin32, pillow and coverage
   - "powershell ./ci/install.ps1"
   - "SET PATH=%PYTHON%;%PYTHON%\\Scripts;%PATH%"
+  - cd %APPVEYOR_BUILD_FOLDER%
+  - git submodule init
+  - git submodule update
+
 
   # Install the build dependencies of the project. If some dependencies contain
   # compiled extensions and are not provided as pre-built wheel packages,
@@ -76,7 +80,6 @@ install:
 
   # Enable desktop (for correct screenshots).
   #- ps: iex ((new-object net.webclient).DownloadString('https://raw.githubusercontent.com/appveyor/ci/master/scripts/enable-desktop.ps1'))
-
 build: false  # Not a C# project, build stuff at the test step instead.
 
 test_script:

pywinauto/base_application.py

@@ -223,6 +223,8 @@ def __find_base(self, criteria_, timeout, retry_interval):
                 if 'backend' not in ctrl_criteria:
                     ctrl_criteria['backend'] = self.backend.name
 
+                if self.app is not None:
+                    ctrl_criteria['pid'] = self.app.process
                 ctrl = self.backend.generic_wrapper_class(findwindows.find_element(**ctrl_criteria))
                 previous_parent = ctrl.element_info
                 ctrls.append(ctrl)
@@ -479,7 +481,6 @@ def by(self, **criteria):
         new_item = WindowSpecification(self.criteria[0], allow_magic_lookup=self.allow_magic_lookup)
         new_item.criteria.extend(self.criteria[1:])
         new_item.criteria.append(criteria)
-
         return new_item
 
     def __getitem__(self, key):
@@ -574,7 +575,16 @@ def __getattribute__(self, attr_name):
             # FIXME - I don't get this part at all, why is it win32-specific and why not keep the same logic as above?
             # if we have been asked for an attribute of the dialog
             # then resolve the window and return the attribute
-            desktop_wrapper = self.backend.generic_wrapper_class(self.backend.element_info_class())
+            if self.backend.name in ('wpf'):
+                if self.app is None and len(self.criteria) > 0 and 'pid' in self.criteria[0]:
+                    pid = self.criteria[0]['pid']
+                else:
+                    pid = self.app.process
+                desktop_wrapper = self.backend.generic_wrapper_class(
+                    self.backend.element_info_class(pid=pid)
+                )
+            else:
+                desktop_wrapper = self.backend.generic_wrapper_class(self.backend.element_info_class())
             need_to_resolve = (len(self.criteria) == 1 and hasattr(desktop_wrapper, attr_name))
             if hasattr(self.backend, 'dialog_class'):
                 need_to_resolve = need_to_resolve and hasattr(self.backend.dialog_class, attr_name)

pywinauto/base_wrapper.py

@@ -170,6 +170,8 @@ def by(self, **criteria):
 
         criteria['backend'] = self.backend.name
         criteria['parent'] = self.element_info
+        if self.backend.name in ('wpf'):
+            criteria['pid'] = self.element_info.pid
         child_specification = WindowSpecification(criteria)
 
         return child_specification

pywinauto/controls/__init__.py

@@ -48,5 +48,8 @@
     from . import common_controls
     from . import win32_controls
 
+    from . import wpfwrapper
+    from . import wpf_controls
+
 
 from ..base_wrapper import InvalidElement

pywinauto/controls/uia_controls.py

@@ -1126,7 +1126,7 @@ def item_by_path(self, path, exact=False):
         menu_items = [p.strip() for p in path.split("->")]
         items_cnt = len(menu_items)
         if items_cnt == 0:
-            raise IndexError()
+            raise IndexError("Menu path has incorrect format, no item identifiers found")
         for item in menu_items:
             if not item:
                 raise IndexError("Empty item name between '->' separators")
@@ -1141,8 +1141,9 @@ def next_level_menu(parent_menu, item_name, is_last):
         # a new Menu control is created and placed on the dialog. It can be
         # a direct child or a descendant.
         # Sometimes we need to re-discover Menu again
+        i = 0
         try:
-            menu = next_level_menu(self, menu_items[0], items_cnt == 1)
+            menu = next_level_menu(self, menu_items[i], items_cnt == 1)
             if items_cnt == 1:
                 return menu
 
@@ -1156,8 +1157,8 @@ def next_level_menu(parent_menu, item_name, is_last):
 
             for i in range(1, items_cnt):
                 menu = next_level_menu(menu, menu_items[i], items_cnt == i + 1)
-        except(AttributeError):
-            raise IndexError()
+        except (AttributeError, IndexError):
+            raise IndexError('Incorrect menu path, item "{}" (index {}) not found'.format(menu_items[i], i))
 
         return menu