ctftime-api 0.1.1__tar.gz → 0.1.3__tar.gz

{ctftime_api-0.1.1 → ctftime_api-0.1.3}/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.3
 Name: ctftime_api
-Version: 0.1.1
+Version: 0.1.3
 Summary: CTFTime API wrapper
 Keywords: ctftime,api,wrapper
 Author: John Toniutti
 Author-email: john.toniutti@gmail.com
 Requires-Python: >=3.11
-Classifier: Development Status :: 3 - Alpha
+Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)
 Classifier: Framework :: Pydantic
@@ -18,10 +18,16 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Internet :: WWW/HTTP
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Typing :: Typed
+Provides-Extra: speed
+Requires-Dist: brotli (>=1.1.0,<2.0.0) ; (platform_python_implementation == "CPython") and (extra == "speed")
+Requires-Dist: brotlicffi (>=1.1.0.0,<2.0.0.0) ; (platform_python_implementation != "CPython") and (extra == "speed")
+Requires-Dist: h2 (>=4.2.0,<5.0.0) ; extra == "speed"
 Requires-Dist: httpx (>=0.28.1,<0.29.0)
 Requires-Dist: pycountry (>=24.6.1,<25.0.0)
 Requires-Dist: pydantic (>=2.10.6,<3.0.0)
 Requires-Dist: pydantic-extra-types (>=2.10.2,<3.0.0)
+Requires-Dist: uvloop (>=0.21.0,<1.0.0) ; (sys_platform != "win32") and (extra == "speed")
+Requires-Dist: zstandard (>=0.23.0,<0.24.0) ; extra == "speed"
 Project-URL: Documentation, https://jotonedev.github.io/ctftime_api/
 Project-URL: Repository, https://github.com/jotonedev/ctftime_api
 Description-Content-Type: text/markdown

{ctftime_api-0.1.1 → ctftime_api-0.1.3}/ctftime_api/client.py

@@ -1,32 +1,47 @@
+import importlib.util
 from datetime import datetime
+from typing import Any
 
 import httpx
 from httpx import URL, Timeout
-
 from pydantic_extra_types.country import CountryAlpha2
 
-from ctftime_api.models.vote import Vote
 from ctftime_api.models.event import Event, EventResult
 from ctftime_api.models.team import TeamRank, Team, TeamComplete
-
+from ctftime_api.models.vote import Vote
 
 __all__ = ["CTFTimeClient"]
 
+h2 = importlib.util.find_spec("httpcore.h2")
+
 
 class CTFTimeClient:
-    def __init__(self, client: httpx.AsyncClient | None = None, *args, **kwargs):
+    def __init__(self, client: httpx.AsyncClient | None = None, **kwargs):
         """
         Initialize the CTFTime API client.
         :param client: The httpx.AsyncClient to use. If None, a new client will be created.
-        :param args: Args that will be passed to the httpx.AsyncClient constructor.
         :param kwargs: Kwargs that will be passed to the httpx.AsyncClient constructor.
         """
         if client is not None:
             self._client = client
         else:
-            self._client = httpx.AsyncClient(*args, **kwargs)
+            if h2 is not None:
+                kwargs.setdefault("http2", True)
+            self._client = httpx.AsyncClient(**kwargs)
         self._base_url = URL("https://ctftime.org/api/v1/")
 
+    async def _get(self, url: str | URL, **kwargs) -> Any:
+        """
+        Perform a GET request.
+        :param url: The url to make the request to.
+        :param kwargs: Additional arguments to pass to the request.
+        :return: The response JSON.
+        :raise httpx.HTTPStatusError: If the response status code is not successful.
+        """
+        response = await self._client.get(url, **kwargs)
+        response.raise_for_status()
+        return response.json()
+
     async def close(self):
         """Close the underlying httpx.AsyncClient."""
         await self._client.aclose()
@@ -39,16 +54,16 @@ class CTFTimeClient:
         :param year: The year to get the top teams for. If None, the current year will be used.
         :param limit: The number of teams to get.
         :return: A list of the top teams.
