jcgoette / hi_mama_homeassistant Goto Github PK

        for h2 in response_h2:
            h2_text = h2.get_text(strip=True)
            h2_next_siblings = h2.next_sibling.contents
            # TODO: needs refactor
            if "Preview" in h2_text:
                child_dict["At Daycare"] = True
            if "Preview" not in h2_text:
                if "Report" in h2_text:
                    if i == 0:

        """Query HiMama for data."""
        pymama_data = pymama_query(self._email, self._password, self._child_id)
        pymama_latest = pymama_data.get("Latest")
        # TODO: move to binary_sensor
        pymama_latest["At Daycare"] = pymama_data.get("At Daycare")
        return pymama_latest

        response = BeautifulSoup(response, "html.parser")

        response_h2 = response.find_all("h2")
        for i2, h2 in enumerate(response_h2):
            h2_text = h2.get_text(strip=True)
            h2_next_siblings = h2.next_sibling.contents
            # TODO: needs refactor
            if i == 0 and i2 == 0:
                child_dict["At Daycare"] = True if "Preview" in h2_text else False
            if "Preview" not in h2_text:
                if "Report" in h2_text:
                    if i == 0: