diff --git a/main.py b/main.py index ff74058..a217973 100644 --- a/main.py +++ b/main.py @@ -1,4 +1,6 @@ import logging +import json +from pathlib import Path from overpass import fetch_overpass from queries.bergbahn import BERGBAHN_QUERY from queries.restaurant import RESTAURANT_QUERY @@ -24,10 +26,31 @@ logger = logging.getLogger(__name__) # --------------------------------------------------------------------------- BBOXEN = { - "davos": (46.72, 9.70, 46.92, 10.00), - "schweiz": (45.8, 5.9, 47.8, 10.5), + "SW": (45.8, 5.9, 46.8, 8.2), + "SO": (45.8, 8.2, 46.8, 10.5), + "NW": (46.8, 5.9, 47.8, 8.2), + "NO": (46.8, 8.2, 47.8, 10.5) } +# BBOXEN = { +# 1: (45.8, 5.9, 46.4667, 7.4333), +# 2: (45.8, 7.4333, 46.4667, 8.9667), +# 3: (45.8, 8.9667, 46.4667, 10.5), +# 4: (46.4667, 5.9, 47.1333, 7.4333), +# 5: (46.4667, 7.4333, 47.1333, 8.9667), +# 6: (46.4667, 8.9667, 47.1333, 10.5), +# 7: (47.1333, 5.9, 47.8, 7.4333), +# 8: (47.1333, 7.4333, 47.8, 8.9667), +# 9: (47.1333, 8.9667, 47.8, 10.5) +# } + +# BBOXEN = { +# 1: (45.8, 5.9, 46.3, 7.05), 2: (45.8, 7.05, 46.3, 8.2), 3: (45.8, 8.2, 46.3, 9.35), 4: (45.8, 9.35, 46.3, 10.5), +# 5: (46.3, 5.9, 46.8, 7.05), 6: (46.3, 7.05, 46.8, 8.2), 7: (46.3, 8.2, 46.8, 9.35), 8: (46.3, 9.35, 46.8, 10.5), +# 9: (46.8, 5.9, 47.3, 7.05), 10: (46.8, 7.05, 47.3, 8.2), 11: (46.8, 8.2, 47.3, 9.35), 12: (46.8, 9.35, 47.3, 10.5), +# 13: (47.3, 5.9, 47.8, 7.05), 14: (47.3, 7.05, 47.8, 8.2), 15: (47.3, 8.2, 47.8, 9.35), 16: (47.3, 9.35, 47.8, 10.5) +# } + QUERY = {"bergbahn": BERGBAHN_QUERY} @@ -36,6 +59,8 @@ QUERY = {"bergbahn": BERGBAHN_QUERY} # --------------------------------------------------------------------------- def main() -> None: + overall = [] + errors = [] query_name = list(QUERY.keys())[0] for name, bbox in BBOXEN.items(): @@ -43,20 +68,25 @@ def main() -> None: try: result = fetch_overpass(overpass_query=QUERY.get(query_name,""), bbox=bbox) except RuntimeError as e: + errors.append(name) logger.error(f"API-Fehler bei '{name}': {e}") continue except requests.Timeout: + errors.append(name) logger.error(f"Timeout bei '{name}' — bbox zu gross oder Server überlastet") continue elements = result.get("elements", []) logger.info(f"'{name}': {len(elements)} Elemente gefunden") + overall.extend(elements) - for el in elements: - el_type = el.get("type", "?") - tags = el.get("tags", {}) - el_name = tags.get("name", "–") - logger.debug(f" [{el_type}] {el_name}") + logger.info(f"Total: {len(overall)} Elemente gefunden") + if errors: + logger.info(f"Fehler in Fragmenten: {errors}") + + # Ergebnisse speichern + with Path("overpass_results.json ").open("w", encoding="utf -8") as file: + json.dump(overall, file, indent=2, ensure_ascii=False) logger.info("Fertig.") @@ -66,17 +96,14 @@ if __name__ == "__main__": # Was ist passiert? - # * Wir haben den Logger eingebaut und können auf verschiedenen Levels loggen - # * Wir haben ein neues Model 'queries' erstellt und darin die Queries für bergbahn und restaurant abgelegt. - # -> Prinzip der Trennung von Daten und Logik! + # * Wir haben die Resultate als .json auf unserer Festplatte gespeichert + # * Wir haben die Schweiz (bbox) verschiedener Grössen aufgesplittet # TASK: - # * bbox für Schweiz scheint zu gross und wirft einen error ... Lösungsmöglichkeiten? - # -> Wir können die Schweiz (Koordinaten) in Unterregionen aufsplitten. Macht das bitte. - # -> entfernt dazu die bbox für 'davos', nehmt die 'schweiz' und splittet sie in 4, 9 oder 16 - # Koordinaten-Tuples auf. - # * Speichert und loggt in welchen Koordinaten-Tuples ein Fehler auftritt (gebt am Schluss eine Zusammenfassung - # dieser fehlerhaften Queries aus) - # * Baut eine neue Liste für die erfolgreichen Resultate und speichert diese Liste am Schluss als .json auf - # eurer Festplatte \ No newline at end of file + # * Die Speicherung der Ergebnisse liegt nun in unserer Hauptlogik -> wir lagern diese in ein neues Modul 'utils' + # als eigenständige Funktion 'store_to_disk()' aus. + # * Dokumentiert diese Funktion (docstrings) und fügt Typehints dazu. + # * Fangt in der Hauptfunktion einen OSError ab, wenn etwas beim Speichern nicht klappt + # * Optional könnt ihr der Funktion 'store_to_disk()' auch einen beliebigen Pfad mitgeben, wo das File gespeichert + # werden soll (Default wäre im aktuellen Verzeichnis) \ No newline at end of file