Model/scripts/hyde/build_217091901.py

"""Elmhurst build for UPRN 217091901 (RdSAP-21.0.1, TOP-FLOOR FLAT, band A,
SOLID BRICK 280 mm UNINSULATED, FLAT ROOF no insulation (As Built), floor over
ANOTHER DWELLING below (zero loss), mains-gas COMBI (PCDB 18658), 6 DG windows
~8.62 m², no doors, TFA 63. Lodged SAP 62; engine 60.82 (Δ −1.18, PE +8.7).

Per-cert worksheet validation. The roof dominates loss (145.6 W/K at flat-default
U 2.30); this build confirms the accredited uninsulated flat-roof + solid-brick
wall U against the engine. Run:
    DISPLAY=:99 python scripts/hyde/build_217091901.py <page>
"""
from __future__ import annotations
import sys
from pathlib import Path
import elmhurst_lib as E

DIM = "TabContainer_TabPanelMain_WebUserControlDimensionsMain_"
WALL = ("TabContainer_TabPanelMain_InnerTabContainerMain_"
        "TabPanelExternalWallMain_WebUserControlWallMain_")
ROOF = "TabContainer_TabPanelMain_WebUserControlRoofMain_"
FLOOR = "TabContainer_TabPanelMain_WebUserControlFloorsMain_"
WP = "TabContainer_TabPanelWindowsPanel_"
DP = "TabContainer_TabPanelDoorsPanel_"
VP = "TabContainer_TabPanelVentilationPanel_"
APT = "TabContainer_TabPanelAirPressureTest_"
LP = "TabContainer_TabPanelLighting_"
MV = "TabContainer_TabPanelMechVent_"
WH = "TabContainer_TabPanelWaterHeating_"
MH1B = "TabContainer_TabPanelMainHeating1_WebUserControlMainHeating1_"
_SAMPLE_DIR = (Path(__file__).parent.parent.parent
               / "backend/epc_api/json_samples/real_life_examples"
               / "RdSAP-Schema-21.0.1/uprn_217091901")


def _pick(page, suffix, contains):
    val = page.evaluate(
        """(a)=>{const s=document.getElementById(a[0]);if(!s)return null;
        for(const o of s.options){if(o.text.toLowerCase().includes(a[1].toLowerCase()))return o.value;}return null;}""",
        [f"{E.FP}{suffix}", contains])
    if val is not None:
        E.set_select(page, suffix, val)
    return val


def _options(page, suffix):
    return page.evaluate(
        """(id)=>{const s=document.getElementById(id);if(!s)return [];
        return Array.from(s.options).map(o=>o.text);}""", f"{E.FP}{suffix}")


def property_description(page):
    E.goto(page, "PropertyDescription", "WebFormPropertyDescription.aspx")
    _pick(page, "DropDownListPropertyType1", "flat") or \
        E.set_select(page, "DropDownListPropertyType1", "F Flat")
    E.set_text(page, "TextBoxStoreys", "1")
    E.set_text(page, "TextBoxHabitableRooms", "3")
    E.set_text(page, "TextBoxHeatedHabitableRooms", "3")
    print("date ->", _pick(page, "DropDownListDateBuiltMain", "before 1900"))
    E.set_select(page, "DropDownListDateBuiltFirst", "")
    E.save_close(page)


def flats(page):
    E.goto(page, "Flats", "WebFormFlats.aspx")
    print("flat positions:", _options(page, "" if False else "DropDownListPositionOfFlat"))
    _pick(page, "DropDownListPositionOfFlat", "top") or \
        E.set_select(page, "DropDownListPositionOfFlat", "Top Floor")
    E.set_text(page, "TextBoxFloor", "3")
    E.set_select(page, "RadioButtonListFlatCoridor", "None")
    E.save_close(page)


