gnomepy 2.3.0__tar.gz → 2.3.2__tar.gz

{gnomepy-2.3.0 → gnomepy-2.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gnomepy
-Version: 2.3.0
+Version: 2.3.2
 Summary: 
 Author: mprey
 Author-email: masonprey7@gmail.com

{gnomepy-2.3.0 → gnomepy-2.3.2}/gnomepy/registry/api.py

@@ -148,13 +148,25 @@ class RegistryClient:
         body = {_to_camel_case(k): v for k, v in kwargs.items()}
         return self._patch("/event-contracts", {"eventContractId": str(event_contract_id)}, body)
 
+    _PAGE_SIZE = 5000
+
     def _get(self, path: str, params: dict, output_type) -> list:
-        res = requests.get(self.base_url + path, params=params, headers={"x-api-key": self.api_key})
-        res.raise_for_status()
-        if dataclasses.is_dataclass(output_type):
-            known = {f.name for f in dataclasses.fields(output_type)}
-            return [output_type(**{k: v for k, v in item.items() if k in known}) for item in res.json()]
-        return [output_type(**item) for item in res.json()]
+        all_items = []
+        offset = 0
+        while True:
+            page_params = {**params, "limit": self._PAGE_SIZE, "offset": offset}
+            res = requests.get(self.base_url + path, params=page_params, headers={"x-api-key": self.api_key})
+            res.raise_for_status()
+            page = res.json()
+            if dataclasses.is_dataclass(output_type):
+                known = {f.name for f in dataclasses.fields(output_type)}
+                all_items.extend(output_type(**{k: v for k, v in item.items() if k in known}) for item in page)
+            else:
+                all_items.extend(output_type(**item) for item in page)
+            if len(page) < self._PAGE_SIZE:
+                break
+            offset += self._PAGE_SIZE
+        return all_items
 
     def _post_bulk(self, path: str, items: list[dict], batch_size: int = 500) -> list[dict]:
         results: list[dict] = []

{gnomepy-2.3.0 → gnomepy-2.3.2}/gnomepy/registry/types.py

@@ -105,7 +105,6 @@ class Event:
     category: str | None
     resolution_source: str | None
     tags: list[str] | None
-    embedding: list[float] | None
     resolved: bool
     resolved_at: str | None
     expiry: str | None
@@ -119,7 +118,6 @@ class EventContract:
     event_id: int
     security_id: int
     outcome_label: str
-    complement_security_id: int | None
     date_created: str
 
 

{gnomepy-2.3.0 → gnomepy-2.3.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "gnomepy"
-version = "2.3.0"
+version = "2.3.2"
 description = ""
 authors = [
     {name = "mprey",email = "masonprey7@gmail.com"}