Update commands/universal.py

commands/universal.py

@@ -12,11 +12,7 @@ def searchQuery(user_search, driver, elem, mode=0, score="", isLoggedIn=True):
                 driver.execute_script("arguments[0].click();", anchor)
         
     user_search = user_search.lower()
-    try:
-        WebDriverWait(driver, timeout=5).until(EC.presence_of_element_located((By.XPATH, elem)))
-    except:
-        driver.refresh()
-        WebDriverWait(driver, timeout=20).until(EC.presence_of_element_located((By.XPATH, elem)))
+    WebDriverWait(driver, timeout=15).until(EC.presence_of_element_located((By.XPATH, elem)))
     search_bar = driver.find_element(By.XPATH, elem)
 
     try:
@@ -31,13 +27,14 @@ def searchQuery(user_search, driver, elem, mode=0, score="", isLoggedIn=True):
     if not mode:
         specific_Query(driver=driver, search_bar=search_bar, user_search=user_search)
     else:
+        time.sleep(1)
         search_bar.send_keys(Keys.ARROW_DOWN)
         time.sleep(2)
     
     if mode:
         search_bar.send_keys(Keys.ARROW_DOWN)
         time.sleep(1.2)
-    search_bar.send_keys(Keys.ENTER)
+        search_bar.send_keys(Keys.ENTER)
     time.sleep(0.3)
 
     try:
@@ -94,44 +91,50 @@ def continue_Search(driver, link, mode=0):
 
 
 def specific_Query(driver, search_bar, user_search):
-    user_search = user_search.replace("_"," ").split()
+    # Replace underscores with spaces and append the full search term
+    user_search_words = user_search.replace("_", " ").split()
+    user_search_words.append(''.join(user_search_words))
+    full_query_found = False
+    query_to_click = None
+
     try:
-        for user_word in user_search:
+        for user_word in user_search_words:
             time.sleep(1.2)
             driver.execute_script('arguments[0].value=arguments[1]', search_bar, user_word)
-            time.sleep(0.6)
+            time.sleep(1)
             search_bar.send_keys(Keys.SPACE)
-            time.sleep(2.2)
-
-            checker = 0
-            tag_pos = 0
-
-            search_bar_queries = driver.find_element(By.XPATH, '//*[@class="sc-1974j38-2 kjWkkt"]').find_elements(By.XPATH, './/*[@class="sc-d98f2c-0"]')
-            for n_iter, query in enumerate(search_bar_queries):
-                query_text = query.find_element(By.XPATH, './/div//div[2]').text.lower()
-
-                for text in user_search:
-                    if query_text.find(text) != -1:
-                        checker += 1
-
-                if checker == len(user_search):
-                    tag_pos = n_iter
+            time.sleep(2.5)
+
+            search_bar_queries = driver.find_elements(By.XPATH, '//*[@class="sc-1974j38-2 kjWkkt"]//a')
+            for query in search_bar_queries:
+                query_text = query.find_element(By.XPATH, './/div').text.lower()
+                if len(query.find_elements(By.XPATH, './/div//div[2]')) > 0:
+                    query_text = query.find_element(By.XPATH, './/div//div[2]').text.lower()
+
+                if all(word in query_text for word in user_search_words[:-1]):
+                    query_to_click = query
+                    full_query_found = True
+                    time.sleep(0.3)
                     break
-                else:
-                    checker = 0
 
-            if checker == len(user_search):
-                for i in range(tag_pos+1):
-                    time.sleep(0.5)
-                    search_bar.send_keys(Keys.ARROW_DOWN)
+            if full_query_found:
                 break
-    except:
-        driver.execute_script('arguments[0].value=arguments[1]', search_bar, user_search)
+
+    except Exception as e:
+        pass
+
+    if full_query_found:
+        driver.execute_script("arguments[0].click();", query_to_click)
+    else:
+        # If full query not found, just press Enter
         search_bar.send_keys(Keys.ENTER)
 
+
 def contains_works(driver, elem):
     try:
         WebDriverWait(driver, timeout=9).until(EC.presence_of_element_located((By.XPATH, elem)))
         return True
     except:
-        return False
+        return False
+    
+