def dimensions(page):
    E.goto(page, "Dimensions", "WebFormDimensions.aspx")
    E.set_text(page, f"{DIM}TextBoxFloorAreaLowestFloor", "63.29")
    E.set_text(page, f"{DIM}TextBoxRoomHeightLowestFloor", "2.85")
    E.set_text(page, f"{DIM}TextBoxWallPerimeterLowestFloor", "14.0")
    E.set_text(page, f"{DIM}TextBoxPartyWallLengthLowestFloor", "22.8")
    E.save_close(page)


def walls(page):
    E.goto(page, "Walls", "WebFormWalls.aspx")
    _pick(page, f"{WALL}DropDownListType", "solid brick")
    page.wait_for_timeout(500)
    print("insulation opts:", _options(page, f"{WALL}DropDownListInsulation"))
    _pick(page, f"{WALL}DropDownListInsulation", "as built") or \
        _pick(page, f"{WALL}DropDownListInsulation", "none")
    # cert lodges wall_thickness=280mm measured=Y — this field was never
    # set here, so the shared assessment silently carried over whatever
    # thickness an earlier build had left (260mm), masking the real
    # RdSAP 10 Table 13 200/280mm boundary bug on this cert's exact
    # thickness. See domain/sap10_ml/rdsap_uvalues.py
    # _u_brick_thin_wall_age_a_to_e's docstring.
    for suf in (f"{WALL}TextBoxThickness", f"{WALL}TextBoxWallThickness"):
        if page.locator(f"#{E.FP}{suf}").count():
            E.set_text(page, suf, "280"); print("wall thickness set 280 via", suf); break
    E.save_close(page)


def roofs(page):
    E.goto(page, "Roofs", "WebFormRoofs.aspx")
    print("ROOF TYPE opts:", _options(page, f"{ROOF}DropDownListType"))
    _pick(page, f"{ROOF}DropDownListType", "flat")
    page.wait_for_timeout(600)
    _pick(page, f"{ROOF}DropDownListInsulationAt", "flat roof insulation")
    page.wait_for_timeout(600)
    # discover the thickness/additional field that appears for flat-roof insulation
    for suf in (f"{ROOF}DropDownListThickness", f"{ROOF}DropDownListInsulation",
                f"{ROOF}DropDownListAdditionalInsulation"):
        if page.locator(f"#{E.FP}{suf}").count():
            print(f"  {suf.split('_')[-1]} opts:", _options(page, suf))
            _pick(page, suf, "as built") or _pick(page, suf, "unknown") or \
                _pick(page, suf, "none") or _pick(page, suf, "nil")
    E.save_close(page)


def floors(page):
    E.goto(page, "Floors", "WebFormFloors.aspx")
    print("FLOOR LOCATION opts:", _options(page, f"{FLOOR}DropDownListLocation"))
    # "(other premises below)" = above a partially-heated space
    _pick(page, f"{FLOOR}DropDownListLocation", "another dwelling below")
    E.save_close(page)


def openings(page):
    E.goto(page, "Openings", "WebFormOpenings.aspx")
    E.click_tab(page, "TabContainer_TabPanelWindowsPanel")
    E.set_single_window(page, 8.62, "North", _glazing(page))
    _delete_zero_rows(page)
    E.click_tab(page, "TabContainer_TabPanelDoorsPanel")
    E.set_text(page, f"{DP}TextBoxDoors", "0")
    E.set_text(page, f"{DP}TextBoxDoorsInsulated", "0")
    E.set_text(page, f"{DP}TextBoxDraughtProofedDoors", "0")
    E.save_close(page)


def _glazing(page):
    # Cert glazed_type=5 → SINGLE glazing (epc_codes.csv, RdSAP-Schema-21).
    for opt in _options(page, f"{WP}DropDownListExtGlazing"):
        if "single" in opt.lower():
            return opt
    return "Single glazed"


