app.py

import rumps
import requests
import datetime
import webbrowser


class CoronaBar(object):
    base_api_url = "https://coronavirus-19-api.herokuapp.com/countries"
    default_country = "USA"
    update_interval = 900  # seconds
    about_url = "https://github.com/duarteocarmo/coronabar"

    # APP
    def __init__(self):
        self.app = rumps.App("Corona Bar", "🦠")
        self.countries = rumps.MenuItem(title="Select Country")
        self.about = rumps.MenuItem(title="About", callback=self.open_page)

        self.country = self.default_country

        self.app.menu = [self.countries, self.about]

        country_list = self.get_country_list()
        self.setup(country_list, self.country)

    def setup(self, country_list, default_country):
        for country in country_list:
            self.countries.add(
                rumps.MenuItem(
                    title=f"{country}", callback=self.update_country_listing
                )
            )

        self.create_country_listing(
            rumps.MenuItem(title=f"{self.default_country}")
        )

    def open_page(self, sender):
        try:
            webbrowser.open(self.about_url)
        except Exception as e:
            print(str(e))
            return False

    def create_country_listing(self, country):
        data = self.get_country_data(country.title)
        for k, v in data.items():
            self.app.menu.add(rumps.MenuItem(self.string_mapper(k, v)))
        current_time = datetime.datetime.now().strftime("%H:%M")
        self.app.menu.add(rumps.MenuItem(title=f"Updated at {current_time}"))
        self.timer = rumps.Timer(self.on_update, self.update_interval)
        self.timer.start()

    def update_country_listing(self, country):
        self.country = country.title
        for k, v in self.app.menu.items():
            if k not in ["Select Country", "Quit", "About"]:
                del self.app.menu[k]

        data = self.get_country_data(country.title)
        for k, v in data.items():
            self.app.menu.insert_before(
                "Quit", rumps.MenuItem(self.string_mapper(k, v))
            )

        current_time = datetime.datetime.now().strftime("%H:%M")
        self.app.menu.insert_before(
            "Quit", rumps.MenuItem(title=f"Updated at {current_time}")
        )

    def on_update(self, sender):
        self.update_country_listing(rumps.MenuItem(title=self.country))

    def run(self):
        self.app.run()

    # DATA
    def get_country_list(self):
        response = requests.request("GET", self.base_api_url)
        data = response.json()
        country_list = [e["country"] for e in data]
        return sorted(country_list)

    def get_country_data(self, country):
        response = requests.request("GET", f"{self.base_api_url}/{country}")
        data = response.json()
        return data

    # STRING MAPPER
    def string_mapper(self, key, value):
        if type(value) == int:
            value = self.human_format(value)

        if not self.is_camel_case(key):
            return f"{key.title()}: {value}"

        elif key == "todayCases":
            return f"Cases Today: {value}"

        elif key == "todayDeaths":
            return f"Deaths Today: {value}"

        elif key == "casesPerOneMillion":
            return f"Cases per Million: {value}"

        elif key == "deathsPerOneMillion":
            return f"Deaths per Million: {value}"

        elif key == "totalTests":
            return f"Total Tests: {value}"

        elif key == "testsPerOneMillion":
            return f"Tests per Million: {value}"

        else:
            return f"{key}: {value}"

    @staticmethod
    def is_camel_case(s):
        # https://stackoverflow.com/a/10182901 thanks :)
        return s != s.lower() and s != s.upper() and "_" not in s

    @staticmethod
    def human_format(num):
        num = int(num)
        magnitude = 0
        while abs(num) >= 1000:
            magnitude += 1
            num /= 1000.0
        # add more suffixes if you need them
        return "%.2f%s" % (num, ["", "K", "M", "G", "T", "P"][magnitude])


if __name__ == "__main__":
    app = CoronaBar()
    app.run()