init

gismeteo/parser.py

@@ -0,0 +1,118 @@
+import datetime
+from typing import Dict, Iterable, List
+
+import dateparser
+from bs4 import Tag
+
+from .core import BaseWidgetParser, RowParser
+
+
+class OneDayParser(BaseWidgetParser):
+    SELECT = ".widget.widget-oneday .widget-items"
+
+
+class DateParser(RowParser[datetime.datetime]):
+    KEY = "date"
+
+    def parse_row(self, tag: Tag) -> Iterable[datetime.datetime]:
+        date_str = (
+            tag.select_one(".widget-row.widget-row-datetime-date > .row-item")
+            .find(text=True, recursive=False)
+            .text
+        )
+        date = dateparser.parse(date_str, languages=["ru"])
+        for item in tag.select(".widget-row.widget-row-datetime-time > .row-item"):
+            time_str = item.text
+            time = dateparser.parse(time_str, languages=["ru"])
+            time = time.replace(year=date.year, month=date.month, day=date.day)
+            yield time
+
+
+class CloudnessParser(RowParser[str]):
+    KEY = "cloudness"
+
+    def parse_row(self, tag: Tag) -> Iterable[str]:
+        for item in tag.select(".widget-row[data-row=icon-tooltip] > .row-item"):
+            yield item.attrs["data-tooltip"]
+
+
+class TemperatureParser(RowParser[int]):
+    KEY = "temperature"
+
+    def parse_row(self, tag: Tag) -> Iterable[int]:
+        for item in tag.select(
+            ".widget-row-chart[data-row=temperature-air] > .chart > .values > .value > temperature-value"
+        ):
+            yield int(item.attrs["value"])
+
+
+class WindSpeedParser(RowParser[int]):
+    KEY = "wind_speed"
+
+    def parse_row(self, tag: Tag) -> Iterable[int]:
+        for item in tag.select(
+            ".widget-row[data-row=wind-speed] > .row-item > speed-value"
+        ):
+            yield int(item.attrs["value"])
+
+
+class WindGustParser(RowParser[int]):
+    KEY = "wind_gust"
+
+    def parse_row(self, tag: Tag) -> Iterable[int]:
+        for item in tag.select(".widget-row[data-row=wind-gust] > .row-item"):
+            value = item.select_one("speed-value")
+            yield int(value.attrs["value"]) if value else 0
+
+
+class WindDirectionParser(RowParser[str]):
+    KEY = "wind_direction"
+
+    def parse_row(self, tag: Tag) -> Iterable[str]:
+        for item in tag.select(
+            ".widget-row[data-row=wind-direction] > .row-item > .direction"
+        ):
+            yield item.text
+
+
+class WindPrecipitationParser(RowParser[float]):
+    KEY = "precipitation"
+
+    def parse_row(self, tag: Tag) -> Iterable[float]:
+        for item in tag.select(
+            ".widget-row[data-row=precipitation-bars] > .row-item > .item-unit"
+        ):
+            yield float(item.text.replace(",", "."))
+
+
+class PressureParser(RowParser[int]):
+    KEY = "pressure"
+
+    def parse_row(self, tag: Tag) -> Iterable[int]:
+        for item in tag.select(
+            ".widget-row-chart[data-row=pressure] > .chart > .values > .value > pressure-value"
+        ):
+            yield int(item.attrs["value"])
+
+
+class HumidityParser(RowParser[int]):
+    KEY = "humidity"
+
+    def parse_row(self, tag: Tag) -> Iterable[int]:
+        for item in tag.select(".widget-row[data-row=humidity] > .row-item"):
+            yield int(item.text)
+
+
+ROW_PARSERS: List[RowParser] = [
+    DateParser(),
+    CloudnessParser(),
+    TemperatureParser(),
+    WindSpeedParser(),
+    WindGustParser(),
+    WindDirectionParser(),
+    WindPrecipitationParser(),
+    PressureParser(),
+    HumidityParser(),
+]
+
+ROW_PARSERS_MAP: Dict[str, RowParser] = {parser.KEY: parser for parser in ROW_PARSERS}