From a74dee22cae3775160b735ac6b0ce0f9a251e6c5 Mon Sep 17 00:00:00 2001 From: Jan-Henrik Bruhn Date: Fri, 17 Feb 2023 21:03:14 +0100 Subject: [PATCH] Different auth scheme --- screenshotter/config.yaml | 2 +- screenshotter/rootfs/app/app.py | 19 ++++++++++--------- 2 files changed, 11 insertions(+), 10 deletions(-) diff --git a/screenshotter/config.yaml b/screenshotter/config.yaml index 01d40c4..322daa8 100644 --- a/screenshotter/config.yaml +++ b/screenshotter/config.yaml @@ -1,6 +1,6 @@ --- name: E-Paper Screenshotter -version: 1.0.1.25 +version: 1.0.1.26 #image: ghcr.io/maxwinterstein/homeassistant-addon-toogoodtogo-ha-mqtt-bridge-{arch} slug: screenshotter description: Screenshot images for E-Paper pricetags diff --git a/screenshotter/rootfs/app/app.py b/screenshotter/rootfs/app/app.py index 7417e80..31d4a7f 100644 --- a/screenshotter/rootfs/app/app.py +++ b/screenshotter/rootfs/app/app.py @@ -62,18 +62,19 @@ while True: driver.implicitly_wait(10) if task.get("ha_auth"): - def interceptor(request): - if request.headers['Authorization']: - del request.headers['Authorization'] # Remember to delete the header first - request.headers['Authorization'] = f"Bearer {ha_token}" # Spoof the referer - driver.request_interceptor = interceptor + #def interceptor(request): + # if request.headers['Authorization']: + # del request.headers['Authorization'] # Remember to delete the header first + # request.headers['Authorization'] = f"Bearer {ha_token}" # Spoof the referer + #driver.request_interceptor = interceptor def response_interceptor(req, response): response.headers['Access-Control-Allow-Origin'] = '*' driver.response_interceptor = response_interceptor - #logging.info("Doing HA Auth") - #driver.execute_script(f"window.localStorage.setItem('hassTokens', JSON.stringify({{hassUrl: '{ha_url}', access_token: '{ha_token}', token_type: 'Bearer'}}));") - #driver.get(task["url"]) - #time.sleep(int(wait)) + + driver.get("http://172.30.32.1:8123") + time.sleep(int(wait)) + logging.info("Doing HA Auth") + driver.execute_script(f"window.localStorage.setItem('hassTokens', JSON.stringify({{hassUrl: '{ha_url}', access_token: '{ha_token}', token_type: 'Bearer'}}));") driver.get(task["url"]) time.sleep(int(wait))