koleo-cli 0.2.137.16__py3-none-any.whl → 0.2.137.18__py3-none-any.whl

koleo_cli-0.2.137.18.dist-info/RECORD

@@ -0,0 +1,26 @@
+koleo/__init__.py,sha256=dp6PfGcuWP50QBWhjnT-zS03l3KzzYnMJnUeYMKBayc,51
+koleo/__main__.py,sha256=wu5N2wk8mvBgyvr2ghmQf4prezAe0_i-p123VVreyYc,62
+koleo/args.py,sha256=gncN8HA8a67U1Wpt3lc32bvPBydn0uCkSmiCZNVQRGk,10275
+koleo/storage.py,sha256=pffcqpTakYp8o6dbVMkD5oF13R6L2eQR3ambU8qTRuM,4345
+koleo/utils.py,sha256=OAjm90SN1BqpgNF8OTAaR9atyLNYIGy5owZ2pLP5OAM,4631
+koleo/api/__init__.py,sha256=3TfO8eBJcDCDXvIFobgyvN3Sfa1kQ2U0lm7gxEAsaWQ,50
+koleo/api/base.py,sha256=8wAasDQ1DBDI1PsPESmS3iUE5csIgaGT4DOX_A-hPwI,2070
+koleo/api/client.py,sha256=Ru5M5ntp2vpQQI4VUaD1TCLIjNtJB4tkNCB4y766DUU,8457
+koleo/api/errors.py,sha256=J9g75K9_yvnZ-7Hi521h0yGcw9NUNnTprQsIZ_mclO0,1324
+koleo/api/logging.py,sha256=VhOFY6N_mofFfg9ZJ5ZnDVCzNuRilqLqrzZU9b-hpZY,1644
+koleo/api/types.py,sha256=udCQeDjRvUqrUkuE_-Jidzsq_FOCjNyGd_bQWm3f9iA,10594
+koleo/cli/__init__.py,sha256=50B_lMtZS27_FVrkIa2TOWMX7SozG1CPPMxbY1W9tqk,266
+koleo/cli/aliases.py,sha256=is3NX5g0-ujJTmyjlFLoNnK_pwoYUVIojWNkgwXRBKk,581
+koleo/cli/base.py,sha256=Rdv8FwenM0ymSHNrW9faGYI6mr65EL-n24bN5YXfLkg,4261
+koleo/cli/connections.py,sha256=LJzoq3ADVUmBeqF4-h_me3pZbBHYg1sG3MaDYprwuLU,6873
+koleo/cli/seats.py,sha256=hIALtopQPD-BcBGI_XpPo3ht6f_D5e16VpI7mFgN0Ao,5477
+koleo/cli/station_board.py,sha256=lRGIeEKEOvUcctM6cwmqVE7Yk9m9R12gfFZahlWLQFc,3669
+koleo/cli/stations.py,sha256=6L3PBWc6xssyR9eeLacuvGBJmaY7Ny3DalKy2Xq7zsA,1633
+koleo/cli/train_info.py,sha256=XS_G4dsevHBPuGSpsbNTh_adSLyedame7WVe71oIe5M,7259
+koleo/cli/utils.py,sha256=FzPGcJdwRwcz10mYiW63Y4zpjM9j6DzNH91UNR3MS8s,666
+koleo_cli-0.2.137.18.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+koleo_cli-0.2.137.18.dist-info/METADATA,sha256=JLxhaXfSUCa7zhq12sGS8LLP94BvnW2Tc3DoFOk9NYU,4731
+koleo_cli-0.2.137.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+koleo_cli-0.2.137.18.dist-info/entry_points.txt,sha256=ouWbMR_XWpEwV7zfkFKeiFLe_IMP-47kTvVahgy4PRg,42
+koleo_cli-0.2.137.18.dist-info/top_level.txt,sha256=AlWdXotkRYzHpFfOBYi6xOXl1H0zq4-tqtZ2XivoWB4,6
+koleo_cli-0.2.137.18.dist-info/RECORD,,

{koleo_cli-0.2.137.16.dist-info → koleo_cli-0.2.137.18.dist-info}/WHEEL

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

koleo_cli-0.2.137.18.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+koleo = koleo.args:main

koleo/api.py

