Run in regular intervals with less intervention

AssignerPal.py

@@ -1,11 +1,15 @@
 from selenium import webdriver
 import AutoTicket
-import AutoColor
+import AutoColor2
 import json
 import os
+import datetime
+import time
 
 urls = {"NOC-PROV": "", "NOC-PROV-ADV" : "", "PROV-DSL" : "", "PROV-FIBER":"", "PROV-IPBB":"","PROV-MISC":"","FUS-PROV":"","TICKETS":""}
 
+tickets_sorted = False
+
 def assign_urls(dictionary):
     for key in dictionary:
         value = input(f"Enter url for '{key}': ")
@@ -43,29 +47,25 @@ def first_time_setup():
     with open('team.json', 'w') as file:
         json.dump(users, file)
 
-def choice(urls,users,driver):
-    print("Tickets or Directly?")
-    user_choice = input()
-    if "ti" in user_choice.lower():
-        AutoTicket.auto_ticket(urls,users,driver)
-    if "di" in user_choice.lower():
-        AutoColor.tag_assist(urls,driver,users)
-    print("Go back to Directly or Tickets choice?")
-    if input("y/n : ") == "y":
-        choice(urls,users,driver)
-
 def main():
     if os.path.exists('urls.json'):
-        print("Do first time set up?")
-        if input("y/n : ") == "y":
-            first_time_setup()
+        print("Skipping first time setup")
     else: first_time_setup()
     driver = webdriver.Firefox()
     with open('urls.json', 'r') as file:
         urls = json.load(file)
     with open('team.json', 'r') as file:
         users = json.load(file)
-    choice(urls,users,driver)
-    driver.close()
+    while True:
+        current_time = datetime.datetime.now().time()
+        start_time = datetime.time(9, 0, 0)
+        end_time = datetime.time(17, 0, 0)
+        if start_time <= current_time <= end_time:
+            AutoTicket.auto_ticket(urls, users, driver, tickets_sorted)
+            AutoColor2.tag_assist(urls, driver, users)
+        else:
+            print("It's after hours. I'm not doing anything.")
+        time.sleep(7200)  # Sleep for 2 hours
+    #driver.close()
 
 main()

AutoColor2.py

