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

koleo/api.py

@@ -136,7 +136,7 @@ class KoleoAPI:
             "query[date]": date.strftime("%d-%m-%Y %H:%M:%S"),
             "query[start_station]": start,
             "query[end_station]": end,
-            "query[only_purchasable]": str(direct).lower(),
+            "query[only_purchasable]": str(purchasable).lower(),
             "query[only_direct]": str(direct).lower(),
             "query[brand_ids][]": brand_ids,
         }

koleo/cli.py

@@ -223,34 +223,57 @@ class CLI:
         self.print("\n".join(parts))
         self.print(self.train_route_table(train_details))
 
-    def connections(self, start: str, end: str, date: datetime, brands: list[str], direct: bool, purchasable: bool):
+    def connections(
+        self,
+        start: str,
+        end: str,
+        date: datetime,
+        brands: list[str],
+        direct: bool,
+        purchasable: bool,
+        length: int = 1
+    ):
         start_station = self.get_station(start)
         end_station = self.get_station(end)
         brands = [i.lower().strip() for i in brands]
         api_brands = self.storage.get_cache("brands") or self.storage.set_cache("brands", self.client.get_brands())
         if not brands:
-            connection_brands = [i["id"] for i in api_brands]
+            connection_brands = {i["name"]: i["id"] for i in api_brands}
         else:
-            connection_brands = [
-                i["id"]
+            connection_brands = {
+                i["name"]:i["id"]
                 for i in api_brands
                 if i["name"].lower().strip() in brands or i["logo_text"].lower().strip() in brands
-            ]
+            }
             if not connection_brands:
                 self.print(f'[bold red]No brands match: "{', '.join(brands)}"[/bold red]')
                 exit(2)
-        connections = self.client.get_connections(
-            start_station["name_slug"], end_station["name_slug"], connection_brands, date, direct, purchasable
+        results = []
+        fetch_date = date
+        while len(results) < length:
+            connections = self.client.get_connections(
+                start_station["name_slug"], end_station["name_slug"], list(connection_brands.values()), fetch_date, direct, purchasable
+            )
+            if connections:
+                fetch_date = arr_dep_to_dt(connections[-1]["departure"]) + timedelta(seconds=(30*60)+1) # wtf
+                results.extend(connections)
+            else:
+                break
+        link = (
+            f"https://koleo.pl/rozklad-pkp/{start_station["name_slug"]}/{end_station["name_slug"]}"
+            + f"/{date.strftime("%d-%m-%Y_%H:%M")}"
+            +f"{"all" if not direct else "direct"}/{"-".join(connection_brands.keys()) if brands else "all"}"
         )
         parts = [
-            f"[bold blue]{start_station["name"]} → {end_station["name"]} at {date.strftime("%H:%M %d-%m")}[/bold blue]"
+            f"[bold blue][link={link}]{start_station["name"]} → {end_station["name"]} at {date.strftime("%H:%M %d-%m")}[/link][/bold blue]"
         ]
-        for i in connections:
+        for i in results:
             arr = arr_dep_to_dt(i["arrival"])
             dep = arr_dep_to_dt(i["departure"])
             travel_time = (arr - dep).seconds
+            date_part = f"{arr.strftime("%d-%m")} " if arr.date() != date.date() else ""
             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"]}]{date_part}{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 len(i["trains"]) == 1:
                 train = i["trains"][0]
@@ -494,7 +517,14 @@ def main():
         action="store_true",
         default=False,
     )
-    connections.set_defaults(func=cli.connections, pass_=["start", "end", "brands", "date", "direct", "purchasable"])
+    connections.add_argument(
+        "-l",
+        "--length",
+        help="fetch at least n connections",
+        type=int,
+        default=1,
+    )
+    connections.set_defaults(func=cli.connections, pass_=["start", "end", "brands", "date", "direct", "purchasable", "length"])
 
     args = parser.parse_args()
 

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

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

koleo_cli-0.2.137.13.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=J05sdOg0S0VgcJNcmmoE8fYMKVVmtqkw5Sy88e-C91k,5713
+koleo/cli.py,sha256=o2Ixo9uegZ7VHL98SoV49Vn2U1VyBp2sqFFZGsbvBb0,24692
+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.13.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+koleo_cli-0.2.137.13.dist-info/METADATA,sha256=tfDdMmERJgsghRGiM0O8yu_qjbawpD6EUb9FDyiSMeE,3096
+koleo_cli-0.2.137.13.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+koleo_cli-0.2.137.13.dist-info/entry_points.txt,sha256=LtCidkVDq8Zd7-fxpRbys1Xa9LTHMZwXVbdcQEscdes,41
+koleo_cli-0.2.137.13.dist-info/top_level.txt,sha256=AlWdXotkRYzHpFfOBYi6xOXl1H0zq4-tqtZ2XivoWB4,6
+koleo_cli-0.2.137.13.dist-info/RECORD,,

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

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

koleo_cli-0.2.137.12.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=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,,