tobi-wan-kenobi/bumblebee-status

"""
Displays information about the public IP address associated with the default route:
    * Public IP address
    * Country Name
    * Country Code
    * City Name
    * Geographic Coordinates

Left mouse click on the widget forces immediate update.
Any change to the default route will cause the widget to update.

Requirements:
    * netifaces

Parameters:
    * publicip.format: Format string (defaults to ‘{ip} ({country_code})’)
    * Available format strings - ip, country_name, country_code, city_name, coordinates

Examples:
    * bumblebee-status -m publicip -p publicip.format="{ip} ({country_code})"
    * bumblebee-status -m publicip -p publicip.format="{ip} which is in {city_name}"
    * bumblebee-status -m publicip -p publicip.format="Your packets are right here: {coordinates}"

contributed by `tfwiii <https://github.com/tfwiii>` - many thanks!
"""

import re
import threading
import netifaces
import time

import core.module
import core.widget
import core.input
import core.decorators

import util.format
import util.location
import logging

log = logging.getLogger(__name__)


class Module(core.module.Module):
    @core.decorators.every(minutes=60)
    def __init__(self, config, theme):
        super().__init__(config, theme, core.widget.Widget(self.publicip))

        self.__previous_default_route = None
        self.__current_default_route = None
        self.background = True

        # Immediate update (override default) when left click on widget
        core.input.register(self, button=core.input.LEFT_MOUSE, cmd=self.__click_update)

        # By default show: <ip> (<2 letter country code>)
        self._format = self.parameter("format", "{ip} ({country_code})")

        self.__monitor = threading.Thread(target=self.monitor, args=())
        self.__monitor.start()

    def monitor(self):
        __previous_ips = set()
        __current_ips = set()
        # Initially set to True to force an info update on first pass
        __information_changed = True

        self.update()

        while threading.main_thread().is_alive():
            __current_ips.clear()
            # Look for any changes to IP addresses
            try:
                for interface in netifaces.interfaces():
                    try:
                        __current_ips.add(netifaces.ifaddresses(interface)[2][0]['addr'])
                    except:
                        pass
            except:
            # If not ip address information found clear __current_ips
                __current_ips.clear()
            
            # If a change of any interfaces' IP then flag change
            if __current_ips.symmetric_difference(__previous_ips):
                __previous_ips = __current_ips.copy()
                __information_changed = True

            # Update if change is flagged
            if __information_changed:
                __information_changed = False
                self.update()
            
            # Throttle the calls to netifaces
            time.sleep(1)

    def publicip(self, widget):
        if widget.get("public_ip") is None:
            return "n/a"
        return self._format.format(
            ip = widget.get("public_ip", "-"),
            country_name = widget.get("country_name", "-"),
            country_code = widget.get("country_code", "-"),
            city_name = widget.get("city_name", "-"),
            coordinates = widget.get("coordinates", "-"),
        )

    def __click_update(self, event):
        util.location.reset()

    def update(self):
        widget = self.widget()

        try:
            util.location.reset()
            time.sleep(5) # wait for reset to complete before querying results

            # Fetch fresh location information
            __info = util.location.location_info()
            __raw_lat = __info["latitude"]
            __raw_lon = __info["longitude"]

            # Contstruct coordinates string if util.location has provided required info
            if isinstance(__raw_lat, float) and isinstance(__raw_lon, float):
                __lat = float("{:.2f}".format(__raw_lat))
                __lon = float("{:.2f}".format(__raw_lon))
                if __lat < 0:
                    __coords = str(__lat) + "°S"
                else:
                    __coords = str(__lat) + "°N"
                __coords += ","
                if __lon < 0:
                    __coords += str(__lon) + "°W"
                else:
                    __coords += str(__lon) + "°E"
            else:
                __coords = "Unknown" 

            # Set widget values
            widget.set("public_ip", __info["public_ip"])
            widget.set("country_name", __info["country"])
            widget.set("country_code", __info["country_code"])
            widget.set("city_name", __info["city_name"])
            widget.set("coordinates", __coords)

            # Update widget values
            core.event.trigger("update", [widget.module.id], redraw_only=True)
        except Exception as ex:
            widget.set("public_ip", None)
            logging.error(str(ex))

    def state(self, widget):
        return widget.get("state", None)


# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4