koleo-cli 0.2.137__py3-none-any.whl → 0.2.137.1__py3-none-any.whl

koleo/api.py

@@ -106,11 +106,11 @@ class KoleoAPI:
         params = {"brand": brand_name, "nr": number}
         if name:
             params["name"] = name.upper()  # WHY!!!!!!!!!
-        return self._get_json("/api/v2/main/train_calendars", params=params)
+        return self._get_json("/pl/train_calendars", params=params)
 
     def get_train(self, id: int) -> TrainDetailResponse:
         # https://koleo.pl/pl/trains/142821312
-        return self._get_json(f"/api/v2/main/trains/{id}")
+        return self._get_json(f"/pl/trains/{id}")
 
     def get_connections(
         self,

koleo/cli.py

@@ -6,7 +6,7 @@ from rich.traceback import install
 
 from .api import KoleoAPI
 from .types import TrainOnStationInfo, TrainDetailResponse
-from .utils import name_to_slug, parse_datetime, time_dict_to_dt
+from .utils import name_to_slug, parse_datetime, time_dict_to_dt, convert_platform_number
 from .storage import Storage, DEFAULT_CONFIG_PATH
 
 install(show_locals=True)
@@ -87,24 +87,24 @@ class CLI:
 
     def train_info(self, brand: str, name: str, date: datetime):
         brand = brand.upper().strip()
-        name = name.upper().strip()
-        cache_id = f"tc-{brand}-{name}"
+        name = name.strip()
         if name.isnumeric():
             number = int(name)
-            name = ""
-        elif len(parts := name.split(" ")) == 2 or len(parts := name.split("-")) == 2:
-            number, name = parts
+            train_name = ""
+        elif len((parts := name.split(" "))) == 2 or len((parts := name.split("-"))) == 2:
+            number, train_name = parts
             number = int(number)
         else:
             raise ValueError("Invalid train name!")
+        cache_id = f"tc-{brand}-{number}-{name}"
         train_calendars = self.storage.get_cache(cache_id) or self.storage.set_cache(
-            cache_id, self.client.get_train_calendars(brand, number, name)
+            cache_id, self.client.get_train_calendars(brand, number, train_name)
         )
         brands = self.storage.get_cache("brands") or self.storage.set_cache("brands", self.client.get_brands())
         train_id = train_calendars["train_calendars"][0]["date_train_map"][date.strftime("%Y-%m-%d")]
         train_details = self.client.get_train(train_id)
         brand = next(iter(i for i in brands if i["id"] == train_details["train"]["brand_id"]), {}).get("name", "")
-        data = f"{brand} {train_details["train"]["train_full_name"]}\n"
+        parts = [f"{brand} {train_details["train"]["train_full_name"]}"]
         vehicle_types: dict[str, str] = {
             stop["station_display_name"]: stop["vehicle_type"]
             for stop in train_details["stops"]
@@ -115,10 +115,10 @@ class CLI:
             start = keys[0]
             for i in range(1, len(keys)):
                 if vehicle_types[keys[i]] != vehicle_types[start]:
-                    data += f"[bold green]  {start} - {keys[i]}:[/bold green] {vehicle_types[start]}\n"
+                    parts.append(f"[bold green]  {start} - {keys[i]}:[/bold green] {vehicle_types[start]}")
                     start = keys[i]
-            data += f"[bold green]  {start} - {keys[-1]}:[/bold green] {vehicle_types[start]}"
-        self.console.print(data)
+            parts.append(f"[bold green]  {start} - {keys[-1]}:[/bold green] {vehicle_types[start]}")
+        self.console.print("\n".join(parts))
         self.console.print(self.train_route_table(train_details))
 
     def route(self, start: str, end: str, date: datetime, direct: bool = False, purchasable: bool = False):
@@ -147,8 +147,9 @@ class CLI:
         for stop in train["stops"]:
             arr = time_dict_to_dt(stop["arrival"])
             dep = time_dict_to_dt(stop["departure"])
+            platform = convert_platform_number(stop["platform"]) or ""
             parts.append(
-                f"[white underline]{stop["distance"] / 1000:0.4}km[/white underline] [bold green]{arr.strftime("%H:%M")}[/bold green] - [bold red]{dep.strftime("%H:%M")}[/bold red] [purple] {stop["station_display_name"]} {stop["platform"]} [/purple]"
+                f"[white underline]{stop["distance"] / 1000:^5.1f}km[/white underline] [bold green]{arr.strftime("%H:%M")}[/bold green] - [bold red]{dep.strftime("%H:%M")}[/bold red] [purple]{stop["station_display_name"]} {platform} [/purple]"
             )
         return "\n".join(parts)
 

koleo/utils.py

@@ -39,3 +39,23 @@ TRANSLITERATIONS = {
 
 def name_to_slug(name: str) -> str:
     return "".join([TRANSLITERATIONS.get(char, char) for char in name.lower()])
+
+
+NUMERAL_TO_ARABIC = {
+    "I": 1,
+    "II": 2,
+    "III": 3,
+    "IV": 4,
+    "V": 5,
+    "VI": 6,
+    "VII": 7,
+    "VIII": 8,
+    "IX": 9,
+    "X": 10,
+    "XI": 11, # wtf poznań???
+    "XII": 12 # just to be safe
+}
+
+
+def convert_platform_number(number: str) -> int | None:
+    return NUMERAL_TO_ARABIC.get(number)

{koleo_cli-0.2.137.dist-info → koleo_cli-0.2.137.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: koleo-cli
-Version: 0.2.137
+Version: 0.2.137.1
 Summary: Koleo CLI
 Home-page: https://github.com/lzgirlcat/koleo-cli
 Author: Zoey !

koleo_cli-0.2.137.1.dist-info/RECORD

@@ -0,0 +1,12 @@
+koleo/__init__.py,sha256=ChPpKyc4vrcgWj-Uk_ZlDw32s_7iFpE3f1FTAd6zxPY,51
+koleo/__main__.py,sha256=wu5N2wk8mvBgyvr2ghmQf4prezAe0_i-p123VVreyYc,62
+koleo/api.py,sha256=uzaOSD_voMRoIxKRuh-RqGKGcJDyAUh3UtOvchW-skc,5052
+koleo/cli.py,sha256=ARSDB6bLYf0c2gSkY_A_b4BoniaMBamCfCUByZsSl5o,9365
+koleo/storage.py,sha256=uCh6edwizAuw1z_Ti5AXvDan2pJAJBSobCVmYw096F8,2015
+koleo/types.py,sha256=8mAsRdNh3jMJwAV5KnHVvDQAekhs0N70EVem4p_w18o,3760
+koleo/utils.py,sha256=vywC_2q_quCdk23GPYXzm6oteCYwHcS3AVwK2pYv1NQ,1224
+koleo_cli-0.2.137.1.dist-info/METADATA,sha256=MXri6goTJkWZs2XgcsdSm-MHie9-53nWXCleo4XVIJU,626
+koleo_cli-0.2.137.1.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
+koleo_cli-0.2.137.1.dist-info/entry_points.txt,sha256=LtCidkVDq8Zd7-fxpRbys1Xa9LTHMZwXVbdcQEscdes,41
+koleo_cli-0.2.137.1.dist-info/top_level.txt,sha256=AlWdXotkRYzHpFfOBYi6xOXl1H0zq4-tqtZ2XivoWB4,6
+koleo_cli-0.2.137.1.dist-info/RECORD,,

koleo_cli-0.2.137.dist-info/RECORD

@@ -1,12 +0,0 @@
-koleo/__init__.py,sha256=ChPpKyc4vrcgWj-Uk_ZlDw32s_7iFpE3f1FTAd6zxPY,51
-koleo/__main__.py,sha256=wu5N2wk8mvBgyvr2ghmQf4prezAe0_i-p123VVreyYc,62
-koleo/api.py,sha256=8aYn4SsavjOx9cPHLgSlwVmbZaZWJNYpIL3lZPuvdTk,5070
-koleo/cli.py,sha256=fd1O_gO5EF-7BrrLe79j7fn7SEKr5yLgWu33UIjLghM,9230
-koleo/storage.py,sha256=uCh6edwizAuw1z_Ti5AXvDan2pJAJBSobCVmYw096F8,2015
-koleo/types.py,sha256=8mAsRdNh3jMJwAV5KnHVvDQAekhs0N70EVem4p_w18o,3760
-koleo/utils.py,sha256=MYfeQkk9spT2lmlFVsHlpzgnAf0hqN-gKEbgYASqW6U,904
-koleo_cli-0.2.137.dist-info/METADATA,sha256=-PLG8A9e3iD-az9tyrdJEitqThaZrBkOhiWHWuMSPXk,624
-koleo_cli-0.2.137.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-koleo_cli-0.2.137.dist-info/entry_points.txt,sha256=LtCidkVDq8Zd7-fxpRbys1Xa9LTHMZwXVbdcQEscdes,41
-koleo_cli-0.2.137.dist-info/top_level.txt,sha256=AlWdXotkRYzHpFfOBYi6xOXl1H0zq4-tqtZ2XivoWB4,6
-koleo_cli-0.2.137.dist-info/RECORD,,