Work on trying to get access to settings...

This commit is contained in:
Dima Gerasimov 2019-06-29 18:29:51 +01:00 committed by Dmitrii Gerasimov
parent 823b2a11de
commit e8260f2f47

View file

@ -1,6 +1,10 @@
#!/usr/bin env python3
from pathlib import Path
from time import sleep
from contextlib import contextmanager
from tempfile import TemporaryDirectory
import re
import json
from selenium import webdriver # type: ignore
import pyautogui # type: ignore
@ -9,11 +13,38 @@ import pyautogui # type: ignore
addon_path = (Path(__file__).parent.parent / 'build').absolute()
@contextmanager
def get_webdriver():
profile = webdriver.FirefoxProfile()
with TemporaryDirectory() as td:
profile = webdriver.FirefoxProfile(td)
driver = webdriver.Firefox(profile, firefox_binary='/L/soft/firefox-dev/firefox/firefox')
driver.install_addon(str(addon_path), temporary=True)
return driver
# necessary to trigger prefs.js initialisation..
driver.get('http://example.com')
sleep(1)
moz_profile = Path(driver.capabilities['moz:profile'])
prefs_file = moz_profile / 'prefs.js'
addon_id = None
print(prefs_file)
for line in prefs_file.read_text().splitlines():
print(line)
# temporary-addon\":\"53104c22-acd0-4d44-904c-22d11d31559a\"}")
m = re.search(r'temporary-addon.....([0-9a-z-]+)."', line)
if m is None:
continue
addon_id = m.group(1)
print("ADDON " + addon_id)
assert addon_id is not None
options = f'moz-extension://{addon_id}/options.html'
print(options)
driver.get(options)
import ipdb; ipdb.set_trace()
yield driver
# TODO close driver?
# TODO need to fix hotkey!
@ -34,11 +65,12 @@ def trigger_grasp():
# pyautogui.locateOnScreen('/L/soft/browser-extensions/grasp/unicorn.png')
print("sending hotkey!")
import ipdb; ipdb.set_trace()
pyautogui.hotkey('ctrl', 'alt', 'c')
def test():
driver = get_webdriver()
with get_webdriver() as driver:
driver.get('https://en.wikipedia.org/wiki/Automation')
sleep(2) # just in case..