python-openpublictransport 0.1.6__tar.gz → 0.1.7__tar.gz

{python_openpublictransport-0.1.6 → python_openpublictransport-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-openpublictransport
-Version: 0.1.6
+Version: 0.1.7
 Summary: Python library for public transport APIs (EFA, OTP2, TRIAS, FPTF, HAFAS, GTFS-RT)
 License: MIT
 Requires-Python: >=3.11

{python_openpublictransport-0.1.6 → python_openpublictransport-0.1.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "python-openpublictransport"
-version = "0.1.6"
+version = "0.1.7"
 description = "Python library for public transport APIs (EFA, OTP2, TRIAS, FPTF, HAFAS, GTFS-RT)"
 readme = "README.md"
 license = { text = "MIT" }

{python_openpublictransport-0.1.6 → python_openpublictransport-0.1.7}/src/openpublictransport/const.py

@@ -44,6 +44,8 @@ KVV_TRANSPORTATION_TYPES = {
     1: "train",
     4: "tram",
     5: "bus",
+    6: "bus",  # Regionalbus (e.g. KVV line 106)
+    7: "bus",  # Schnellbus
 }
 
 HVV_TRANSPORTATION_TYPES = {

{python_openpublictransport-0.1.6 → python_openpublictransport-0.1.7}/src/openpublictransport/providers/efa_base.py

@@ -17,6 +17,28 @@ from .base import BaseProvider
 _LOGGER = logging.getLogger(__name__)
 
 
+def _transport_type_from_name(name: str) -> str:
+    """Best-effort transport type from an EFA product name.
+
+    Used as a fallback when the numeric product class is not in a provider's
+    mapping (e.g. KVV Regionalbus class 6), so unmapped classes are not dropped
+    as "unknown". Checks subway/tram before the generic "bahn" → train.
+    """
+    n = (name or "").lower()
+    if "u-bahn" in n or "u_bahn" in n or "ubahn" in n or "subway" in n or "metro" in n:
+        return "subway"
+    if "straßenbahn" in n or "strassenbahn" in n or "stadtbahn" in n or "tram" in n:
+        return "tram"
+    if "fähre" in n or "faehre" in n or "schiff" in n or "ferry" in n:
+        return "ferry"
+    if "bus" in n or "ast" in n or "ruf" in n or "ersatz" in n:
+        # Stadt-/Regional-/Schnell-/Nachtbus, AST, Rufbus, (Schienen-)Ersatzverkehr
+        return "bus"
+    if "bahn" in n or "zug" in n or "train" in n:  # S-Bahn, Regionalbahn, Zug, …
+        return "train"
+    return "unknown"
+
+
 class EFABaseProvider(BaseProvider):
     """Base class for all EFA-based providers (VRR, KVV, HVV, MVV, etc.)."""
 
@@ -122,10 +144,14 @@ class EFABaseProvider(BaseProvider):
             product = transportation.get("product", {})
             product_class = product.get("class", 0)
             transport_type = type_mapping.get(product_class, "unknown")
+            if transport_type == "unknown":
+                # Fall back to the product name so unmapped classes aren't dropped.
+                transport_type = _transport_type_from_name(product.get("name", ""))
             if transport_type == "unknown":
                 _LOGGER.debug(
-                    "Unknown transport class %s for line %s",
+                    "Unknown transport class %s / name %r for line %s",
                     product_class,
+                    product.get("name"),
                     transportation.get("number", "unknown"),
                 )
             return transport_type

{python_openpublictransport-0.1.6 → python_openpublictransport-0.1.7}/src/python_openpublictransport.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-openpublictransport
-Version: 0.1.6
+Version: 0.1.7
 Summary: Python library for public transport APIs (EFA, OTP2, TRIAS, FPTF, HAFAS, GTFS-RT)
 License: MIT
 Requires-Python: >=3.11