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

koleo/cli.py

@@ -89,6 +89,37 @@ class CLI:
         self.print(station_info)
         self.get_arrivals(st["id"], date)
 
+    def all_trains(self, station: str, date: datetime):
+        st = self.get_station(station)
+        station_info = f"[bold blue][link=https://koleo.pl/dworzec-pkp/{st["name_slug"]}/{date.strftime("%Y-%m-%d")}]{st["name"]} at {date.strftime("%d-%m %H:%M")}[/bold blue] ID: {st["id"]}[/link]"
+        self.print(station_info)
+        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)
+        )
+        departures = self.storage.get_cache(dep_cache_id) or self.storage.set_cache(
+            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()
+        )
+        trains = [
+            (i, type)
+            for i, type in trains
+            if datetime.fromisoformat(i["departure"] if type == 1 else i["arrival"]).timestamp() > date.timestamp()  # type: ignore
+        ]
+        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]"
+            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]"
+            )
+        self.print("\n".join(parts))
+
     def find_station(self, query: str | None):
         if query:
             stations = self.client.find_station(query)
@@ -356,6 +387,26 @@ def main():
     arrivals.add_argument("-s", "--save", help="save the station as your default one", action="store_true")
     arrivals.set_defaults(func=cli.full_arrivals, pass_=["station", "date"])
 
+    all_trains = subparsers.add_parser(
+        "all", aliases=["w", "wszystkie", "all_trains", "pociagi"], help="Allows you to list all station trains"
+    )
+    all_trains.add_argument(
+        "station",
+        help="The station name",
+        default=None,
+        nargs="*",
+        action=RemainderString,
+    )
+    all_trains.add_argument(
+        "-d",
+        "--date",
+        help="the date",
+        type=lambda s: parse_datetime(s),
+        default=datetime.now(),
+    )
+    all_trains.add_argument("-s", "--save", help="save the station as your default one", action="store_true")
+    all_trains.set_defaults(func=cli.all_trains, pass_=["station", "date"])
+
     train_route = subparsers.add_parser(
         "trainroute",
         aliases=["r", "tr", "t", "poc", "pociąg"],
@@ -445,7 +496,6 @@ def main():
         args.station = storage.favourite_station
     elif hasattr(args, "station") and args.save:
         storage.favourite_station = args.station
-        storage.save()
     if not hasattr(args, "func"):
         if storage.favourite_station:
             cli.full_departures(storage.favourite_station, datetime.now())
@@ -454,3 +504,6 @@ def main():
             exit()
     else:
         args.func(**{k: v for k, v in args.__dict__.items() if k in getattr(args, "pass_", [])})
+
+    if storage._dirty:
+        storage.save()

koleo/storage.py

@@ -33,6 +33,7 @@ class Storage:
 
     def __post_init__(self):
         self._path: str
+        self._dirty = False
 
     @classmethod
     def load(cls, *, path: str = DEFAULT_CONFIG_PATH) -> t.Self:
@@ -57,13 +58,13 @@ class Storage:
             return item
         else:
             self.cache.pop(id)
-            self.save()
+            self._dirty = True
 
     def set_cache(self, id: str, item: T, ttl: int = 86400) -> T:
         if self.disable_cache:
             return item
         self.cache[id] = (int(time() + ttl), item)
-        self.save()
+        self._dirty = True
         return item
 
     def save(self):
@@ -72,4 +73,12 @@ class Storage:
             if not ospath.exists(dir):
                 makedirs(dir)
         with open(self._path, "w+") as f:
+            self.clean()
             dump(asdict(self), f, indent=True)
+
+    def clean(self):
+        now = time()
+        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

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

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

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

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

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

koleo_cli-0.2.137.9.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=_ahP1RWnlCtoM857fKZgbdO_eQQiuyDEdhYA2RoX784,20627
-koleo/storage.py,sha256=_VztM8d3YTnmZOR9JvOsobThrQhb4gEzT1azC8N7baY,2024
-koleo/types.py,sha256=pjVCIH39rUNeipazeRKOuTRCWoslLdbGivKiQmfd5Pw,5226
-koleo/utils.py,sha256=CsYWNf3IPhHJln445tzXSA7r2z3NpmqVSmw8Rrb7YT0,1822
-koleo_cli-0.2.137.9.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-koleo_cli-0.2.137.9.dist-info/METADATA,sha256=xfSWLLtbGywsL7CqQ9buD9bKWE1okzZyvo3kyWqyj2Q,3095
-koleo_cli-0.2.137.9.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-koleo_cli-0.2.137.9.dist-info/entry_points.txt,sha256=LtCidkVDq8Zd7-fxpRbys1Xa9LTHMZwXVbdcQEscdes,41
-koleo_cli-0.2.137.9.dist-info/top_level.txt,sha256=AlWdXotkRYzHpFfOBYi6xOXl1H0zq4-tqtZ2XivoWB4,6
-koleo_cli-0.2.137.9.dist-info/RECORD,,