def _add_window(page, area, orientation, glazing):
    print("glazing ->", glazing)
    E.set_select(page, f"{WP}DropDownListExtGlazing", glazing)
    page.wait_for_timeout(400)
    ft = page.locator(f"#{E.FP}{WP}DropDownListExtFrameType")
    if ft.count():
        ft.select_option("PVC")
    gg = page.locator(f"#{E.FP}{WP}DropDownListExtGlazingGap")
    if gg.count():
        gg.select_option("12 mm")
    wid = f"{E.FP}{WP}TextBoxExtWidth"
    page.evaluate(
        """(a)=>{const e=document.getElementById(a[0]);if(e){e.value=a[1];
        e.dispatchEvent(new Event('input',{bubbles:true}));
        e.dispatchEvent(new Event('change',{bubbles:true}));
        e.dispatchEvent(new Event('blur',{bubbles:true}));}}""", [wid, str(area)])
    page.locator(f"#{E.FP}{WP}TextBoxExtHeight").fill("1.00")
    page.locator(f"#{E.FP}{WP}DropDownListExtOrientation").select_option(orientation)
    page.locator(f"#{E.FP}{WP}DropDownListExtBuildingPartId").select_option("Main")
    page.locator(f"#{E.FP}{WP}DropDownListExtLocation").select_option("External wall")
    page.wait_for_timeout(300)
    before = E.window_row_count(page)
    page.evaluate("(id)=>{const e=document.getElementById(id); if(e)e.click();}", f"{E.FP}{WP}ButtonAddWindow")
    for _ in range(25):
        page.wait_for_timeout(200)
        if E.window_row_count(page) > before:
            break


def _grid_rows(page):
    return page.evaluate(
        """()=>{const t=document.querySelector("[id*=GridViewExtendedWidows]");
        if(!t)return[];return Array.from(t.querySelectorAll('tr')).slice(1)
        .map(r=>Array.from(r.querySelectorAll('td')).map(c=>c.innerText.trim()));}""")


def _delete_zero_rows(page):
    g = 0
    while g < 6 and E.window_row_count(page) > 1:
        g += 1
        rows = _grid_rows(page)
        bad = next((i for i, c in enumerate(rows) if len(c) > 1 and c[1] in ("0.00", "0", "0.0")), None)
        if bad is None:
            break
        _delete_row(page, bad)
        page.wait_for_timeout(400)


def _delete_row(page, idx):
    before = E.window_row_count(page)
    btn = page.evaluate(
        """(i)=>{const b=document.querySelectorAll("[id*='GridViewExtendedWidows_DeleteButton_']");return b[i]?b[i].id:null;}""", idx)
    if not btn:
        return
    page.evaluate("(id)=>{const e=document.getElementById(id); if(e)e.click();}", btn)
    page.wait_for_selector(f"#{E.FP}DeleteWindowDialog_LinkButtonYes", state="visible", timeout=5000)
    page.evaluate("(id)=>{const e=document.getElementById(id); if(e)e.click();}", f"{E.FP}DeleteWindowDialog_LinkButtonYes")
    for _ in range(20):
        page.wait_for_timeout(200)
        if E.window_row_count(page) < before:
            break


def ventilation(page):
    E.goto(page, "VentilationAndCooling", "WebFormVentilationAndCooling.aspx")
    E.click_tab(page, "TabContainer_TabPanelVentilationPanel")
    E.set_text(page, f"{VP}TextBoxIntermittentFans", "0")
    cool = page.locator(f"#{E.FP}{VP}CheckBoxFixedSpaceCooling")
    if cool.count() and cool.is_checked():
        E.commit(page, cool.uncheck)
    E.click_tab(page, "TabContainer_TabPanelMechVent")
    mv = page.locator(f"#{E.FP}{MV}CheckBoxMechanicalVentilation")
    if mv.count() and mv.is_checked():
        E.commit(page, mv.uncheck)
    E.click_tab(page, "TabContainer_TabPanelAirPressureTest")
    E.set_select(page, f"{APT}DropDownListTestMethod", "Not available")
    E.click_tab(page, "TabContainer_TabPanelLighting")
    E.set_text(page, f"{LP}TextBoxLightsTotal", "6")
    E.set_text(page, f"{LP}TextBoxLedLightsTotal", "0")
    E.set_text(page, f"{LP}TextBoxCflLightsTotal", "0")
    E.save_close(page)


