diff --git a/bumblebee/modules/publicip.py b/bumblebee/modules/publicip.py index bb2f4ab..99899fe 100644 --- a/bumblebee/modules/publicip.py +++ b/bumblebee/modules/publicip.py @@ -1,9 +1,15 @@ """Displays public IP address +Requires the following python packages: + * requests + +Parameters: + * publicip.region: us-central (default), us-east, us-west, uk, de, pl, nl + * publicip.service: web address that returns plaintext ip address (ex. "http://l2.io/ip") """ try: - from urllib2 import urlopen + from requests import get except ImportError: pass @@ -15,7 +21,15 @@ class Module(bumblebee.engine.Module): super(Module, self).__init__(engine, config, bumblebee.output.Widget(full_text=self.public_ip) ) - + self._avail_regions = {"us-east":"http://l2.io/ip", + "us-central":"http://whatismyip.akamai.com", + "us-west":"http://ipv4bot.whatismyipaddress.com", + "pl":"http://ip.42.pl/raw", + "de":"http://myexternalip.com/raw", + "nl":"http://tnx.nl/ip", + "uk":"http://ident.me"} + self._region = self.parameter("region", "us-central") + self._service = self.parameter("service", "") self._ip = "" @@ -24,7 +38,11 @@ class Module(bumblebee.engine.Module): def update(self, widgets): try: - self._ip = urlopen("http://ip.42.pl/raw").read() + if self._service: + self.address = self._service + else: + self.address = self._avail_regions[self._region] + self._ip = get(self.address).text.rstrip() except Exception: - self._ip = "Not Connected" + self._ip = "No Connection"