+        :raise httpx.HTTPStatusError: If the response status code is not successful.
         """
         if year is None:
             url = self._base_url.join("top/")
             year = datetime.now().year
         else:
             url = self._base_url.join(f"top/{year}/")
-        response = await self._client.get(url, params={"limit": limit})
-        response.raise_for_status()
 
-        teams = response.json().get(f"{year}", [])
+        response: dict[str, list[dict]] = await self._get(url, params={"limit": limit})
+        teams = response.get(f"{year}", [])
 
         return [TeamRank.model_validate(team) for team in teams]
 
@@ -60,6 +75,8 @@ class CTFTimeClient:
         :param country: The country to get the top teams for.
             It can be a pycountry Country object or a two-letter country code.
         :return: A list of the top teams.
+        :raise httpx.HTTPStatusError: If the response status code is not successful.
+        :raise ValueError: If the country is not a two-letter country code or a pycountry Country object.
         """
         if isinstance(country, CountryAlpha2):
             country = country
@@ -70,10 +87,7 @@ class CTFTimeClient:
                 )
 
         url = self._base_url.join("top-by-country/").join(f"{country}/")
-        response = await self._client.get(url)
-        response.raise_for_status()
-
-        teams = response.json()
+        teams: list[dict[str, Any]] = await self._get(url)
 
         return [TeamRank.model_validate(team) for team in teams]
 
@@ -88,6 +102,8 @@ class CTFTimeClient:
             It can be a Unix timestamp or a datetime object.
         :param limit: The number of events to get.
         :return: A list of events.
+        :raise httpx.HTTPStatusError: If the response status code is not successful.
+        :raise ValueError: If the start date is after the end date.
         """
         if isinstance(start, datetime):
             start = int(start.timestamp())
@@ -98,12 +114,9 @@ class CTFTimeClient:
             raise ValueError("The start date must be before the end date.")
 
         url = self._base_url.join("events/")
-        response = await self._client.get(
+        events: list[dict[str, Any]] = await self._get(
             url, params={"start": start, "finish": end, "limit": limit}
         )
-        response.raise_for_status()
-
-        events = response.json()
 
         return [Event.model_validate(event) for event in events]
 
@@ -112,12 +125,10 @@ class CTFTimeClient:
         Get information about a specific event.
         :param event_id: The ID of the event.
         :return: The event information.
+        :raise httpx.HTTPStatusError: If the response status code is not successful.
         """
         url = self._base_url.join(f"events/{event_id}/")
-        response = await self._client.get(url)
-        response.raise_for_status()
-
-        event = response.json()
+        event: dict[str, Any] = await self._get(url)
 
         return Event.model_validate(event)
 
@@ -129,14 +140,14 @@ class CTFTimeClient:
         :param limit: The number of teams to get.
         :param offset: The offset to start from.
         :return: A list of teams.
+        :raise httpx.HTTPStatusError: If the response status code is not successful.
         """
         url = self._base_url.join("teams/")
-        response = await self._client.get(
+
+        response: dict[str, Any] = await self._get(
             url, params={"limit": limit, "offset": offset}
         )
-        response.raise_for_status()
-
-        teams = response.json().get("results", [])
+        teams: list[dict[str, Any]] = response.get("results", [])
 
         return [Team.model_validate(team) for team in teams]
 
@@ -145,12 +156,10 @@ class CTFTimeClient:
         Get information about a specific team.
         :param team_id: The ID of the team.
         :return: The team information.
+        :raise httpx.HTTPStatusError: If the response status code is not successful.
         """
         url = self._base_url.join(f"teams/{team_id}/")
-        response = await self._client.get(url)
-        response.raise_for_status()
-
-        team = response.json()
+        team: dict[str, Any] = await self._get(url)
 
         return TeamComplete.model_validate(team)
 
@@ -162,19 +171,17 @@ class CTFTimeClient:
         :param year: The year to get the results for.
             If None, the current year will be used.
         :return: A dictionary of event results.
