Selaa lähdekoodia

Add automated click of submit button to checkIn()

Signed-off-by: Michael Tang <michael.h.tang@gmail.com>
Michael Tang 5 vuotta sitten
vanhempi
commit
357ff34fd9
1 muutettua tiedostoa jossa 18 lisäystä ja 1 poistoa
  1. 18 1
      session.py

+ 18 - 1
session.py

@@ -44,6 +44,23 @@ def checkIn():
     # click next
     buttons[1].click()
 
+    print('waiting 2 seconds for final page to load')
+    time.sleep(2)
+
+    buttons =  driver.find_elements_by_class_name("quantumWizButtonPaperbuttonLabel")
+    # find the submit button 
+    buttonText = buttons[1].text
+    pressSubmit = input("Press submit? (y/n)")
+    if pressSubmit == "y":
+        print(buttons[1].text)
+        buttons[1].click()
+    elif pressSubmit == "n":
+        print("done")
+    else:
+        print("invalid key pressed.")
+    
+
+
     ########################## end of Check in ###########################
 def checkOut():
     ############################ start Check Out ############################
@@ -72,7 +89,7 @@ def locationSelector():
         print("not a valid selection.")
 
 
-DRIVER_PATH = "/Users/michaeltang/.wdm/drivers/chromedriver/88.0.4324.96/mac64/chromedriver"
+DRIVER_PATH = "/home/michtang/selenium/chromedriver"
 
 
 # URL = input("Enter the URL: ")