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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
import argparse
import importlib
import os
import signal
import subprocess
from time import sleep
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.wait import WebDriverWait
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument("--browser", required=True)
parser.add_argument("--headless", action="store_true")
parser.add_argument("--outfile", "-o", required=True)
parser.add_argument("--width", type=int, default=900)
parser.add_argument("--height", type=int, default=700)
return parser.parse_args()
def create_webdriver_options(browser, headless):
mod = importlib.import_module(f"selenium.webdriver.{browser}.options")
options = mod.Options()
if headless:
options.headless = True
return options
def create_webdriver(browser, headless):
mod = importlib.import_module(f"selenium.webdriver.{browser}.webdriver")
return mod.WebDriver(options=create_webdriver_options(browser, headless))
def example_server():
proc = subprocess.Popen(["make", "example"])
# `make example` runs a few things before runserver.
sleep(2)
return proc
def set_viewport_size(selenium, width, height):
script = """
return [
window.outerWidth - window.innerWidth + arguments[0],
window.outerHeight - window.innerHeight + arguments[1],
];
"""
window_width, window_height = selenium.execute_script(script, width, height)
selenium.set_window_size(window_width, window_height)
def submit_form(selenium, data):
url = selenium.current_url
for name, value in data.items():
el = selenium.find_element(By.NAME, name)
el.send_keys(value)
el.send_keys(Keys.RETURN)
WebDriverWait(selenium, timeout=5).until(EC.url_changes(url))
def main():
args = parse_args()
with example_server() as p:
try:
with create_webdriver(args.browser, args.headless) as selenium:
set_viewport_size(selenium, args.width, args.height)
selenium.get("http://localhost:8000/admin/login/")
submit_form(selenium, {"username": os.environ["USER"], "password": "p"})
selenium.get("http://localhost:8000/admin/auth/user/")
# Check if SQL Panel is already visible:
sql_panel = selenium.find_element(By.ID, "djdt-SQLPanel")
if not sql_panel:
# Open the admin sidebar.
el = selenium.find_element(By.ID, "djDebugToolbarHandle")
el.click()
sql_panel = selenium.find_element(By.ID, "djdt-SQLPanel")
# Open the SQL panel.
sql_panel.click()
selenium.save_screenshot(args.outfile)
finally:
p.send_signal(signal.SIGTERM)
if __name__ == "__main__":
main()
|
