105 lines
3.6 KiB
Python
105 lines
3.6 KiB
Python
import configparser
|
|
import icalendar
|
|
import datetime
|
|
import requests
|
|
import urllib
|
|
import json
|
|
import os
|
|
|
|
def read_config():
|
|
config = configparser.ConfigParser()
|
|
config.read('config.ini')
|
|
return config
|
|
|
|
config = read_config()
|
|
current_year = datetime.datetime.now().year
|
|
|
|
# request locations supportet by the endpoint
|
|
# https://nuernberg-abfallapp.regioit.de/abfall-app-nuernberg/rest/orte
|
|
# Returns:
|
|
# [{"id":2049161,"name":"Nürnberg"}]
|
|
req = requests.get("https://nuernberg-abfallapp.regioit.de/abfall-app-nuernberg/rest/orte")
|
|
locations = req.json()
|
|
for location in locations:
|
|
if location.get("name") == config["address"]["city"]:
|
|
nbg_id = location.get("id")
|
|
|
|
# request all known streets for a given location id (can be obtained by previous request)
|
|
req = requests.get("https://nuernberg-abfallapp.regioit.de/abfall-app-nuernberg/rest/orte/{location_id}/strassen".format(location_id=nbg_id))
|
|
streets = req.json()
|
|
|
|
for street in streets:
|
|
if street.get("name") == config["address"]["street"]:
|
|
street_id = street.get("id")
|
|
|
|
# request all known house numbers for a given street id (can be obtained by previous request)
|
|
req = requests.get("https://nuernberg-abfallapp.regioit.de/abfall-app-nuernberg/rest/strassen/{street_id}".format(street_id=street_id))
|
|
housenumbers = req.json().get("hausNrList", [])
|
|
|
|
for number in housenumbers:
|
|
if number.get("nr") == config["address"]["house_number"]:
|
|
housenumber_id = number.get("id")
|
|
|
|
# request trash types picked up at that location
|
|
req = requests.get("https://nuernberg-abfallapp.regioit.de/abfall-app-nuernberg/rest/hausnummern/{housenumber_id}/fraktionen".format(housenumber_id=housenumber_id))
|
|
trash_types = req.json()
|
|
|
|
for trash_type in trash_types:
|
|
print(trash_type.get("name"))
|
|
|
|
#requested_trash_types = list(map(lambda x: str(x.get("id")), trash_types))
|
|
|
|
query = {
|
|
"format": ["ics"],
|
|
"jahr": [current_year],
|
|
"ort": ["Nürnberg"],
|
|
"strasse": [street_id],
|
|
"hnr": [housenumber_id],
|
|
"fraktion": []
|
|
}
|
|
|
|
radicale_props = {
|
|
"C:supported-calendar-component-set": "VEVENT",
|
|
"D:displayname": None,
|
|
"ICAL:calendar-color": "#000000ff",
|
|
"tag": "VCALENDAR"
|
|
}
|
|
|
|
target_path = config.get("calendar", "target", fallback=os.path.join(".", "calendars"))
|
|
if not os.path.exists(target_path):
|
|
os.makedirs(target_path)
|
|
|
|
for trash_type in trash_types:
|
|
radicale_props["D:displayname"] = trash_type.get("name")
|
|
radicale_props["ICAL:calendar-color"] = "#{}ff".format(trash_type.get("farbeRgb", "000000"))
|
|
folder_name = trash_type.get("name").replace("/", "_")
|
|
trash_type_folder = os.path.join(target_path, folder_name)
|
|
if not os.path.exists(trash_type_folder):
|
|
os.makedirs(trash_type_folder)
|
|
|
|
radicale_prop_file = os.path.join(trash_type_folder, ".Radicale.props")
|
|
with open(radicale_prop_file, "w") as f:
|
|
f.write(json.dumps(radicale_props))
|
|
|
|
query["fraktion"] = [str(trash_type["id"])]
|
|
query_string = urllib.parse.urlencode(query, doseq=True)
|
|
query_url = urllib.parse.urlunparse(("https", "abfallkalender.regioit.de", "/kalender-nuernberg/downloadfile.jsp", "", query_string, ""))
|
|
req = requests.get(query_url)
|
|
ical = req.text
|
|
|
|
cal = icalendar.Calendar.from_ical(ical)
|
|
for event in cal.walk():
|
|
if isinstance(event, icalendar.cal.Event):
|
|
uid = event.get("UID")
|
|
new_cal = icalendar.Calendar()
|
|
new_cal.add('prodid', 'Trash dates')
|
|
new_cal.add('version', '2.0')
|
|
new_cal.add_component(event)
|
|
ical_name = "{}.ical".format(uid)
|
|
ical_path = os.path.join(trash_type_folder, ical_name)
|
|
with open(ical_path, "wb") as single_file:
|
|
single_file.write(new_cal.to_ical())
|
|
else:
|
|
pass
|
|
|