@@ -1,161 +0,0 @@
-import typing as t
-from datetime import datetime
-
-from requests import PreparedRequest, Response, Session
-
-from koleo.types import *
-
-
-class errors:
-    class KoleoAPIException(Exception):
-        status: int
-        request: PreparedRequest
-        response: Response
-
-        def __init__(self, response: Response, *args: object) -> None:
-            super().__init__(*args)
-            self.status = response.status_code
-            self.request = response.request
-            self.response = response
-
-        @classmethod
-        def from_response(cls, response: Response) -> "t.Self":
-            if response.status_code == 404:
-                return errors.KoleoNotFound(response)
-            elif response.status_code == 401:
-                return errors.KoleoUnauthorized(response)
-            elif response.status_code == 403:
-                return errors.KoleoForbidden(response)
-            elif response.status_code == 429:
-                return errors.KoleoRatelimited(response)
-            else:
-                return cls(response)
-
-    class KoleoNotFound(KoleoAPIException):
-        pass
-
-    class KoleoForbidden(KoleoAPIException):
-        pass
-
-    class KoleoUnauthorized(KoleoAPIException):
-        pass
-
-    class KoleoRatelimited(KoleoAPIException):
-        pass
-
-
-class KoleoAPI:
-    errors = errors
-
-    def __init__(self) -> None:
-        self.session = Session()
-        self.base_url = "https://koleo.pl"
-        self.version = 2
-        self.base_headers = {
-            "x-koleo-version": str(self.version),
-            "User-Agent": "Koleo-CLI(https://pypi.org/project/koleo-cli)",
-        }
-
-    def _get(self, path, *args, **kwargs) -> Response:
-        headers = {**self.base_headers, **kwargs.get("headers", {})}
-        r = self.session.get(self.base_url + path, *args, headers=headers, **kwargs)
-        if not r.ok:
-            raise errors.KoleoAPIException.from_response(r)
-        return r
-
-    def _get_json(self, path, *args, **kwargs) -> t.Any:
-        r = self._get(path, *args, **kwargs)
-        res = r.json()
-        if res is None:
-            raise self.errors.KoleoNotFound(r)
-        return res
-
-    def _get_bytes(self, path, *args, **kwargs) -> bytes:
-        r = self._get(path, *args, **kwargs)
-        return r.content
-
-    def get_stations(self) -> list[ExtendedStationInfo]:
-        return self._get_json("/api/v2/main/stations")
-
-    def find_station(self, query: str, language: str = "pl") -> list[ExtendedStationInfo]:
-        # https://koleo.pl/ls?q=tere&language=pl
-        return self._get_json("/ls", params={"q": query, "language": language})["stations"]
-
-    def get_station_by_id(self, id: int) -> ExtendedBaseStationInfo:
-        # https://koleo.pl/api/v2/main/stations/by_id/24000
-        return self._get_json(
-            f"/api/v2/main/stations/by_id/{id}",
-        )
-
-    def get_station_by_slug(self, slug: str) -> ExtendedBaseStationInfo:
-        # https://koleo.pl/api/v2/main/stations/by_slug/inowroclaw
-        return self._get_json(
-            f"/api/v2/main/stations/by_slug/{slug}",
-        )
-
-    def get_station_info_by_slug(self, slug: str) -> StationDetails:
-        # https://koleo.pl/api/v2/main/station_info/inowroclaw
-        return self._get_json(
-            f"/api/v2/main/station_info/{slug}",
-        )
-
-    def get_departures(self, station_id: int, date: datetime) -> list[TrainOnStationInfo]:
-        # https://koleo.pl/api/v2/main/timetables/18705/2024-03-25/departures
-        return self._get_json(
-            f"/api/v2/main/timetables/{station_id}/{date.strftime("%Y-%m-%d")}/departures",
-        )
-
-    def get_arrivals(self, station_id: int, date: datetime) -> list[TrainOnStationInfo]:
-        # https://koleo.pl/api/v2/main/timetables/18705/2024-03-25/arrivals
-        return self._get_json(
-            f"/api/v2/main/timetables/{station_id}/{date.strftime("%Y-%m-%d")}/arrivals",
-        )
-
-    def get_train_calendars(self, brand_name: str, number: int, name: str | None = None) -> TrainCalendarResponse:
-        # https://koleo.pl/pl/train_calendars?brand=REG&nr=10417
-        # https://koleo.pl/pl/train_calendars?brand=IC&nr=1106&name=ESPERANTO ; WHY!!!! WHY!!!!!!1
-        params = {"brand": brand_name, "nr": number}
-        if name:
-            params["name"] = name.upper()  # WHY!!!!!!!!!
-        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"/pl/trains/{id}")
-
-    def get_connections(
-        self,
-        start: str,
-        end: str,
-        brand_ids: list[int],
-        date: datetime,
-        direct: bool = False,
-        purchasable: bool = False,
-    ) -> list[ConnectionDetail]:
-        params = {
-            "query[date]": date.strftime("%d-%m-%Y %H:%M:%S"),
-            "query[start_station]": start,
-            "query[end_station]": end,
-            "query[only_purchasable]": str(purchasable).lower(),
-            "query[only_direct]": str(direct).lower(),
-            "query[brand_ids][]": brand_ids,
-        }
-        return self._get_json("/api/v2/main/connections", params=params)["connections"]
-
-    def get_brands(self) -> list[ApiBrand]:
-        # https://koleo.pl/api/v2/main/brands
-        return self._get_json(
-            "/api/v2/main/brands",
-        )
-
-    def get_carriers(self) -> list[Carrier]:
-        # https://koleo.pl/api/v2/main/carriers
-        return self._get_json(
-            "/api/v2/main/carriers",
-        )
-
-    def get_discounts(self) -> list[DiscountInfo]:
-        # https://koleo.pl/api/v2/main/discounts
-        return self._get_json(
-            "/api/v2/main/discounts",
-        )