koleo-cli 0.2.137.10__py3-none-any.whl → 0.2.137.12__py3-none-any.whl

koleo/cli.py

@@ -96,14 +96,16 @@ class CLI:
         arr_cache_id = f"arr-{st['id']}-{date.strftime("%Y-%m-%d")}"
         dep_cache_id = f"dep-{st['id']}-{date.strftime("%Y-%m-%d")}"
         arrivals = self.storage.get_cache(arr_cache_id) or self.storage.set_cache(
-            arr_cache_id, self.client.get_arrivals(st['id'], date)
+            arr_cache_id, self.client.get_arrivals(st["id"], date)
         )
         departures = self.storage.get_cache(dep_cache_id) or self.storage.set_cache(
-            dep_cache_id, self.client.get_departures(st['id'], date)
+            dep_cache_id, self.client.get_departures(st["id"], date)
         )
         trains = sorted(
             [(i, 1) for i in departures] + [(i, 2) for i in arrivals],
-            key=lambda train: datetime.fromisoformat(train[0]["departure"] if train[1] == 1 else train[0]["arrival"]).timestamp()
+            key=lambda train: datetime.fromisoformat(
+                train[0]["departure"] if train[1] == 1 else train[0]["arrival"]
+            ).timestamp(),
         )
         trains = [
             (i, type)
@@ -113,7 +115,11 @@ class CLI:
         brands = self.storage.get_cache("brands") or self.storage.set_cache("brands", self.client.get_brands())
         parts = []
         for train, type in trains:
-            time = f"[bold green]{train['departure'][11:16]}[/bold green]" if type == 1 else f"[bold yellow]{train['arrival'][11:16]}[/bold yellow]"
+            time = (
+                f"[bold green]{train['departure'][11:16]}[/bold green]"
+                if type == 1
+                else f"[bold yellow]{train['arrival'][11:16]}[/bold yellow]"
+            )
             brand = next(iter(i for i in brands if i["id"] == train["brand_id"]), {}).get("logo_text")
             parts.append(
                 f"{time} [red]{brand}[/red] {train["train_full_name"]}[purple] {train["stations"][0]["name"]} {self.format_position(train["platform"], train["track"])}[/purple]"
@@ -164,14 +170,16 @@ class CLI:
         brands = self.storage.get_cache("brands") or self.storage.set_cache("brands", self.client.get_brands())
         for calendar in train_calendars:
             brand = next(iter(i for i in brands if i["id"] == calendar["trainBrand"]), {}).get("logo_text", "")
-            parts = [f"[red]{brand}[/red] [bold blue]{calendar['train_nr']}{" "+ v if (v:=calendar.get("train_name")) else ""}[/bold blue]:"]
+            parts = [
+                f"[red]{brand}[/red] [bold blue]{calendar['train_nr']}{" "+ v if (v:=calendar.get("train_name")) else ""}[/bold blue]:"
+            ]
             for k, v in calendar["date_train_map"].items():
                 parts.append(f"  [bold green]{k}[/bold green]: [purple]{v}[/purple]")
             self.print("\n".join(parts))
 
     def train_info(self, brand: str, name: str, date: datetime):
         train_calendars = self.get_train_calendars(brand, name)
-        if not (train_id:=train_calendars[0]["date_train_map"].get(date.strftime("%Y-%m-%d"))):
+        if not (train_id := train_calendars[0]["date_train_map"].get(date.strftime("%Y-%m-%d"))):
             self.print(f"[bold red]This train doesn't run on the selected date: {date.strftime("%Y-%m-%d")}[/bold red]")
             exit(2)
         self.train_detail(train_id)
@@ -182,7 +190,8 @@ class CLI:
         brand = next(iter(i for i in brands if i["id"] == train_details["train"]["brand_id"]), {}).get("logo_text", "")
 
         parts = [f"[red]{brand}[/red] [bold blue]{train_details["train"]["train_full_name"]}[/bold blue]"]
-        parts.append(f"  {train_details["train"]["run_desc"]}")
+        if train_details["train"]["run_desc"]:
+            parts.append(f"  {train_details["train"]["run_desc"]}")
 
         route_start = arr_dep_to_dt(train_details["stops"][0]["departure"])
         route_end = arr_dep_to_dt(train_details["stops"][-1]["arrival"])
@@ -241,11 +250,8 @@ class CLI:
             dep = arr_dep_to_dt(i["departure"])
             travel_time = (arr - dep).seconds
             parts.append(
-                f"[bold green][link=https://koleo.pl/travel-options/{i["id"]}]{dep.strftime("%H:%M")} - {arr.strftime("%H:%M")}[/bold green] {travel_time//3600}h{(travel_time % 3600)/60:.0f}m {i['distance']}km: [/link]"
+                f"[bold green][link=https://koleo.pl/travel-options/{i["id"]}]{dep.strftime("%H:%M")} - {arr.strftime("%H:%M")}[/bold green] {travel_time//3600}h{(travel_time % 3600)/60:.0f}m {i['distance']}km:[/link]"
             )
-            if i["constriction_info"]:
-                for constriction in i["constriction_info"]:
-                    parts.append(f" [bold red]- {constriction} [/bold red]")
             if len(i["trains"]) == 1:
                 train = i["trains"][0]
                 stop = next(iter(i for i in train["stops"] if i["station_id"] == train["start_station_id"]), {})
@@ -258,10 +264,15 @@ class CLI:
                     )
                 )
                 brand = next(iter(i for i in api_brands if i["id"] == train["brand_id"]), {}).get("logo_text")
