mirror of
https://github.com/searxng/searxng.git
synced 2025-10-30 18:22:31 -04:00
e16b6cb148
The ``JSONEncoder`` (``format="json"``) must perform a conversion to the
built-in types for the ``msgspec.Struct``::
if isinstance(o, msgspec.Struct):
return msgspec.to_builtins(o)
The result types are already of type ``msgspec.Struct``, so they can be
converted into built-in types.
The field types (in the result type) that were not yet of type ``msgspec.Struct``
have been converted to::
searx.weather.GeoLocation@dataclass -> msgspec.Struct
searx.weather.DateTime -> msgspec.Struct
searx.weather.Temperature -> msgspec.Struct
searx.weather.PressureUnits -> msgspec.Struct
searx.weather.WindSpeed -> msgspec.Struct
searx.weather.RelativeHumidity -> msgspec.Struct
searx.weather.Compass -> msgspec.Struct
BTW: Wherever it seemed sensible, the typing was also modernized in the modified
files.
Closes: https://github.com/searxng/searxng/issues/5250
Signed-off-by: Markus Heiser <markus.heiser@darmarit.de>
128 lines
4.0 KiB
Python
128 lines
4.0 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
"""
|
|
DuckDuckGo Weather
|
|
~~~~~~~~~~~~~~~~~~
|
|
"""
|
|
|
|
import typing as t
|
|
from json import loads
|
|
from urllib.parse import quote
|
|
|
|
from dateutil import parser as date_parser
|
|
|
|
from searx.engines.duckduckgo import fetch_traits # pylint: disable=unused-import
|
|
from searx.engines.duckduckgo import get_ddg_lang
|
|
|
|
from searx.result_types import EngineResults
|
|
from searx.extended_types import SXNG_Response
|
|
from searx import weather
|
|
|
|
|
|
about = {
|
|
"website": 'https://duckduckgo.com/',
|
|
"wikidata_id": 'Q12805',
|
|
"official_api_documentation": None,
|
|
"use_official_api": True,
|
|
"require_api_key": False,
|
|
"results": "JSON",
|
|
}
|
|
|
|
send_accept_language_header = True
|
|
|
|
# engine dependent config
|
|
categories = ["weather"]
|
|
base_url = "https://duckduckgo.com/js/spice/forecast/{query}/{lang}"
|
|
|
|
# adapted from https://gist.github.com/mikesprague/048a93b832e2862050356ca233ef4dc1
|
|
WEATHERKIT_TO_CONDITION: dict[str, weather.WeatherConditionType] = {
|
|
"BlowingDust": "fog",
|
|
"Clear": "clear sky",
|
|
"Cloudy": "cloudy",
|
|
"Foggy": "fog",
|
|
"Haze": "fog",
|
|
"MostlyClear": "clear sky",
|
|
"MostlyCloudy": "partly cloudy",
|
|
"PartlyCloudy": "partly cloudy",
|
|
"Smoky": "fog",
|
|
"Breezy": "partly cloudy",
|
|
"Windy": "partly cloudy",
|
|
"Drizzle": "light rain",
|
|
"HeavyRain": "heavy rain",
|
|
"IsolatedThunderstorms": "rain and thunder",
|
|
"Rain": "rain",
|
|
"SunShowers": "rain",
|
|
"ScatteredThunderstorms": "heavy rain and thunder",
|
|
"StrongStorms": "heavy rain and thunder",
|
|
"Thunderstorms": "rain and thunder",
|
|
"Frigid": "clear sky",
|
|
"Hail": "heavy rain",
|
|
"Hot": "clear sky",
|
|
"Flurries": "light snow",
|
|
"Sleet": "sleet",
|
|
"Snow": "light snow",
|
|
"SunFlurries": "light snow",
|
|
"WintryMix": "sleet",
|
|
"Blizzard": "heavy snow",
|
|
"BlowingSnow": "heavy snow",
|
|
"FreezingDrizzle": "light sleet",
|
|
"FreezingRain": "sleet",
|
|
"HeavySnow": "heavy snow",
|
|
"Hurricane": "rain and thunder",
|
|
"TropicalStorm": "rain and thunder",
|
|
}
|
|
|
|
|
|
def _weather_data(location: weather.GeoLocation, data: dict[str, t.Any]):
|
|
|
|
return EngineResults.types.WeatherAnswer.Item(
|
|
location=location,
|
|
temperature=weather.Temperature(val=data['temperature'], unit="°C"),
|
|
condition=WEATHERKIT_TO_CONDITION[data["conditionCode"]],
|
|
feels_like=weather.Temperature(val=data['temperatureApparent'], unit="°C"),
|
|
wind_from=weather.Compass(data["windDirection"]),
|
|
wind_speed=weather.WindSpeed(val=data["windSpeed"], unit="mi/h"),
|
|
pressure=weather.Pressure(val=data["pressure"], unit="hPa"),
|
|
humidity=weather.RelativeHumidity(data["humidity"] * 100),
|
|
cloud_cover=data["cloudCover"] * 100,
|
|
)
|
|
|
|
|
|
def request(query: str, params: dict[str, t.Any]):
|
|
|
|
eng_region = traits.get_region(params['searxng_locale'], traits.all_locale)
|
|
eng_lang = get_ddg_lang(traits, params['searxng_locale'])
|
|
|
|
# !ddw paris :es-AR --> {'ad': 'es_AR', 'ah': 'ar-es', 'l': 'ar-es'}
|
|
params['cookies']['ad'] = eng_lang
|
|
params['cookies']['ah'] = eng_region
|
|
params['cookies']['l'] = eng_region
|
|
logger.debug("cookies: %s", params['cookies'])
|
|
|
|
params["url"] = base_url.format(query=quote(query), lang=eng_lang.split('_')[0])
|
|
return params
|
|
|
|
|
|
def response(resp: SXNG_Response):
|
|
res = EngineResults()
|
|
|
|
if resp.text.strip() == "ddg_spice_forecast();":
|
|
return res
|
|
|
|
json_data = loads(resp.text[resp.text.find('\n') + 1 : resp.text.rfind('\n') - 2])
|
|
|
|
geoloc = weather.GeoLocation.by_query(resp.search_params["query"])
|
|
|
|
weather_answer = EngineResults.types.WeatherAnswer(
|
|
current=_weather_data(geoloc, json_data["currentWeather"]),
|
|
service="duckduckgo weather",
|
|
)
|
|
|
|
for forecast in json_data['forecastHourly']['hours']:
|
|
forecast_time = date_parser.parse(forecast['forecastStart'])
|
|
forecast_data = _weather_data(geoloc, forecast)
|
|
forecast_data.datetime = weather.DateTime(forecast_time)
|
|
weather_answer.forecasts.append(forecast_data)
|
|
|
|
res.add(weather_answer)
|
|
return res
|