@@ -0,0 +1,212 @@
+from selenium import webdriver
+from selenium.webdriver.common.keys import Keys
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.wait import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.common.action_chains import ActionChains
+import datetime
+import random
+
+# Condensed string finder for lists of strings
+def found_string(string_list, main_string):
+    for string in string_list:
+        if string in main_string:
+            return True
+    return False
+
+# Get tag_count dictionary
+def count_tags(queues,tags):
+    tag_count = {}
+    for queue in queues:
+        count = 0
+        for tag in tags:
+            if queue in tag.text.lower():
+                count += 1
+        tag_count[queue] = count
+    return tag_count
+
+def next_string(list_of_strings):
+    index = 0
+    while True:
+        yield list_of_strings[index]
+        index = (index + 1) % len(list_of_strings)
+
+
+
+# Assign steps in order of queue list without checking tag_count
+def dumb_make_tags(queues,tags,driver):
+    ac = ActionChains(driver)
+    original_window = driver.current_window_handle
+    next_queue = next_string(queues)
+    for tag in tags:
+        if found_string(queues, tag.text.lower()) == False:
+            driver.execute_script("arguments[0].scrollIntoView(true);", tag)
+            ac.context_click(tag).perform()
+            WebDriverWait(driver, 10).until(EC.number_of_windows_to_be(2))
+            for window_handle in driver.window_handles:
+                if window_handle != original_window:
+                    driver.switch_to.window(window_handle)
+                    break
+            WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.TAG_NAME, 'h2')))
+            driver.find_element(By.CSS_SELECTOR, "a.edit").click()
+            try:
+                element = WebDriverWait(driver, 10).until(
+                    EC.presence_of_element_located((By.ID, "tag_id"))
+                )
+            finally:
+                tag_field = driver.find_element(By.ID, "tag_id")
+                tag_field.send_keys(Keys.END)
+                tag_field.send_keys("/",next(next_queue))
+                tag_field.send_keys(Keys.ENTER)
+                WebDriverWait(driver,5).until(EC.presence_of_element_located((By.ID,'stepListHeader')))
+                driver.close()
+                driver.switch_to.window(original_window)
+
+# Tool to create a menu based on a list.
+def select_list_item(list_of_strings):
+    for i, item in enumerate(list_of_strings):
+        print(f"{chr(97 + i)}) {item}")
+
+    while True:
+        choice = input("Please select an option (a, b, c, etc.): ").lower()
+        if choice.isalpha() and ord(choice) - 97 in range(len(list_of_strings)):
+            return list_of_strings[ord(choice) - 97]
+        else:
+            print("Invalid choice. Please select a valid option.")
+
+# Select every unassigned tag, print queue count, print step name, use user input to choose what tag to assign
+def guided_make_tags(queues,tags,driver):
+    ac = ActionChains(driver)
+    original_window = driver.current_window_handle
+    for tag in tags:
+        if found_string(queues, tag.text.lower()) == False:
+            driver.execute_script("arguments[0].scrollIntoView(true);", tag)
+            ac.context_click(tag).perform()
+            WebDriverWait(driver, 10).until(EC.number_of_windows_to_be(2))
+            for window_handle in driver.window_handles:
+                if window_handle != original_window:
+                    driver.switch_to.window(window_handle)
+                    break
+            WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.TAG_NAME, 'h2')))
+            print("What should we tag this workflow? ", driver.find_element(By.TAG_NAME, 'h2').text)
+            guided_tag = select_list_item(queues)                
+            driver.find_element(By.CSS_SELECTOR, "a.edit").click()
+            try:
+                element = WebDriverWait(driver, 10).until(
+                    EC.presence_of_element_located((By.ID, "tag_id"))
+                )
+            finally:
+                tag_field = driver.find_element(By.ID, "tag_id")
+                tag_field.send_keys(Keys.END)
+                tag_field.send_keys("/",guided_tag)
+                tag_field.send_keys(Keys.ENTER)
+                WebDriverWait(driver,5).until(EC.presence_of_element_located((By.ID,'stepListHeader')))
+                driver.close()
+                driver.switch_to.window(original_window)
+
+
+
+
+# Get lowest_queue string, and increment tag_count for the value for that queue by one.
+def find_lowest_and_increment(dictionary):
+    if not dictionary:
+        return None
+
+    min_key = min(dictionary, key=dictionary.get)
+
+    dictionary[min_key] += 1
+
+    return min_key
+
+# Check which queue has the least number of assigned workflows.
+def smart_make_tags(queues,tags,driver):
+    ac = ActionChains(driver)
+    original_window = driver.current_window_handle
+    tag_count = count_tags(queues,tags)
+    for tag in tags:
+        if found_string(queues, tag.text.lower()) == False:
+            lowest_queue = find_lowest_and_increment(tag_count)
+            driver.execute_script("arguments[0].scrollIntoView(true);", tag)
+            ac.context_click(tag).perform()
+            WebDriverWait(driver, 10).until(EC.number_of_windows_to_be(2))
+            for window_handle in driver.window_handles:
+                if window_handle != original_window:
+                    driver.switch_to.window(window_handle)
+                    break
+            WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.TAG_NAME, 'h2')))
+            driver.find_element(By.CSS_SELECTOR, "a.edit").click()
+            try:
+                element = WebDriverWait(driver, 10).until(
+                    EC.presence_of_element_located((By.ID, "tag_id"))
+                )
+            finally:
+                tag_field = driver.find_element(By.ID, "tag_id")
+                tag_field.send_keys(Keys.END)
+                tag_field.send_keys("/",lowest_queue)
+                tag_field.send_keys(Keys.ENTER) 
+                WebDriverWait(driver,5).until(EC.presence_of_element_located((By.ID,'stepListHeader')))
+                driver.close()
+                driver.switch_to.window(original_window)
+
+# Ignore future steps.
+def future_tag_manip(driver):
+        future_tag_field = driver.find_element(By.ID, "cf-future-tag")
+        future_tag_field.clear()
+        future_tag_field.send_keys("Automation In Progress")
+
+def filter_steps_manip(driver):
+    step_field = driver.find_element(By.ID, "cf-due-step")
+    step_field.clear()
+    print("")
+    print("Wanna avoid touching a step?")
+    print("Take a look at what's in the queue.")
+    print("If there's anything in there you don't want to tag,")
+    print("copy the step name here, with | to seperate steps.")
+    print("e.g. 'Confirm Disconnect Date Has Passed|Reserve Numbers'")
+    print("or, press Enter to skip")
+    filters = input()
+    if filters != "":
+        step_field.send_keys("^(?!.*(",filters,")).*$")
+    print("")
+    print("This look good? y/n")
+    answer = input()
+    if "n" in answer:
+        print("Let's try that again...")
+        print("")
+        filter_steps_manip()
+
+
+def automation_picker(queues,tags,driver,choice):
+        if choice.lower() == "smart":
+            print("Okay, I'll make you proud (> O ω O)>✎")
+            smart_make_tags(queues,tags,driver)
+        elif choice.lower() == "dumb":
+            print("Okay, I make thing go now ─=≡Σ((( つ＞＜)つ")
+            dumb_make_tags(queues,tags,driver)
+        else:
+            print("Skipping...")
+
+def tag_assist(urls,driver,users):
+    print("This is someone's automation tool. If you don't know what you're doing, please abort.")
+    urllist = urls
+    urllist.popitem()
+    for key, url in urllist.items():
+        print("Working on ",key)
+        driver.get(url)
+        print("")
+        print("Please use the browser to log in. I won't look (づ_ど)")
+        try:
+            element = WebDriverWait(driver, 500).until(
+                EC.presence_of_element_located((By.ID, "dueWorkflowSteps_wrapper"))
+            )
+        finally:
+            future_tag_manip(driver)
+            #filter_steps_manip(driver)
+            tags = driver.find_elements(By.CSS_SELECTOR, "td.tag")
+            if "NOC" in key:
+                automation_picker(users,tags,driver,"smart")
+            else:
+                automation_picker(users,tags,driver,"dumb")
+    print("Done! ✧⋆٩(ᵔ ᗜ ᵔ )و ♡✧")
+    print("Goodbye! ヾ(˶ᵔ ᗜ ᵔ˶)")
+        

