import logging from actions import open_page from classes.application_result import ApplicationResult from language import _ from providers._provider import ApplyContext, Provider logger = logging.getLogger("flat-apply") class Wbm(Provider): @property def domain(self) -> str: return "wbm.de" async def apply_for_flat(self, url: str, ctx: ApplyContext) -> ApplicationResult: r = ctx.recorder p = ctx.profile async with open_page(url, recorder=r) as page: await r.step_snap(page, "page.404_check") if await page.get_by_role("heading", name="Page Not Found").is_visible(): await r.step_snap(page, "abort.not_found", status="warn") return ApplicationResult(False, message=_("not_found")) await r.step_snap(page, "cookies.check") cookie_accept_btn = page.get_by_text("Alle zulassen") if await cookie_accept_btn.is_visible(): await cookie_accept_btn.click() await r.step_snap(page, "cookies.accepted") r.step("chatbot.remove") try: await page.locator('#removeConvaiseChat').click() except Exception as e: r.step("chatbot.remove_failed", "warn", str(e)) await r.step_snap(page, "ad.offline_check") if page.url == "https://www.wbm.de/wohnungen-berlin/angebote/": await r.step_snap(page, "abort.offline", status="warn") return ApplicationResult(False, message=_("ad_offline")) await r.step_snap(page, "form.open") await page.locator('.openimmo-detail__contact-box-button').click() await r.step_snap(page, "form.wbs_section") if p.is_possessing_wbs: await page.locator('label[for="powermail_field_wbsvorhanden_1"]').click() await page.locator("input[name*='[wbsgueltigbis]']").fill(p.wbs_valid_till_dt().strftime("%Y-%m-%d")) await page.locator("select[name*='[wbszimmeranzahl]']").select_option(str(p.wbs_rooms)) await page.locator("#powermail_field_einkommensgrenzenacheinkommensbescheinigung9").select_option(p.wbs_type) if p.is_prio_wbs: await page.locator("#powermail_field_wbsmitbesonderemwohnbedarf_1").check(force=True) else: await page.locator('label[for="powermail_field_wbsvorhanden_2"]').click() await r.step_snap(page, "form.personal") await page.locator("#powermail_field_anrede").select_option(p.salutation) await page.locator("#powermail_field_name").fill(p.lastname) await page.locator("#powermail_field_vorname").fill(p.firstname) await page.locator("#powermail_field_strasse").fill(p.street) await page.locator("#powermail_field_plz").fill(p.postcode) await page.locator("#powermail_field_ort").fill(p.city) await page.locator("#powermail_field_e_mail").fill(p.email) await page.locator("#powermail_field_telefon").fill(p.telephone) await page.locator("#powermail_field_datenschutzhinweis_1").check(force=True) await r.step_snap(page, "form.filled") if not ctx.submit_forms: await r.step_snap(page, "submit.dry_run") return ApplicationResult(success=True, message=_("application_success_dry")) await r.step_snap(page, "submit.click") await page.get_by_role("button", name="Anfrage absenden").click() await r.step_snap(page, "success.check") if await page.get_by_text("Wir haben Ihre Anfrage für das Wohnungsangebot erhalten.").is_visible(): return ApplicationResult(True) elif await self._missing_fields_warning(page): await r.step_snap(page, "missing_fields_detected", status="warn") return ApplicationResult(False, _("missing_fields")) await r.step_snap(page, "success.not_found", status="warn") return ApplicationResult(success=False, message=_("submit_conformation_msg_not_found")) async def _missing_fields_warning(self, page) -> bool: missing_field_msg = page.get_by_text("Dieses Feld muss ausgefüllt werden!").first return await missing_field_msg.first.is_visible()