Merge pull request #332 from AutomationSolutionz/ProcessID_Generate_Windows

Stored the application's process id while launching

Author: Muntasib-creator

Framework/Built_In_Automation/Desktop/Windows/BuiltInFunctions.py

@@ -35,6 +35,7 @@
 
 
 python_folder = []
+current_pid_list = []
 for location in subprocess.getoutput("where python").split("\n"):
     if "Microsoft" not in location:
         python_folder.append(location)
@@ -386,6 +387,7 @@ def Element_only_search(
 ):
     # max_time is built in wait function.  It will try every seconds 15 times.
     try:
+
         if Parent_Element is not None:
             ParentElement = Parent_Element
             element_index = -1
@@ -405,6 +407,7 @@ def Element_only_search(
 
         )
         all_elements += tmp_elements
+
         if all_elements:
             return all_elements
         else:
@@ -1675,6 +1678,8 @@ def _get_main_window(WindowName):
         MainWindowsList = AutomationElement.RootElement.FindAll(
             TreeScope.Children, Condition.TrueCondition
         )
+
+        found_windows = []
         for MainWindowElement in MainWindowsList:
             try:
                 if WindowName[2] == "pid":
@@ -1686,12 +1691,19 @@ def _get_main_window(WindowName):
                 else:
                     NameS = MainWindowElement.Current.Name
                     if _found(WindowName, NameS):
-                        CommonUtil.ExecLog(sModuleInfo, "Switching to window: %s" % NameS, 1)
-                        autoit.win_activate(NameS)
-                        return MainWindowElement
+                        if MainWindowElement.Current.ProcessId in current_pid_list:
+                            CommonUtil.ExecLog(sModuleInfo, "Switching to window: %s" % NameS, 1)
+                            CommonUtil.ExecLog(sModuleInfo, f"pid matched: {MainWindowElement.Current.ProcessId}", 5)
+                            autoit.win_activate(NameS)
+                            return MainWindowElement
+                        else:
+                            found_windows.append(MainWindowElement)
             except:
                 pass
-
+        if len(found_windows) > 0:
+            CommonUtil.ExecLog(sModuleInfo, "Switching to window: %s" % found_windows[0].Current.Name, 1)
+            autoit.win_activate(found_windows[0].Current.Name)
+            return found_windows[0]
         return None
     except Exception:
         CommonUtil.Exception_Handler(sys.exc_info())
@@ -2096,6 +2108,8 @@ def _open_inspector(inspector, args):
 @logger
 def Run_Application(data_set):
     sModuleInfo = inspect.currentframe().f_code.co_name + " : " + MODULE_NAME
+    global current_pid_list
+    current_pid_list = []
     try:
         args = {"shell": True, "stdin": None, "stdout": None, "stderr": None}
         launch_cond = ""
@@ -2142,6 +2156,7 @@ def Run_Application(data_set):
                 subprocess.Popen(cmd, **args)
                 # Desktop_app = os.path.basename(Desktop_app)
             else:
+                #last_start_time = time.time()
                 autoit.send("^{ESC}")
                 time.sleep(0.5)
                 autoit.send(Desktop_app)
@@ -2154,6 +2169,10 @@ def Run_Application(data_set):
             #     Desktop_app += ".exe"
             CommonUtil.ExecLog(sModuleInfo, "Waiting for the app to launch for maximum %s seconds" % wait, 1)
             s = time.time()
+            for process in psutil.process_iter(['pid', 'name']):
+                if Desktop_app.lower() in process.info['name'].lower():
+                    current_pid_list.append(process.info['pid'])
+
             while time.time() - s < wait:
                 # if len(pygetwindow.getWindowsWithTitle(Desktop_app)) > 0:     # This is case in-sensitive
                 if pygetwindow.getActiveWindow() is None or Desktop_app in pygetwindow.getActiveWindow().title:          # This is case sensitive
@@ -2165,7 +2184,6 @@ def Run_Application(data_set):
                 else:
                     CommonUtil.ExecLog(sModuleInfo, "Could not find any launched app with title: %s however continuing" % Desktop_app, 2)
                 return "passed"
-
         if maximize:
             win = pygetwindow.getWindowsWithTitle(Desktop_app)[0]
             win.maximize()