apyefa 0.0.3__py3-none-any.whl → 0.0.5__py3-none-any.whl

apyefa/client.py

@@ -13,6 +13,7 @@ from apyefa.data_classes import (
     CoordFormat,
     Departure,
     Line,
+    LineRequestType,
     Location,
     LocationFilter,
     LocationType,
@@ -162,20 +163,24 @@ 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
+    async def lines_by_location(
+        self, location: str | Location, req_types: list[LineRequestType] = []
+    ) -> list[Line]:
+        """Search for lines that pass `location`. Location can be location ID like `de:08111:6221` or a `Location` object
 
         Args:
             location (str | Location): Location
+            req_types (list[LineRequestType], optional): List of types for the request. Defaults to empty.
 
         Raises:
-            ValueError: If not a stop location provided but e.g. POI or Address
+            ValueError: Wrong location type provided e.g. LocationType.POI or LocationType.ADDRESS
 
         Returns:
             list[Transport]: List of lines
         """
         _LOGGER.info("Request lines by location")
         _LOGGER.debug(f"location:{location}")
+        _LOGGER.debug(f"filters :{req_types}")
 
         if isinstance(location, Location):
             if location.loc_type != LocationType.STOP:
@@ -186,6 +191,9 @@ class EfaClient:
 
         command = CommandServingLines("odv", location)
 
+        if req_types:
+            command.add_param("lineReqType", sum(req_types))
+
         response = await self._run_query(self._build_url(command))
 
         return command.parse(response)

apyefa/commands/command_serving_lines.py

@@ -1,9 +1,9 @@
 import logging
 
-from voluptuous import Any, Optional, Required, Schema
+from voluptuous import Any, Optional, Range, Required, Schema
 
 from apyefa.commands.command import Command
-from apyefa.data_classes import Line
+from apyefa.data_classes import Line, LineRequestType
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -48,7 +48,9 @@ class CommandServingLines(Command):
                 Optional("name_sl"): str,
                 # mode 'line'
                 Optional("lineName"): str,
-                Optional("lineReqType"): int,
+                Optional("lineReqType"): Range(
+                    min=0, max=sum([x.value for x in LineRequestType])
+                ),
                 Optional("mergeDir"): Any("0", "1", 0, 1),
                 Optional("lsShowTrainsExplicit"): Any("0", "1", 0, 1),
                 Optional("line"): str,

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):
@@ -46,6 +76,15 @@ class LocationFilter(IntEnum):
     POST_CODES = 64
 
 
+class LineRequestType(IntEnum):
+    NONE = 0
+    DEPARTURE_MONITOR = 1
+    STOP_TIMETABLE = 2
+    TIMETABLE = 4
+    ROUTE_MAPS = 8
+    STATION_TIMETABLE = 16
+
+
 class CoordFormat(StrEnum):
     WGS84 = "WGS84[dd.ddddd]"
 
@@ -63,7 +102,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 +183,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 +221,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 +408,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.3.dist-info → apyefa-0.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: apyefa
-Version: 0.0.3
+Version: 0.0.5
 Summary: Python API for EFA(Elektronische Fahrplanauskunft) async requests
 Author-email: Alex Jung <jungdevelop@gmail.com>
 License: MIT License
