from WeatherForecast import WeatherForecast
from WeatherInterface import WeatherInterface
import pyowm
from datetime import datetime
from settings import units

class OwmForecasts (WeatherInterface):
    """Fetches weather through the Openweathermap-api."""
    def __init__ (self, location, api_key, paid_api=False):
        self.subscription = "pro" if paid_api else None
        self.api_key = api_key
        self.units = units
        self.location = location
        self.api = pyowm.OWM(self.api_key, subscription_type=self.subscription)

    def is_available (self):
        try:
            return self.api.is_API_online()
        except:
            return False
    
    def reload(self):
        pass

    def get_today_forecast (self, location=None):
        if self.is_available() is False:
            return None
        
        try:
            location = self.location if location is None else location

            observation = self.api.weather_at_place(location)
            weather = observation.get_weather()

            return self.__get_forecast_from_weather__(weather, location=location)
        except:
            return None

    def get_forecast_in_days (self, offset_by_days, location=None):
        if offset_by_days is 0:
            return self.get_today_forecast(location)
        
        if self.is_available() is False:
            return None
                
        location = self.location if location is None else location
        try:
            forecast = self.api.daily_forecast(location, limit=offset_by_days)
            target_weather = forecast.get_forecast().get_weathers()[-1]

            return self.__get_forecast_from_weather__(target_weather, location=location)
        except: # only allowed for paid membership
            return None

    def __get_forecast_from_weather__ (self, weather, location):
        forecast_object = WeatherForecast()
        forecast_object.units = self.units
        forecast_object.fetch_datetime = datetime.now()
        forecast_object.location = location
        forecast_object.datetime = weather.get_reference_time(timeformat='date')

        forecast_object.icon = weather.get_weather_icon_name()
        forecast_object.air_humidity = str(weather.get_humidity())
        forecast_object.clouds = str(weather.get_clouds())
        forecast_object.short_description = str(weather.get_status())
        forecast_object.detailed_description = str(weather.get_detailed_status())
        forecast_object.air_pressure = str(weather.get_pressure()['press'])

        if forecast_object.units == "metric":
            forecast_object.air_temperature = str(int(weather.get_temperature(unit='celsius')['temp']))
            forecast_object.wind_speed = str(int(weather.get_wind()['speed']))

        if forecast_object.units == "imperial":
            forecast_object.air_temperature = str(int(weather.get_temperature('fahrenheit')['temp']))
            forecast_object.wind_speed = str(int(weather.get_wind()['speed'] * 0.621))

        forecast_object.sunrise = datetime.fromtimestamp(int(weather.get_sunrise_time(timeformat='unix')))
        forecast_object.sunset = datetime.fromtimestamp(int(weather.get_sunset_time(timeformat='unix')))

        return forecast_object