From 7c659f59635866beeceb2dc4520dbaf631a00d95 Mon Sep 17 00:00:00 2001 From: M Pacer Date: Tue, 20 Mar 2018 10:35:58 -0700 Subject: [PATCH] add quick_selenium script for quickly starting up a selenium interactively --- notebook/tests/selenium/quick_selenium.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 notebook/tests/selenium/quick_selenium.py diff --git a/notebook/tests/selenium/quick_selenium.py b/notebook/tests/selenium/quick_selenium.py new file mode 100644 index 000000000..6eff52bd3 --- /dev/null +++ b/notebook/tests/selenium/quick_selenium.py @@ -0,0 +1,16 @@ +from selenium.webdriver import Firefox +from notebook.notebookapp import list_running_servers + + +def quick_driver(): + """Quickly create a selenium driver pointing at an active noteboook server. + + """ + try: + server = list(list_running_servers())[0] + except IndexError as e: + e.message = 'You need a server running before you can run this command' + driver = Firefox() + auth_url = f'{server["url"]}?token={server["token"]}' + driver.get(auth_url) + return driver