def space_heating(page):
    E.goto(page, "SpaceHeating", "WebFormSpaceHeating.aspx")
    page.wait_for_timeout(1000)
    mhc = page.locator(f"#{E.MH1}TextBoxMainHeatingCode")
    code = mhc.input_value() if mhc.count() else ""
    if code and code not in ("0",):
        print(f"clearing leftover MainHeatingCode {code}")
        page.evaluate("""(id)=>{const e=document.getElementById(id);if(e){e.value='';
            e.dispatchEvent(new Event('change',{bubbles:true}));}}""", f"{E.MH1}TextBoxMainHeatingCode")
        page.wait_for_timeout(400)
        E.save_close(page)
        return
    E.set_heating_dialog(page, f"{MH1B}ButtonMainHeatingCode",
                         "^Gas", "Mains gas", "Boilers", "Post 1998", "Condensing",
                         "Combi condens")
    print("code:", page.locator(f"#{E.MH1}TextBoxMainHeatingCode").input_value())
    E.set_heating_dialog(page, f"{MH1B}ButtonMainHeatingControls",
                         "^Boilers", "^Standard", "CBE Programmer, room thermostat and TRVs")
    print("control:", page.locator(f"#{E.MH1}TextBoxMainHeatingControls").input_value())
    E.set_select(page, "DropDownListSecondaryHeatingPresent", "No")
    E.save_close(page)


def water_heating(page):
    E.goto(page, "WaterHeating", "WebFormWaterHeating.aspx")
    E.click_tab(page, "TabContainer_TabPanelWaterHeating")
    page.wait_for_timeout(400)
    E.clear_hot_water_cylinder(page)
    E.set_heating_dialog(page, f"{WH}ButtonWaterHeatingCode",
                         "From Space Heating", "From the primary heating system")
    print("water code:", page.locator(f"#{E.FP}{WH}TextBoxWaterHeatingCode").input_value())
    E.save_close(page)


def download(page):
    _SAMPLE_DIR.mkdir(parents=True, exist_ok=True)
    E.goto(page, "Recommendations", "WebFormRecommendations.aspx")
    page.wait_for_timeout(1500)
    with page.expect_navigation(wait_until="networkidle", timeout=40000):
        page.click("#ContentBody_buttonActionSummary_Link", timeout=15000)
    page.wait_for_timeout(2000)
    for btn, fname in (("ContentBody_ContentPlaceHolder1_LinkButtonSummary", "elmhurst_summary.pdf"),
                       ("ContentBody_ContentPlaceHolder1_ButtonDebugInforPdf", "elmhurst_worksheet.pdf")):
        try:
            with page.expect_download(timeout=60000) as dl:
                page.click(f"#{btn}", timeout=15000)
            dl.value.save_as(str(_SAMPLE_DIR / fname))
            print("saved", fname)
        except Exception as e:
            print("!! download failed", fname, type(e).__name__, str(e)[:120])


_ORDER = ["property_description", "flats", "dimensions", "walls", "roofs",
          "floors", "openings", "ventilation", "space_heating", "water_heating",
          "download"]


def main():
    if len(sys.argv) < 2 or sys.argv[1] not in _ORDER:
        print("usage: build_217091901.py <" + "|".join(_ORDER) + ">")
        return 2
    with E.session() as (ctx, page):
        globals()[sys.argv[1]](page)
        print("done:", sys.argv[1], "->", page.url)
    return 0


if __name__ == "__main__":
    raise SystemExit(main())