+                s = f" [red]{brand}[/red] {train["train_full_name"]}[purple] {stop_station['name']} {self.format_position(stop["platform"], stop["track"])}[/purple]"
                 parts[-1] += (
                     f" [red]{brand}[/red] {train["train_full_name"]}[purple] {stop_station['name']} {self.format_position(stop["platform"], stop["track"])}[/purple]"
                 )
+                for constriction in i["constriction_info"]:
+                    parts.append(f" [bold red]- {constriction}[/bold red]")
             else:
+                for constriction in i["constriction_info"]:
+                    parts.append(f" [bold red]- {constriction}[/bold red]")
                 previous_arrival: datetime | None = None
                 for train in i["trains"]:
                     brand = next(iter(i for i in api_brands if i["id"] == train["brand_id"]), {}).get("logo_text")

koleo/storage.py

@@ -81,4 +81,4 @@ class Storage:
         copy = self.cache.copy()
         self.cache = {k: data for k, data in copy.items() if data[0] > now}
         if copy != self.cache:
-            self._dirty = True
+            self._dirty = True

{koleo_cli-0.2.137.10.dist-info → koleo_cli-0.2.137.12.dist-info}/METADATA

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

koleo_cli-0.2.137.12.dist-info/RECORD

@@ -0,0 +1,13 @@
+koleo/__init__.py,sha256=N_IkOBZCSPCCw31Hu72CFys707PziGFmXpNVl0CXAz8,47
+koleo/__main__.py,sha256=wu5N2wk8mvBgyvr2ghmQf4prezAe0_i-p123VVreyYc,62
+koleo/api.py,sha256=07PSwLFmirdJ_JhPBJ7rO1nv_v90njIcwmOoxT4P_4M,5708
+koleo/cli.py,sha256=5QmvtqjB7kirt1iEG5SZDzOV6QTHTiJMNRtJoZa2Jik,23678
+koleo/storage.py,sha256=L31IrXxdWiGGOraVBdFeHc1Z-a2fNvyRn-uC7gJWe9E,2305
+koleo/types.py,sha256=pjVCIH39rUNeipazeRKOuTRCWoslLdbGivKiQmfd5Pw,5226
+koleo/utils.py,sha256=CsYWNf3IPhHJln445tzXSA7r2z3NpmqVSmw8Rrb7YT0,1822
+koleo_cli-0.2.137.12.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+koleo_cli-0.2.137.12.dist-info/METADATA,sha256=WAaZyvFSeL7O3M2Ss9YNz2SyyPTW3m7tzU8KKdY-pnY,3096
+koleo_cli-0.2.137.12.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
+koleo_cli-0.2.137.12.dist-info/entry_points.txt,sha256=LtCidkVDq8Zd7-fxpRbys1Xa9LTHMZwXVbdcQEscdes,41
+koleo_cli-0.2.137.12.dist-info/top_level.txt,sha256=AlWdXotkRYzHpFfOBYi6xOXl1H0zq4-tqtZ2XivoWB4,6
+koleo_cli-0.2.137.12.dist-info/RECORD,,

{koleo_cli-0.2.137.10.dist-info → koleo_cli-0.2.137.12.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.2.0)
+Generator: setuptools (75.5.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

koleo_cli-0.2.137.10.dist-info/RECORD

@@ -1,13 +0,0 @@
-koleo/__init__.py,sha256=N_IkOBZCSPCCw31Hu72CFys707PziGFmXpNVl0CXAz8,47
-koleo/__main__.py,sha256=wu5N2wk8mvBgyvr2ghmQf4prezAe0_i-p123VVreyYc,62
-koleo/api.py,sha256=07PSwLFmirdJ_JhPBJ7rO1nv_v90njIcwmOoxT4P_4M,5708
-koleo/cli.py,sha256=CeZBwuS2ibZZ_HQq8bRwcY4HLq59_tEb9ux7daOvgc4,23239
-koleo/storage.py,sha256=U1ydt8AZ3WsiNBOmThD-cxomLzLrpHRG4BZbMg-14Fc,2304
-koleo/types.py,sha256=pjVCIH39rUNeipazeRKOuTRCWoslLdbGivKiQmfd5Pw,5226
-koleo/utils.py,sha256=CsYWNf3IPhHJln445tzXSA7r2z3NpmqVSmw8Rrb7YT0,1822
-koleo_cli-0.2.137.10.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-koleo_cli-0.2.137.10.dist-info/METADATA,sha256=wwrlHORzN11td36xS_fjWglZpIWA11EryaJ6rpm9kXk,3096
-koleo_cli-0.2.137.10.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-koleo_cli-0.2.137.10.dist-info/entry_points.txt,sha256=LtCidkVDq8Zd7-fxpRbys1Xa9LTHMZwXVbdcQEscdes,41
-koleo_cli-0.2.137.10.dist-info/top_level.txt,sha256=AlWdXotkRYzHpFfOBYi6xOXl1H0zq4-tqtZ2XivoWB4,6
-koleo_cli-0.2.137.10.dist-info/RECORD,,