Merge pull request #33 from behdanisohrab/weather

weather plugin for moa

searx/plugins/weather.py

@@ -0,0 +1,192 @@
+from flask_babel import gettext
+import requests
+import re
+
+name = "MOA Weather Data Plugin"
+description = gettext("Displays weather data for a given location using the Open Meteo API, with improved formatting.")
+default_on = False
+preference_section = 'general'
+
+query_re = re.compile(r'weather\s+(?:in\s+)?(.+)', re.IGNORECASE)
+
+conditions = {
+    99: {
+        "name": gettext("Thunderstorm with heavy hail"),
+        "emoji": "⛈️"
+    },
+    96: {
+        "name": gettext("Thunderstorm with slight hail"),
+        "emoji": "⛈️"
+    },
+    95: {
+        "name": gettext("Thunderstorm"),
+        "emoji": "🌩️"
+    },
+    86: {
+        "name": gettext("Heavy snow showers"),
+        "emoji": "🌧️"
+    },
+    85: {
+        "name": gettext("Slight snow showers"),
+        "emoji": "🌧️"
+    },
+    82: {
+        "name": gettext("Violent rain showers"),
+        "emoji": "🌧️"
+    },
+    81: {
+        "name": gettext("Moderate rain showers"),
+        "emoji": "🌧️"
+    },
+    80: {
+        "name": gettext("Slight rain showers"),
+        "emoji": "🌧️"
+    },
+    77: {
+        "name": gettext("Snow grains"),
+        "emoji": "🌨️"
+    },
+    75: {
+        "name": gettext("Heavy snowfall"),
+        "emoji": "🌨️"
+    },
+    73: {
+        "name": gettext("Moderate snowfall"),
+        "emoji": "🌨️"
+    },
+    71: {
+        "name": gettext("Slight snowfall"),
+        "emoji": "🌨️"
+    },
+    67: {
+        "name": gettext("Heavy freezing rain"),
+        "emoji": "🌧️"
+    },
+    66: {
+        "name": gettext("Light freezing rain"),
+        "emoji": "🌧️"
+    },
+    65: {
+        "name": gettext("Heavy rain"),
+        "emoji": "🌧️"
+    },
+    63: {
+        "name": gettext("Moderate rain"),
+        "emoji": "🌧️"
+    },
+    61: {
+        "name": gettext("Slight rain"),
+        "emoji": "🌧️"
+    },
+    57: {
+        "name": gettext("Dense freezing drizzle"),
+        "emoji": "🌧️"
+    },
+    56: {
+        "name": gettext("Light freezing drizzle"),
+        "emoji": "🌧️"
+    },
+    55: {
+        "name": gettext("Dense drizzle"),
+        "emoji": "🌧️"
+    },
+    53: {
+        "name": gettext("Moderate drizzle"),
+        "emoji": "🌧️"
+    },
+    51: {
+        "name": gettext("Light drizzle"),
+        "emoji": "🌧️"
+    },
+    48: {
+        "name": gettext("Depositing rime fog"),
+        "emoji": "🌫️"
+    },
+    45: {
+        "name": gettext("Fog"),
+        "emoji": "🌫️"
+    },
+    3: {
+        "name": gettext("Overcast"),
+        "emoji": "☁️"
+    },
+    2: {
+        "name": gettext("Partly cloudy"),
+        "emoji": "🌥️"
+    },
+    1: {
+        "name": gettext("Mainly clear"),
+        "emoji": "🌤️"
+    },
+    0: {
+        "name": gettext("Clear"),
+        "emoji": "☀️",
+        "emoji_night": "🌙"
+    }
+}
+
+def query_location(location_query):
+    geocoding_url = f'https://nominatim.openstreetmap.org/search?format=json&q={location_query}'
+    response = requests.get(geocoding_url).json()
+    if response:
+        lat = response[0]['lat']
+        lon = response[0]['lon']
+        return {
+            'lat': lat,
+            'lon': lon,
+            'display_coords': gettext('%s °N, %s °E') % (lat, lon),
+            'display_name': response[0]['display_name']
+        }
+
+    return None
+
+def fetch_weather_data(lat, lon, params):
+    weather_url = f'https://api.open-meteo.com/v1/forecast?latitude={lat}&longitude={lon}'
+    if params:
+        weather_url += f'&{params}'
+    response = requests.get(weather_url).json()
+    return response
+
+def get_named_weather_condition(current_weather, with_emoji=False):
+    for code, condition in conditions.items():
+        if current_weather['weather_code'] >= code:
+            condition = conditions[code]
+            emoji = condition['emoji'] if current_weather.get('is_day', True) else condition.get('emoji_night', condition['emoji'])
+            return f"{emoji} {condition['name']}" if with_emoji else condition['name']
+    return gettext('Unknown')
+
+def post_search(request, search):
+    match = query_re.match(search.search_query.query)
+    if not match:
+        return True  # Continue with normal search if no match
+
+    location_query = match.group(1)
+    location = query_location(location_query)
+
+    if not location:
+        search.result_container.answers.clear()
+        search.result_container.answers['weather_error'] = {'answer': gettext('Location not found or invalid coordinates format.')}
+        return True
+
+    weather_data = fetch_weather_data(location['lat'], location['lon'], 'current=weather_code,is_day,temperature_2m,wind_speed_10m&hourly=temperature_2m,relative_humidity_2m,wind_speed_10m')
+    if 'current' in weather_data:
+        current_weather = weather_data['current']
+    else:
+        search.result_container.answers.clear()
+        search.result_container.answers['weather_error'] = {'answer': gettext('Weather data not available.')}
+        return True
+
+    search.result_container.answers.clear()
+    search.result_container.answers['weather'] = {
+        'title': gettext('Weather for %s') % f"{location['display_name']} ({location['display_coords']})",
+        'fields': [
+            {'label': gettext('Condition'), 'value': get_named_weather_condition(current_weather, with_emoji=True)},
+            {'label': gettext('Temperature'), 'value': f"{current_weather['temperature_2m']}°C"},
+            {'label': gettext('Wind Speed'), 'value': f"{current_weather['wind_speed_10m']}m/s"}
+        ],
+        'sources': [
+            {'title': 'Open Meteo', 'url': 'https://open-meteo.com/'},
+            {'title': 'OpenStreetMap', 'url': 'https://www.openstreetmap.org/'}
+        ],
+    }
+    return True