apyefa 0.0.2__py3-none-any.whl → 0.0.4__py3-none-any.whl

apyefa/client.py

@@ -4,6 +4,7 @@ import aiohttp
 
 from apyefa.commands import (
     Command,
+    CommandAdditionalInfo,
     CommandDepartures,
     CommandServingLines,
     CommandStopFinder,
@@ -163,7 +164,7 @@ class EfaClient:
         return command.parse(response)
 
     async def lines_by_location(self, location: str | Location) -> list[Line]:
-        """Search for lines that pass `location`. Location can be location ID like `de:08111:6221` or a Location object
+        """Search for lines that pass `location`. Location can be location ID like `de:08111:6221` or a `Location` object
 
         Args:
             location (str | Location): Location
@@ -193,6 +194,15 @@ class EfaClient:
     async def locations_by_line(self, line: str | Line) -> list[Location]:
         raise NotImplementedError
 
+    async def additional_info(self):
+        _LOGGER.info("Request additional info")
+
+        command = CommandAdditionalInfo()
+
+        response = await self._run_query(self._build_url(command))
+
+        return command.parse(response)
+
     async def _run_query(self, query: str) -> str:
         _LOGGER.info(f"Run query {query}")
 

apyefa/commands/__init__.py

@@ -1,4 +1,5 @@
 from .command import Command
+from .command_add_info import CommandAdditionalInfo
 from .command_departures import CommandDepartures
 from .command_serving_lines import CommandServingLines
 from .command_stop_finder import CommandStopFinder
@@ -8,6 +9,7 @@ from .command_trip import CommandTrip
 __all__ = [
     "Command",
     "CommandDepartures",
+    "CommandAdditionalInfo",
     "CommandStopFinder",
     "CommandSystemInfo",
     "CommandTrip",

apyefa/commands/command_add_info.py

@@ -0,0 +1,51 @@
+import logging
+
+from voluptuous import Any, Optional, Required, Schema
+
+from apyefa.commands.command import Command
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class CommandAdditionalInfo(Command):
+    def __init__(self) -> None:
+        super().__init__("XML_ADDINFO_REQUEST", "addinfo")
+
+    def parse(self, data: dict):
+        data = self._get_parser().parse(data)
+
+        result = []
+
+        return result
+
+    def _get_params_schema(self) -> Schema:
+        return Schema(
+            {
+                Required("outputFormat", default="rapidJSON"): Any("rapidJSON"),
+                Required("coordOutputFormat", default="WGS84"): Any("WGS84"),
+                Optional("filterDateValid"): str,
+                Optional("filterDateValidDay"): str,
+                Optional("filterDateValidMonth"): str,
+                Optional("filterDateValidYear"): str,
+                Optional("filterDateValidComponentsActive"): Any("0", "1", 0, 1),
+                Optional("filterPublicationStatus"): Any("current", "history"),
+                Optional("filterValidIntervalStart"): str,
+                Optional("filterValidIntervalEnd"): str,
+                Optional("filterOMC"): str,
+                Optional("filterOMC_PlaceID"): str,
+                Optional("filterLineNumberIntervalStart"): str,
+                Optional("filterLineNumberIntervalEnd"): str,
+                Optional("filterMOTType"): str,
+                Optional("filterPNLineDir"): str,
+                Optional("filterPNLineSub"): str,
+                Optional("itdLPxx_selLine"): str,
+                Optional("itdLPxx_selOperator"): str,
+                Optional("itdLPxx_selStop"): str,
+                Optional("line"): str,
+                Optional("filterInfoID"): str,
+                Optional("filterInfoType"): str,
+                Optional("filterPriority"): str,
+                Optional("filterProviderCode"): str,
+                Optional("filterSourceSystemName"): str,
+            }
+        )

apyefa/commands/command_serving_lines.py

@@ -48,7 +48,7 @@ class CommandServingLines(Command):
                 Optional("name_sl"): str,
                 # mode 'line'
                 Optional("lineName"): str,
-                Optional("lineReqType"): int,
+                Optional("lineReqType"): list[0, 2, 4, 8, 16],
                 Optional("mergeDir"): Any("0", "1", 0, 1),
                 Optional("lsShowTrainsExplicit"): Any("0", "1", 0, 1),
                 Optional("line"): str,

apyefa/commands/command_stop_finder.py

@@ -41,6 +41,10 @@ class CommandStopFinder(Command):
                 Optional("anyResSort_sf"): str,
                 Optional("anyObjFilter_sf"): int,
                 Optional("doNotSearchForStops_sf"): Any("0", "1", 0, 1),
+                Optional("locationInfoActive_sf"): Any("0", "1", 0, 1),
+                Optional("useHouseNumberList_sf"): Any("0", "1", 0, 1),
+                Optional("useLocalityMainStop"): Any("0", "1", 0, 1),
+                Optional("prMinQu"): int,
                 Optional("anyObjFilter_origin"): Range(
                     min=0, max=sum([x.value for x in LocationFilter])
                 ),

apyefa/data_classes.py

@@ -21,18 +21,48 @@ class LocationType(StrEnum):
     UNKNOWN = "unknown"
 
 
+class InfoType(StrEnum):
+    AREA_INFO = "areaInfo"
+    STOP_INFO = "stopInfo"
+    STOP_BLOCKING = "stopBlocking"
+    LINE_INFO = "lineInfo"
+    LINE_BLOCKING = "lineBlocking"
+    ROUTE_INFO = "routeInfo"
+    ROUTE_BLOCKING = "routeBlocking"
+    GENERAL_INFO = "generalInfo"
+    BANNER_INFO = "bannerInfo"
+    TRAFFIC_INFO = "trafficInformation"
+
+
+class InfoPriority(StrEnum):
+    VERY_LOW = "veryLow"
+    LOW = "low"
+    NORMAL = "normal"
+    HIGH = "high"
+    VERY_HIGH = "veryHigh"
+
+
 class TransportType(IntEnum):
-    RAIL = 0  # RB
+    TRAIN = 0  # Zug
     SUBURBAN = 1  # S-Bahn
     SUBWAY = 2  # U-Bahn
     CITY_RAIL = 3  # Stadtbahn
     TRAM = 4  # Straßenbahn
-    BUS = 5  # Bus
-    RBUS = 6  # Regional Bus
+    CITY_BUS = 5  # Stadtbus
+    REGIONAL_BUS = 6  # Regionalbus
     EXPRESS_BUS = 7  # Schnellbus
-    CABLE_TRAM = 8  # Seilbahn
+    CABLE_RAIL = 8  # Seilbahn
     FERRY = 9  # Schief
     AST = 10  # Anruf-Sammel-Taxi
+    SUSPENSION_RAIL = 11  # Schwebebahn
+    AIRPLANE = 12  # Flugzeug
+    REGIONAL_TRAIN = 13  # Reginalzug (z.B. IRE, RE und RB)
+    NATIONAL_TRAIN = 14  # Nationaler Zug (z.B. IR und D)
+    INTERNATINAL_TRAIN = 15  # Internationaler Zug (z.B. IC und EC)
+    HIGH_SPEED_TRAIN = 16  # Hochgeschwindigkeitzüge (z.B. ICE)
+    RAIL_REPLACEMENT_TRANSPORT = 17  # Schienenersatzverkehr
+    SHUTTLE_TRAIN = 18  # Schuttlezug
+    CITIZEN_BUS = 19  # Bürgerbus
 
 
 class LocationFilter(IntEnum):
@@ -63,7 +93,8 @@ SCHEMA_PROPERTIES = vol.Schema(
         vol.Optional("area"): str,
         vol.Optional("platform"): str,
         vol.Optional("platformName"): str,
-    }
+    },
+    extra=vol.ALLOW_EXTRA,
 )
 
 SCHEMA_LINE_PROPERTIES: Final = vol.Schema(
@@ -143,14 +174,14 @@ SCHEMA_TRANSPORTATION: Final = vol.Schema(
     {
         vol.Required("id"): str,
         vol.Required("name"): str,
-        vol.Required("disassembledName"): str,
         vol.Required("number"): str,
-        vol.Required("description"): str,
         vol.Required("product"): SCHEMA_PRODUCT,
+        vol.Optional("description"): str,
         vol.Optional("operator"): SCHEMA_OPERATOR,
         vol.Optional("destination"): SCHEMA_LOCATION,
         vol.Optional("origin"): SCHEMA_LOCATION,
         vol.Optional("properties"): dict,
+        vol.Optional("disassembledName"): str,
     }
 )
 
@@ -181,7 +212,8 @@ SCHEMA_DEPARTURE: Final = vol.Schema(
         vol.Required("transportation"): SCHEMA_TRANSPORTATION,
         vol.Optional("infos"): list,
         vol.Optional("hints"): list,
-    }
+    },
+    extra=vol.ALLOW_EXTRA,
 )
 
 
