# coding: utf-8
"""
Weather - query various weather services for info
Copyright (C) 2010  Brian S. Stephan

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
"""

import re

from extlib import irclib
from extlib import pywapi

from urllib import quote
from urllib2 import URLError

from Module import Module

class Weather(Module):
    """Provide weather lookup services to the bot."""

    def do(self, connection, event, nick, userhost, what, admin_unlocked):
        """Query Google Weather for a location's weather."""

        match = re.search('^!weather\s+(.*)$', what)
        if match:
            query = match.group(1)
            try:
                google_weather = pywapi.get_weather_from_google(quote(query))
                city = google_weather['forecast_information']['city'].encode('utf-8')
                condition = google_weather['current_conditions']['condition'].encode('utf-8')
                temp_f = google_weather['current_conditions']['temp_f'].encode('utf-8')
                temp_c = google_weather['current_conditions']['temp_c'].encode('utf-8')
                wind = google_weather['current_conditions']['wind_condition'].encode('utf-8')
                humidity = google_weather['current_conditions']['humidity'].encode('utf-8')

                wind_speed_regex = re.compile("Wind: [NSEW]+ at ([0-9]+) mph")
                matches = wind_speed_regex.search(wind)

                if matches is not None and float(temp_f) < 50.0:
                    wind_speed = matches.group(1)
                    windchill = 35.74 + (0.6215 * float(temp_f)) - (35.75 * float(wind_speed)**0.16) + (0.4275 * float(temp_f) * float(wind_speed)**0.16)
                else:
                    windchill = ""

                weatherstr = "Current weather for " + city + ": " + condition + ". " + temp_f + "°F (" + temp_c + "°C), " + wind + ", " + humidity
                if windchill is not "":
                    weatherstr += ", Wind chill: " + str(round(windchill, 2)) + "°F"
                weatherstr += "."

                for city in google_weather['forecasts']:
                    weatherstr += " " + city['day_of_week'].encode('utf-8') + ": " + city['condition'].encode('utf-8') + ". High " + city['high'].encode('utf-8') + "°F, Low " + city['low'].encode('utf-8') + "°F."

                return weatherstr
            except URLError as e:
                return "error connecting to google weather:" + str(e)
            except IndexError as e:
                return "error in pywapi: " + str(e)

# vi:tabstop=4:expandtab:autoindent
# kate: indent-mode python;indent-width 4;replace-tabs on;