1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
"""NSW Rural Fire Service Incidents feed."""
from __future__ import annotations
import logging
from datetime import datetime
from aio_geojson_client.feed import GeoJsonFeed
from aiohttp import ClientSession
from geojson import FeatureCollection
from .consts import URL
from .feed_entry import NswRuralFireServiceIncidentsFeedEntry
_LOGGER = logging.getLogger(__name__)
class NswRuralFireServiceIncidentsFeed(
GeoJsonFeed[NswRuralFireServiceIncidentsFeedEntry]
):
"""NSW Rural Fire Services Incidents feed."""
def __init__(
self,
websession: ClientSession,
home_coordinates: tuple[float, float],
filter_radius: float = None,
filter_categories: list[str] = None,
):
"""Initialise this service."""
super().__init__(websession, home_coordinates, URL, filter_radius=filter_radius)
self._filter_categories = filter_categories
def __repr__(self):
"""Return string representation of this feed."""
return "<{}(home={}, url={}, radius={}, categories={})>".format(
self.__class__.__name__,
self._home_coordinates,
self._url,
self._filter_radius,
self._filter_categories,
)
def _new_entry(
self, home_coordinates: tuple[float, float], feature, global_data: dict
) -> NswRuralFireServiceIncidentsFeedEntry:
"""Generate a new entry."""
return NswRuralFireServiceIncidentsFeedEntry(home_coordinates, feature)
def _filter_entries(
self, entries: list[NswRuralFireServiceIncidentsFeedEntry]
) -> list[NswRuralFireServiceIncidentsFeedEntry]:
"""Filter the provided entries."""
filtered_entries = super()._filter_entries(entries)
if self._filter_categories:
filtered_entries = list(
filter(
lambda entry: entry.category in self._filter_categories,
filtered_entries,
)
)
return filtered_entries
def _extract_last_timestamp(
self, feed_entries: list[NswRuralFireServiceIncidentsFeedEntry]
) -> datetime | None:
"""Determine latest (newest) entry from the filtered feed."""
if feed_entries:
dates = sorted(
filter(None, [entry.publication_date for entry in feed_entries]),
reverse=True,
)
return dates[0]
return None
def _extract_from_feed(self, feed: FeatureCollection) -> dict | None:
"""Extract global metadata from feed."""
return None
|