AutoTicket.py

@@ -6,6 +6,7 @@ from selenium.webdriver.support import expected_conditions as EC
 from selenium.webdriver.common.action_chains import ActionChains
 import datetime
 import json
+import time
 
 def get_filters(driver):
     print("Setting up filters...")
@@ -60,6 +61,7 @@ def assign_tickets(namecount,driver):
                     driver.switch_to.window(window_handle)
                     break
             WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.ID, "owner")))
+            time.sleep(1)
             controls = driver.find_element(By.ID, 'ticket-controls').find_elements(By.CLASS_NAME, "ticket-control")
             WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.CLASS_NAME, 'select2-container')))
             owner_button = controls[3].find_element(By.CLASS_NAME, 'select2-container')
@@ -68,16 +70,14 @@ def assign_tickets(namecount,driver):
             owner_field = search_fields[3]
             owner_field.send_keys(lowest_name)
             owner_field.send_keys(Keys.ENTER)
-            print("Check to see if this worked, then press enter")
-            input()
-            #assigned_json[lowest_name] += 1
+            #assigned_json[lowest_name] += 
             driver.find_element(By.CLASS_NAME, 'update').click()
             driver.close()
             driver.switch_to.window(original_window)
 
 #assigned_json = {}
 
-def auto_ticket(urls,users,driver):
+def auto_ticket(urls,users,driver,tickets_sorted):
     #for user in users:
     #    assigned_json[user] = 0
     driver.get(urls['TICKETS'])
@@ -88,7 +88,9 @@ def auto_ticket(urls,users,driver):
             EC.presence_of_element_located((By.CLASS_NAME, "select2-selection__clear"))
         )
     finally:
-        get_filters(driver)
+        if tickets_sorted == False:
+            get_filters(driver)
+        time.sleep(2.5)
         namecount = get_names(users,driver)
         print(namecount)
         assign_tickets(namecount,driver)