initial working env

2025-12-11 03:30:35 -05:00
commit 25469a54b3
11 changed files with 799 additions and 0 deletions
--- a/download_xml.py
+++ b/download_xml.py
@@ -0,0 +1,66 @@
+from pathlib import Path
+from playwright.async_api import async_playwright
+from schemas import DownloadParameters
+
+
+async def download_xml_files(params: DownloadParameters) -> Path:
+    async with async_playwright() as p:
+        browser = await p.chromium.launch(headless=True)
+        page = await browser.new_page()
+        await page.goto("https://alectrautilitiesgbportal.savagedata.com")
+        await page.wait_for_load_state("networkidle")
+
+        account_name = page.locator("""xpath=//*[@id="account-name"]""")
+        await account_name.wait_for()
+        await account_name.click()
+        await account_name.fill(params.account_name)
+
+        account_number = page.locator("""xpath=//*[@id="idAccountNumber"]""")
+        await account_number.wait_for()
+        await account_number.click()
+        await account_number.fill(params.account_number)
+
+        await page.keyboard.press("Tab")
+        await page.keyboard.type(params.account_phone)
+
+        try:
+            # check if it was filled correctly
+            value = await account_name.input_value()
+            if value != params.account_name:
+                raise ValueError("Account name input failed")
+        except Exception as e:
+            print(f"Error locating or filling account name: {e}")
+            print("Retrying to fill account name...")
+            await account_name.fill(params.account_name)
+
+        submit_button = page.locator("""xpath=//*[@class="btn btn-primary btn-block"]""")
+        await submit_button.wait_for()
+        await submit_button.click()
+
+        download_page_button = page.locator("""xpath=//*[@href="DownloadMyData"]""")
+        await download_page_button.wait_for()
+        await download_page_button.click()
+
+        start_date_picker = page.locator("""xpath=//*[@id="start"]""")
+        await start_date_picker.wait_for()
+        await start_date_picker.fill(params.start_date.strftime("%m-%d-%Y"))
+
+        end_date_picker = page.locator("""xpath=//*[@id="end"]""")
+        await end_date_picker.wait_for()
+        await end_date_picker.fill(params.end_date.strftime("%m-%d-%Y"))
+
+        await page.keyboard.press("Tab")
+        await page.keyboard.press("Space")
+
+        download_button = page.locator("""xpath=//*[@class="btn"]""")
+        await download_button.wait_for()
+
+        # Start waiting for the download
+        async with page.expect_download() as download_info:
+            # Perform the action that initiates download
+            await download_button.click()
+        download = await download_info.value
+        download_path = params.output_dir / download.suggested_filename
+        await download.save_as(download_path)
+        await browser.close()
+        return download_path