@@ -28,7 +28,7 @@ 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
@@ -280,14 +280,13 @@ Find lines pass provided location.
 |Arguments|Type                |Required|Description|
 |---------|--------------------|--------|-----------|
 |location |str \| [Location](#location) |required|The location passed by searched line(s)|
+|req_types|list[[LineRequestType](#lineRequestType)]|optional|The result presentation type(s) can be defined with this argument. Default value is `[]`
 
 ### Return value
 |Type|Description|
 |----|-----------|
 |list[[Line](#line)]|List of lines found for provided location|
 
-> The attribute `origin` of returned `line` objects is None
-
 ### Examples
 ``` python
 async with EfaClient("https://efa.vgn.de/vgnExt_oeffi/") as client:
@@ -367,17 +366,37 @@ async with EfaClient("https://efa.vgn.de/vgnExt_oeffi/") as client:
 ## TransportType
 ```python
 class TransportType(IntEnum):
-    RAIL        = 0 
-    SUBURBAN    = 1
-    SUBWAY      = 2 
-    CITY_RAIL   = 3 
-    TRAM        = 4
-    BUS         = 5 
-    RBUS        = 6 
-    EXPRESS_BUS = 7
-    CABLE_TRAM  = 8
-    FERRY       = 9 
-    AST         = 10  # Anruf-Sammel-Taxi
+    TRAIN                      = 0  # Zug
+    SUBURBAN                   = 1  # S-Bahn
+    SUBWAY                     = 2  # U-Bahn
+    CITY_RAIL                  = 3  # Stadtbahn
+    TRAM                       = 4  # Straßenbahn
+    CITY_BUS                   = 5  # Stadtbus
+    REGIONAL_BUS               = 6  # Regionalbus
+    EXPRESS_BUS                = 7  # Schnellbus
+    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
+```
+
+## LineRequestType
+```python
+class LineRequestType(IntEnum):
+    NONE              = 0
+    DEPARTURE_MONITOR = 1
+    STOP_TIMETABLE    = 2
+    TIMETABLE         = 4
+    ROUTE_MAPS        = 8
+    STATION_TIMETABLE = 16
 ```
 
 ## CoordFormat

{apyefa-0.0.3.dist-info → apyefa-0.0.5.dist-info}/RECORD

@@ -1,12 +1,12 @@
 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=UYt1kvxvPjg-VfdlUqPsazyEdc97J27_DMoprjF8C8g,6924
+apyefa/data_classes.py,sha256=Njt7r8ONEY5KxUz9uVFnPHU9qz9sp5RWmmhiR8ywDiY,11552
 apyefa/exceptions.py,sha256=Vhc8FEtI1xSxbVRLFXd3BlNTekY2w3byEvd3Jhhg8h4,240
 apyefa/helpers.py,sha256=EJyj-Pw3xDrn8WKGbpfqbnaUZjHtVrE2A2LYpTkACps,1695
 apyefa/commands/__init__.py,sha256=kC7Zr8IcahAN8xEOytDUS6WCADQ5oe_nqDIPAjlzxy4,425
 apyefa/commands/command.py,sha256=wq3rEaYd3NUpkNqo2sX3Yw9-lcRerp8scAy2SVpx1hs,3577
 apyefa/commands/command_departures.py,sha256=kHOyxxSHn7F7ClAV1fQdl1O4cOVRs6a60pkqc8-r_9o,1736
-apyefa/commands/command_serving_lines.py,sha256=C8PrMVWCp71fyPzvsTYMmSaYORs60yUNdEShXEawrK0,2021
+apyefa/commands/command_serving_lines.py,sha256=V-HVYgFyeEWBLebKbp6NNXshxw0fVPttGZLu-CTmFC4,2137
 apyefa/commands/command_stop_finder.py,sha256=oDCbjVoQVK_ARDEdmHXuaUJ-IJA0P6f99cVDRS2hqT8,1688
 apyefa/commands/command_system_info.py,sha256=0bQ0oYbp7FFqPA1b5RgaxANm9-BYlN5KiwRILCCuDVY,783
 apyefa/commands/command_trip.py,sha256=kOlchWyBRlvWBqwjqpx130VXEh5kly6KMSv_pgCBljs,1095
@@ -14,8 +14,8 @@ apyefa/commands/parsers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG
 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.3.dist-info/LICENSE,sha256=C2Gdvb1B39BeEP-RGqVd7w6j94GnJo4gnYyiC_l3SFQ,1066
-apyefa-0.0.3.dist-info/METADATA,sha256=mcFjLRl8QNUwgYkepVGceQ8TbzlTazk7MbSSqBjpEi8,15624
-apyefa-0.0.3.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-apyefa-0.0.3.dist-info/top_level.txt,sha256=b9VSv2S7lxdaypCumxO92IEQFpJdFuB8vQs03j5gZxY,7
-apyefa-0.0.3.dist-info/RECORD,,
+apyefa-0.0.5.dist-info/LICENSE,sha256=C2Gdvb1B39BeEP-RGqVd7w6j94GnJo4gnYyiC_l3SFQ,1066
+apyefa-0.0.5.dist-info/METADATA,sha256=bop1VP_hhMO9A_9inje-HerUa1uArdE2jkvA1mZUHa0,16750
+apyefa-0.0.5.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+apyefa-0.0.5.dist-info/top_level.txt,sha256=b9VSv2S7lxdaypCumxO92IEQFpJdFuB8vQs03j5gZxY,7
+apyefa-0.0.5.dist-info/RECORD,,