pollen.1h.py

pollen.1h.py

#!/usr/bin/env LC_ALL=en_US.UTF-8 /usr/local/bin/python3
#
# <bitbar.title>Pollenkoll</bitbar.title>
# <bitbar.version>v0.2</bitbar.version>
# <bitbar.author>lindskogen</bitbar.author>
# <bitbar.author.github>lindskogen</bitbar.author.github>
# <bitbar.desc></bitbar.desc>
# <bitbar.image></bitbar.image>
# <bitbar.dependencies>python</bitbar.dependencies>

from urllib.request import urlopen, Request
from urllib import parse
from bs4 import BeautifulSoup

BASE_URL = 'https://www.klart.se/se/pollenprognoser/stationer/'

icons = {
    "extreme": "infinity.circle.fill",
    "high": "circle.fill",
    "medium": "largecircle.fill.circle",
    "low": "smallcircle.fill.circle",
}

def get_icon(value):
    return icons.get(value) or "circle"

colors = {
    "extreme": "#f1002e",
    "high": "#e96385",
    "medium": "#d88b00",
    "low": "#91c79c",
}

def get_color(value):
    return colors.get(value) or "#cccccc"

city = "göteborg"
favvopollen = ["Björk"]

def render_row(key, value):
    print(f":{get_icon(value)}: {key} {format_value(value)}")

def format_value(value):
    return f"| symbolize=True sfcolor={get_color(value)}"

def get_information(url):
    request = Request(url)
    r = urlopen(request).read()
    soup = BeautifulSoup(r, 'html.parser')

    pollen_table = soup.find('table', {'class': 'pollen-table'}).find('tbody')
    reports = pollen_table.findAll('tr', {'class': 'row'})
    pollen_information = {}
    for report in reports:
        type = str(report.find('td', {'class': 'type'}).text).strip()
        day = report.find('td', {'class': 'day'})
        if day:
            level = str(day.find('i', {'class': 'level'})['class'][1]).replace('-', '')
        else:
            text = str(report.find('td', {'colspan': '5'}).text).strip().split()
            level = text[0] + ' ' + text[1]

        pollen_information[type] = level
    return pollen_information

information = get_information(BASE_URL + parse.quote(city))


if len(favvopollen) > 0:
    for pollen in favvopollen:
        if pollen in information:
            value = information[pollen]
            render_row(pollen, value)
else:
    print("Pollen | sfimage=aqi.low")

print("---")

for key,value in information.items():
    render_row(key, value)