From c5a155561806bfb4997f5987d8e94d1dd31d672e Mon Sep 17 00:00:00 2001 From: Robert Sacks Date: Mon, 5 Jun 2017 03:30:23 -0400 Subject: [PATCH] [modules/publicip] Added region option to publicip module and switched to 'requests' package --- bumblebee/modules/publicip.py | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) diff --git a/bumblebee/modules/publicip.py b/bumblebee/modules/publicip.py index bb2f4ab..f6576eb 100644 --- a/bumblebee/modules/publicip.py +++ b/bumblebee/modules/publicip.py @@ -1,9 +1,14 @@ """Displays public IP address +Requires the following python packages: + * requests + +Parameters: + * publicip.region: us-central (default), us-east, us-west, uk, de, pl, nl """ try: - from urllib2 import urlopen + from requests import get except ImportError: pass @@ -15,7 +20,14 @@ 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._ip = "" @@ -24,7 +36,7 @@ class Module(bumblebee.engine.Module): def update(self, widgets): try: - self._ip = urlopen("http://ip.42.pl/raw").read() + self._ip = get(self._avail_regions[self._region]).text.rstrip() except Exception: self._ip = "Not Connected"