71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
from selenium import webdriver
|
|
import AutoTicket
|
|
import AutoColor
|
|
import json
|
|
import os
|
|
|
|
urls = {"NOC-PROV": "", "NOC-PROV-ADV" : "", "PROV-DSL" : "", "PROV-FIBER":"", "PROV-IPBB":"","PROV-MISC":"","FUS-PROV":"","TICKETS":""}
|
|
|
|
def assign_urls(dictionary):
|
|
for key in dictionary:
|
|
value = input(f"Enter url for '{key}': ")
|
|
dictionary[key] = value
|
|
|
|
def get_team():
|
|
user_list = []
|
|
print("What are the usernames?")
|
|
while True:
|
|
user_input = input("Add to list (Press Enter to finish): ")
|
|
if user_input:
|
|
user_list.append(user_input.lower())
|
|
else:
|
|
break
|
|
print("I'm working with these users:", user_list)
|
|
print("")
|
|
print("This look good? y/n")
|
|
answer = input()
|
|
if "n" in answer:
|
|
print("Let's try that again...")
|
|
print("")
|
|
get_team()
|
|
else:
|
|
return user_list
|
|
|
|
def first_time_setup():
|
|
print("Initiating first time set up")
|
|
print("To ensure internal security, you will need to provide urls for each location,")
|
|
print("As well as employee usernames (first.last)")
|
|
print("If you don't know, you shouldn't be using this script.")
|
|
assign_urls(urls)
|
|
with open('urls.json', 'w') as file:
|
|
json.dump(urls, file)
|
|
users = get_team()
|
|
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()
|
|
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()
|
|
|
|
main() |