[modules] Reimport weather module

modules/core/weather.py

@@ -0,0 +1,138 @@
+# -*- coding: UTF-8 -*-
+# pylint: disable=C0111,R0903
+
+"""Displays the temperature on the current location based on the ip
+
+Requires the following python packages:
+    * requests
+
+Parameters:
+    * weather.location: Set location, defaults to 'auto' for getting location from http://ipinfo.io
+                        If set to a comma-separated list, left-click and right-click can be used to rotate the locations.
+                        Locations should be city names or city ids.
+    * weather.unit: metric (default), kelvin, imperial
+    * weather.showcity: If set to true, show location information, otherwise hide it (defaults to true)
+    * weather.showminmax: If set to true, show the minimum and maximum temperature, otherwise hide it (defaults to false)
+    * weather.apikey: API key from http://api.openweathermap.org
+"""
+
+import bumblebee.input
+import bumblebee.output
+import bumblebee.engine
+import re
+
+try:
+    import requests
+    from requests.exceptions import RequestException
+except ImportError:
+    pass
+
+class Module(bumblebee.engine.Module):
+    def __init__(self, engine, config):
+        super(Module, self).__init__(engine, config,
+            bumblebee.output.Widget(full_text=self.output)
+        )
+        self._temperature = 0
+        self._apikey = self.parameter("apikey", "af7bfe22287c652d032a3064ffa44088")
+        self._location = self.parameter("location", "auto")
+        if "," in self._location:
+            self._location = self._location.split(",")
+        else:
+            self._location = [self._location]
+        self._index = 0
+        self._showcity = bumblebee.util.asbool(self.parameter("showcity", True))
+        self._showminmax = bumblebee.util.asbool(self.parameter("showminmax", False))
+        self._unit = self.parameter("unit", "metric")
+        self._valid = False
+        self.interval_factor(60)
+        self.interval(15)
+
+        engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE, cmd=self._next_location)
+        engine.input.register_callback(self, button=bumblebee.input.RIGHT_MOUSE, cmd=self._prev_location)
+
+    def _next_location(self, event):
+        self._index = 0 if self._index >= len(self._location) - 1 else self._index + 1
+        self.update(self.widgets())
+
+    def _prev_location(self, event):
+        self._index = len(self._location)-1 if self._index <= 0 else self._index - 1
+        self.update(self.widgets())
+
+    def _unit_suffix(self):
+        if self._unit == "metric":
+            return "C"
+        if self._unit == "kelvin":
+            return "K"
+        if self._unit == "imperial":
+            return "F"
+        return ""
+
+    def temperature(self):
+        return u"{}°{}".format(self._temperature, self._unit_suffix())
+
+    def tempmin(self):
+        return u"{}°{}".format(self._tempmin, self._unit_suffix())
+
+    def tempmax(self):
+        return u"{}°{}".format(self._tempmax, self._unit_suffix())
+
+    def city(self):
+        city = re.sub('[_-]', ' ', self._city)
+        return u"{} ".format(city)
+
+    def output(self, widget):
+        if not self._valid:
+            return u"?"
+        if self._showminmax:
+            self._showcity=False
+            return self.city() + self.temperature() + "  Hi:" + self.tempmax() + "  Lo:" + self.tempmin()
+        elif self._showcity:
+            return self.city() + self.temperature()
+        else:
+            return self.temperature()
+
+    def state(self, widget):
+        if self._valid:
+            if "thunderstorm" in self._weather:
+                return ['thunder']
+            elif "drizzle" in self._weather:
+                return ['rain']
+            elif "rain" in self._weather:
+                return ['rain']
+            elif "snow" in self._weather:
+                return ['snow']
+            elif "sleet" in self._weather:
+                return ['sleet']
+            elif "clear" in self._weather:
+                return ['clear']
+            elif "cloud" in self._weather:
+                return ['clouds']
+            else:
+                return []
+
+        return []
+
+    def update(self, widgets):
+        try:
+            weather_url = "http://api.openweathermap.org/data/2.5/weather?appid={}".format(self._apikey)
+            weather_url = "{}&units={}".format(weather_url, self._unit)
+            if self._location[self._index] == "auto":
+                location_url = "http://ipinfo.io/json"
+                location = requests.get(location_url).json()
+                coord = location["loc"].split(",")
+                weather_url = "{url}&lat={lat}&lon={lon}".format(url=weather_url, lat=coord[0], lon=coord[1])
+            elif self._location[self._index].isdigit():
+                weather_url = "{url}&id={id}".format(url=weather_url, id=self._location[self._index])
+            else:
+                weather_url = "{url}&q={city}".format(url=weather_url, city=self._location[self._index])
+            weather = requests.get(weather_url).json()
+            self._city = weather['name']
+            self._temperature = int(weather['main']['temp'])
+            self._tempmin = int(weather['main']['temp_min'])
+            self._tempmax = int(weather['main']['temp_max'])
+            self._weather = weather['weather'][0]['main'].lower()
+            self._valid = True
+        except Exception:
+            self._valid = False
+
+# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4