@@ -367,7 +399,8 @@ class Line(_Base):
         # number = data.get("number")
         description = data.get("description")
         product = TransportType(data.get("product").get("class"))
-        operator = data.get("operator").get("name")
+        # operator = data.get("operator", None).get("name", None)
+        operator = "None"
         destination = Location.from_dict(data.get("destination"))
         origin = Location.from_dict(data.get("origin"))
         properties = data.get("properties", {})

{apyefa-0.0.2.dist-info → apyefa-0.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: apyefa
-Version: 0.0.2
+Version: 0.0.4
 Summary: Python API for EFA(Elektronische Fahrplanauskunft) async requests
 Author-email: Alex Jung <jungdevelop@gmail.com>
 License: MIT License
@@ -28,20 +28,20 @@ Project-URL: Homepage, https://github.com/alex-jung/apyefa
 Project-URL: Documentation, https://github.com/alex-jung/apyefa
 Project-URL: Repository, https://github.com/alex-jung/apyefa
 Project-URL: Issues, https://github.com/alex-jung/apyefa/issues
-Keywords: efa,vgn,gtfs,public transport,traffic
+Keywords: efa,public transport,traffic
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: aiohttp==3.11.7
-Requires-Dist: voluptuous==0.15.2
-Requires-Dist: tzdata==2024.2
+Requires-Dist: aiohttp>=3.11.7
+Requires-Dist: voluptuous>=0.15.2
+Requires-Dist: tzdata>=2024.2
 Provides-Extra: tests
 Requires-Dist: coverage>=5.0.3; extra == "tests"
 Requires-Dist: pytest-cov; extra == "tests"
 Requires-Dist: pytest; extra == "tests"
-Requires-Dist: pytest-asyncio==0.24.0; extra == "tests"
+Requires-Dist: pytest-asyncio>=0.24.0; extra == "tests"
 Requires-Dist: pytest-benchmark[histogram]>=3.2.1; extra == "tests"
-Requires-Dist: requests==2.32.3; extra == "tests"
+Requires-Dist: requests>=2.32.3; extra == "tests"
 
 # apyefa
 [![Python package](https://github.com/alex-jung/apyefa/actions/workflows/python-package.yml/badge.svg)](https://github.com/alex-jung/apyefa/actions/workflows/python-package.yml)

{apyefa-0.0.2.dist-info → apyefa-0.0.4.dist-info}/RECORD

@@ -1,21 +1,22 @@
 apyefa/__init__.py,sha256=kohSwa1VlugPyxKa0QZqQFwRi4hIGrKFd74W7NDgunA,342
-apyefa/client.py,sha256=mxlxxBMEBRkklWhhbujvN5HpL8QWllSRFwIhkICcmQI,6591
-apyefa/data_classes.py,sha256=UccVL6xvq5xs5M_LR4HOYT_cgM4XjsLIriBOqsJbDTs,10335
+apyefa/client.py,sha256=lwAATLU_-QlAkE_n1kqn2BymvWngH84RDb2D2Z1DYbo,6857
+apyefa/data_classes.py,sha256=KKNYuVHF2d_U-CGxmkyf2RoWCP0GPUvWrrzafytxS_0,11392
 apyefa/exceptions.py,sha256=Vhc8FEtI1xSxbVRLFXd3BlNTekY2w3byEvd3Jhhg8h4,240
 apyefa/helpers.py,sha256=EJyj-Pw3xDrn8WKGbpfqbnaUZjHtVrE2A2LYpTkACps,1695
-apyefa/commands/__init__.py,sha256=kC7Zr8IcahAN8xEOytDUS6WCADQ5oe_nqDIPAjlzxy4,425
+apyefa/commands/__init__.py,sha256=_E7sD_tVa7xI7vUN1EJUwHovSU1xg0RFtbnQb7be1sQ,506
 apyefa/commands/command.py,sha256=wq3rEaYd3NUpkNqo2sX3Yw9-lcRerp8scAy2SVpx1hs,3577
+apyefa/commands/command_add_info.py,sha256=7RLV81i6sCKM3fW1DDX3RJz6GvkICqjxDICF_464ppU,1983
 apyefa/commands/command_departures.py,sha256=kHOyxxSHn7F7ClAV1fQdl1O4cOVRs6a60pkqc8-r_9o,1736
-apyefa/commands/command_serving_lines.py,sha256=C8PrMVWCp71fyPzvsTYMmSaYORs60yUNdEShXEawrK0,2021
-apyefa/commands/command_stop_finder.py,sha256=oDCbjVoQVK_ARDEdmHXuaUJ-IJA0P6f99cVDRS2hqT8,1688
+apyefa/commands/command_serving_lines.py,sha256=NAS_9cvTq7-Lc9lFEFgVsWavEA730FH1UYqohOCDnR0,2038
+apyefa/commands/command_stop_finder.py,sha256=SXi09wA5fmGxEcMEu3XYEK4mpth-bKUoRzDyspDATWo,1944
 apyefa/commands/command_system_info.py,sha256=0bQ0oYbp7FFqPA1b5RgaxANm9-BYlN5KiwRILCCuDVY,783
 apyefa/commands/command_trip.py,sha256=kOlchWyBRlvWBqwjqpx130VXEh5kly6KMSv_pgCBljs,1095
 apyefa/commands/parsers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 apyefa/commands/parsers/parser.py,sha256=1PSlLpfYrNeKRTpvCcYsCzAKlNrs0oHvSN7zpF4Xcs8,140
 apyefa/commands/parsers/rapid_json_parser.py,sha256=UMexRiMKkJLcK5EEj80B6RYkWBuR8ZWD3YpeGa0J3lQ,212
 apyefa/commands/parsers/xml_parser.py,sha256=ru52QtBP68KoTZ8OgjFWrLy-PD4_j1miY8Zv5umNSE8,151
-apyefa-0.0.2.dist-info/LICENSE,sha256=C2Gdvb1B39BeEP-RGqVd7w6j94GnJo4gnYyiC_l3SFQ,1066
-apyefa-0.0.2.dist-info/METADATA,sha256=qRG4UZtgJpcFqSGjF8YgVMtR_ChVHdKcNOAWX64hMgA,15624
-apyefa-0.0.2.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-apyefa-0.0.2.dist-info/top_level.txt,sha256=b9VSv2S7lxdaypCumxO92IEQFpJdFuB8vQs03j5gZxY,7
-apyefa-0.0.2.dist-info/RECORD,,
+apyefa-0.0.4.dist-info/LICENSE,sha256=C2Gdvb1B39BeEP-RGqVd7w6j94GnJo4gnYyiC_l3SFQ,1066
+apyefa-0.0.4.dist-info/METADATA,sha256=OV6RcIfCeFWX_ta-4_bXp0JC3gr_j-9w5LtTBW1uO6g,15615
+apyefa-0.0.4.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+apyefa-0.0.4.dist-info/top_level.txt,sha256=b9VSv2S7lxdaypCumxO92IEQFpJdFuB8vQs03j5gZxY,7
+apyefa-0.0.4.dist-info/RECORD,,