Skip to content

Commit

Permalink
test
Browse files Browse the repository at this point in the history
  • Loading branch information
Bram van Dartel committed Dec 29, 2024
1 parent f167a74 commit 275a23d
Show file tree
Hide file tree
Showing 4 changed files with 36 additions and 28 deletions.
46 changes: 23 additions & 23 deletions custom_components/afvalwijzer/__init__.py
Original file line number Diff line number Diff line change
@@ -1,31 +1,31 @@
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.typing import ConfigType
from .const.const import DOMAIN
# from homeassistant.config_entries import ConfigEntry
# from homeassistant.core import HomeAssistant
# from homeassistant.helpers.typing import ConfigType
# from .const.const import DOMAIN


async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Set up the Afvalwijzer integration."""
hass.data.setdefault(DOMAIN, {})
return True
# async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
# """Set up the Afvalwijzer integration."""
# hass.data.setdefault(DOMAIN, {})
# return True


async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up Afvalwijzer from a config entry."""
# Store config entry data
hass.data[DOMAIN][entry.entry_id] = entry.data
# async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
# """Set up Afvalwijzer from a config entry."""
# # Store config entry data
# hass.data[DOMAIN][entry.entry_id] = entry.data

# Forward the setup to the sensor platform
await hass.config_entries.async_forward_entry_setups(entry, ["sensor"])
return True
# # Forward the setup to the sensor platform
# await hass.config_entries.async_forward_entry_setups(entry, ["sensor"])
# return True


async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload a config entry."""
# Remove stored data
if entry.entry_id in hass.data[DOMAIN]:
hass.data[DOMAIN].pop(entry.entry_id)
# async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
# """Unload a config entry."""
# # Remove stored data
# if entry.entry_id in hass.data[DOMAIN]:
# hass.data[DOMAIN].pop(entry.entry_id)

# Unload the sensor platform
await hass.config_entries.async_forward_entry_unload(entry, "sensor")
return True
# # Unload the sensor platform
# await hass.config_entries.async_forward_entry_unload(entry, "sensor")
# return True
5 changes: 4 additions & 1 deletion custom_components/afvalwijzer/collector/mijnafvalwijzer.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,13 +25,16 @@ def get_waste_data_raw(provider, postal_code, street_number, suffix):

try:
response = raw_response.json()
print(response)
ophaaldagen_data = response.get("ophaaldagen", {}).get("data", [])
ophaaldagen_next_data = response.get("ophaaldagenNext", {}).get("data", [])

# print(ophaaldagen_next_data)

if not ophaaldagen_data and not ophaaldagen_next_data:
_LOGGER.error("Address not found or no data available!")
raise KeyError
except KeyError as err:
raise KeyError(f"Invalid and/or no data received from {url}") from err

return ophaaldagen_data + ophaaldagen_next_data[:25]
return ophaaldagen_data + ophaaldagen_next_data
2 changes: 1 addition & 1 deletion custom_components/afvalwijzer/tests/test.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,7 +119,7 @@ def parse_data(self, data):
if __name__ == "__main__":
logging.basicConfig(level=logging.INFO)

collector = AfvalwijzerCollector("mijnafvalwijzer", "5146EA", 73, "")
collector = AfvalwijzerCollector("mijnafvalwijzer", "5685AB", 57, "")
repository = WasteCollectionRepository()

try:
Expand Down
11 changes: 8 additions & 3 deletions custom_components/afvalwijzer/tests/test_module.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,14 +42,19 @@
# street_number = "2"

# Afvalwijzer
provider = "mijnafvalwijzer"
postal_code = "5146eg"
street_number = "1"
# provider = "mijnafvalwijzer"
# postal_code = "5146eg"
# street_number = "1"

# provider = "mijnafvalwijzer"
# postal_code = "5563CM"
# street_number = "22"

provider = "mijnafvalwijzer"
postal_code = "5685AB"
street_number = "57"


# provider = "mijnafvalwijzer"
# postal_code = "3601AC"
# street_number = "10"
Expand Down

0 comments on commit 275a23d

Please sign in to comment.