archivist_insert_workaround_clone/click_archivist_delete_button.py at master · beckyoldroyd/archivist_insert_workaround_clone · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
#!/usr/bin/env python3

"""
Python 3
    Web scraping using selenium to delete everything
"""

import sys
import time

from selenium import webdriver
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities

driver = webdriver.Remote(
         command_executor='http://selenium-standalone-firefox:4444/wd/hub',
         desired_capabilities=DesiredCapabilities.FIREFOX)

# driver = webdriver.Firefox(executable_path="/home/jenny/Documents/python_scripts.git/geckodriver")


def archivist_login(url, uname, pw):
    """
    Log in to archivist
    """
    driver.get(url)
    time.sleep(5)
    driver.find_element_by_id("login-email").send_keys(uname)
    driver.find_element_by_id("login-password").send_keys(pw)
    driver.find_element_by_class_name("btn-default").click()
    time.sleep(5)


def click_delete_button(uname, pw):
    """
    Click 'Delete' for the only study
    """
    # log in
    archivist_url = "https://closer-temp.herokuapp.com"
    export_url = "https://closer-temp.herokuapp.com/admin/instruments"
    archivist_login(archivist_url, uname, pw)
    driver.get(export_url)
    time.sleep(10)

    # locate id and link
    trs = driver.find_elements_by_xpath("html/body/div/div/div/div/div/div/table/tbody/tr")
    for i in range(1, len(trs)):
        # row 0 is header: tr has "th" instead of "td"
        tr = trs[i]

        # column 2 is "Prefix"
        prefix = tr.find_elements_by_xpath("td")[1].text

        # column 6 is "Actions", click on "Delete"
        deleteButton = tr.find_elements_by_xpath("td")[3].find_elements_by_tag_name('button')[-1]

        print("Click delete button for " + prefix)
        deleteButton.click()

        time.sleep(5)
        driver.find_element_by_id("confirm-prefix").send_keys(prefix)
        driver.find_element_by_class_name("btn-danger").click()

    driver.quit()


def main():
    uname = sys.argv[1]
    pw = sys.argv[2]

    click_delete_button(uname, pw)


if __name__ == "__main__":
    main()