diff --git a/main.py b/main.py index e552c2b..9457837 100644 --- a/main.py +++ b/main.py @@ -205,11 +205,6 @@ def main(driver: WebDriver, logger = logging.getLogger('main')): logger.warning('Error while fetching data from %s, retrying... (%d)', url, attempt, exc_info=e) assert attempt < retry - flow = ActionFlow() - profile = None - progress = { 'task': '' } - selection = 0 - class Wait(Action): @classmethod def prepare(cls): @@ -217,8 +212,8 @@ def main(driver: WebDriver, logger = logging.getLogger('main')): @classmethod def perform(cls): - flow.do(cls) wait(1) + return True class Cancel(Action): @classmethod @@ -244,12 +239,17 @@ def main(driver: WebDriver, logger = logging.getLogger('main')): driver.switch_to.window(driver.current_window_handle) raise cls + flow = ActionFlow() flow.append(Wait) flow.allow(Wait) flow.do(Wait) flow.append(Cancel) flow.append(Skip) + profile = None + progress = { 'task': '' } + selection = 0 + jsonrpc2.define('actions', lambda: flow.capabilities()) jsonrpc2.define('cancel', lambda: flow.do(Cancel)) jsonrpc2.define('skip', lambda: flow.do(Skip)) diff --git a/requirements.txt b/requirements.txt index 8244844..8c5210c 100644 Binary files a/requirements.txt and b/requirements.txt differ