diff --git a/scripts/download_ecmk_files.py b/backend/ecmk_fetcher/handler/handler.py similarity index 84% rename from scripts/download_ecmk_files.py rename to backend/ecmk_fetcher/handler/handler.py index d1647b4f..14ca4d4c 100644 --- a/scripts/download_ecmk_files.py +++ b/backend/ecmk_fetcher/handler/handler.py @@ -1,8 +1,21 @@ import os +from enum import Enum +from typing import Any, Mapping from playwright.sync_api import sync_playwright, TimeoutError as PlaywrightTimeoutError -def download_report(username: str, password: str): +class file_download_button_types(Enum): + ASSESSOR_HUB_SITENOTE_REPORT = 11 + CERTIFICATE = 9 + SITENOTE_REPORT = 8 + RAW_XML = 7 + SAP_WORK_SHEET = 15 + + +def download_report(): + username = "" + password = "" + with sync_playwright() as p: browser = p.chromium.launch(headless=True) @@ -66,7 +79,10 @@ def download_report(username: str, password: str): browser.close() +def handler(event: Mapping[str, Any], context: Any) -> None: + download_report() + + if __name__ == "__main__": - email = "" - password = "" - download_report(email, password) + event = {"Records": [{"body": "{}"}]} + handler(event, None)