+        :raise httpx.HTTPStatusError: If the response status code is not successful.
         """
         if year is None:
             url = self._base_url.join("results/")
         else:
             url = self._base_url.join(f"results/{year}/")
 
-        response = await self._client.get(url)
-        response.raise_for_status()
-
-        event = response.json()
+        event: dict[str, dict] = await self._get(url)
 
         return {
-            int(ctf_id): EventResult(**result, ctf_id=ctf_id)
+            int(ctf_id): EventResult(**result, ctf_id=int(ctf_id))
             for ctf_id, result in event.items()
         }
 
@@ -188,6 +195,7 @@ class CTFTimeClient:
         :param timeout: The timeout for the request.
             If None, the session timeout will be used.
         :return: A list of votes.
+        :raise httpx.HTTPStatusError: If the response status code is not successful.
         """
         if year is None:
             year = datetime.now().year
@@ -196,9 +204,6 @@ class CTFTimeClient:
             timeout = self._client.timeout
 
         url = self._base_url.join(f"votes/{year}/")
-        response = await self._client.get(url, timeout=timeout)
-        response.raise_for_status()
-
-        votes = response.json()
+        votes: list[dict] = await self._get(url, timeout=timeout)
 
         return [Vote(**vote) for vote in votes]

{ctftime_api-0.1.1 → ctftime_api-0.1.3}/ctftime_api/models/__init__.py

@@ -1,4 +1,4 @@
+from .duration import Duration
 from .event import Event, EventResult
-from .team import Team, TeamComplete, BaseTeam, TeamRank, TeamResult
 from .rating import Rating
-from .duration import Duration
+from .team import Team, TeamComplete, BaseTeam, TeamRank, TeamResult

{ctftime_api-0.1.1 → ctftime_api-0.1.3}/ctftime_api/models/duration.py

@@ -1,6 +1,5 @@
 from pydantic import BaseModel
 
-
 __all__ = ["Duration"]
 
 

{ctftime_api-0.1.1 → ctftime_api-0.1.3}/ctftime_api/models/event.py

@@ -1,6 +1,6 @@
-from pydantic import BaseModel, HttpUrl
 from datetime import datetime
 
+from pydantic import BaseModel, HttpUrl
 
 from .duration import Duration
 from .team import BaseTeam, TeamResult

{ctftime_api-0.1.1 → ctftime_api-0.1.3}/ctftime_api/models/team.py

@@ -5,7 +5,6 @@ from pydantic_extra_types.country import CountryAlpha2
 
 from .rating import Rating
 
-
 __all__ = ["BaseTeam", "Team", "TeamRank", "TeamComplete", "TeamResult"]
 
 

{ctftime_api-0.1.1 → ctftime_api-0.1.3}/ctftime_api/models/vote.py

@@ -2,7 +2,6 @@ from datetime import datetime
 
 from pydantic import BaseModel
 
-
 __all__ = ["Vote"]
 
 

{ctftime_api-0.1.1 → ctftime_api-0.1.3}/pyproject.toml

@@ -1,13 +1,13 @@
 [project]
 name = "ctftime_api"
-version = "0.1.1"
+version = "0.1.3"
 description = "CTFTime API wrapper"
 authors = [
     {name = "John Toniutti",email = "john.toniutti@gmail.com"}
 ]
 keywords = ["ctftime", "api", "wrapper"]
 classifiers = [
-    "Development Status :: 3 - Alpha",
+    "Development Status :: 4 - Beta",
     "Intended Audience :: Developers",
     "License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)",
     "Framework :: Pydantic",
@@ -29,6 +29,15 @@ dependencies = [
     "pycountry (>=24.6.1,<25.0.0)",
 ]
 
+[project.optional-dependencies]
+speed = [
+    "uvloop (>=0.21.0,<1.0.0) ; sys_platform != 'win32'",
+    "brotli (>=1.1.0,<2.0.0) ; platform_python_implementation == 'CPython' ",
+    "brotlicffi (>=1.1.0.0,<2.0.0.0) ; platform_python_implementation != 'CPython' ",
+    "h2 (>=4.2.0,<5.0.0)",
+    "zstandard (>=0.23.0,<0.24.0)",
+]
+
 [project.urls]
 repository = "https://github.com/jotonedev/ctftime_api"
 documentation = "https://jotonedev.github.io/ctftime_api/"