hyperstudy 0.2.2__py3-none-any.whl → 0.2.3__py3-none-any.whl

hyperstudy/__init__.py

@@ -19,7 +19,7 @@ from .exceptions import (
     ValidationError,
 )
 
-__version__ = "0.2.2"
+__version__ = "0.2.3"
 
 __all__ = [
     "HyperStudy",

hyperstudy/_downloads.py

@@ -38,13 +38,39 @@ def build_filename(recording: dict[str, Any]) -> str:
 
 
 def download_file(url: str, dest: Path, timeout: int = 300) -> int:
-    """Stream-download *url* to *dest* and return bytes written."""
+    """Stream-download *url* to *dest* and return bytes written.
+
+    Verifies ``Content-Length`` (when the server provides it) against bytes
+    actually written. On any error — HTTP, mid-stream disconnect, or length
+    mismatch — the partial file on disk is removed so ``skip_existing``
+    logic won't mistake it for a complete download on retry.
+    """
     resp = requests.get(url, stream=True, timeout=timeout)
     resp.raise_for_status()
 
+    expected: int | None
+    if "Content-Length" in resp.headers:
+        try:
+            expected = int(resp.headers["Content-Length"])
+        except (TypeError, ValueError):
+            expected = None
+    else:
+        expected = None
+
     written = 0
-    with open(dest, "wb") as fh:
-        for chunk in resp.iter_content(chunk_size=_CHUNK_SIZE):
-            fh.write(chunk)
-            written += len(chunk)
+    # BaseException so Ctrl-C also cleans up, not just exceptions.
+    try:
+        with open(dest, "wb") as fh:
+            for chunk in resp.iter_content(chunk_size=_CHUNK_SIZE):
+                fh.write(chunk)
+                written += len(chunk)
+
+        if expected is not None and written != expected:
+            raise IOError(
+                f"Truncated download: wrote {written} of {expected} bytes for {dest.name}"
+            )
+    except BaseException:
+        dest.unlink(missing_ok=True)
+        raise
+
     return written

hyperstudy/_http.py

@@ -63,8 +63,14 @@ class HttpTransport:
     # Public helpers
     # ------------------------------------------------------------------
 
-    def get(self, path: str, params: dict[str, Any] | None = None) -> dict:
-        return self._request("GET", path, params=params)
+    def get(
+        self,
+        path: str,
+        params: dict[str, Any] | None = None,
+        *,
+        timeout: int | float | None = None,
+    ) -> dict:
+        return self._request("GET", path, params=params, timeout=timeout)
 
     def post(self, path: str, json: dict[str, Any] | None = None) -> dict:
         return self._request("POST", path, json=json)
@@ -81,7 +87,8 @@ class HttpTransport:
 
     def _request(self, method: str, path: str, **kwargs) -> dict:
         url = f"{self.base_url}/{path.lstrip('/')}"
-        kwargs.setdefault("timeout", self.timeout)
+        if kwargs.get("timeout") is None:
+            kwargs["timeout"] = self.timeout
 
         resp = self._session.request(method, url, **kwargs)
         return self._handle_response(resp)

hyperstudy/_pagination.py

@@ -16,6 +16,7 @@ def fetch_all_pages(
     *,
     page_size: int = 1000,
     progress: bool = True,
+    timeout: int | float | None = None,
 ) -> tuple[list[dict], dict]:
     """Fetch every page of a paginated endpoint.
 
@@ -28,7 +29,7 @@ def fetch_all_pages(
     params.setdefault("offset", 0)
 
     # First request to learn total
-    body = transport.get(path, params=params)
+    body = transport.get(path, params=params, timeout=timeout)
     metadata = body.get("metadata", {})
     pagination = metadata.get("pagination", {})
     total = pagination.get("total", 0)
@@ -43,7 +44,7 @@ def fetch_all_pages(
     while has_more:
         params["offset"] = pagination.get("nextOffset", params["offset"] + page_size)
 
-        body = transport.get(path, params=params)
+        body = transport.get(path, params=params, timeout=timeout)
         metadata = body.get("metadata", {})
         pagination = metadata.get("pagination", {})
 

hyperstudy/client.py

@@ -13,6 +13,7 @@ from ._downloads import build_filename, download_file, get_download_url
 from ._http import HttpTransport
 from ._pagination import fetch_all_pages
 from ._types import Scope
+from .exceptions import HyperStudyError
 from .experiments import ExperimentMixin
 
 
@@ -533,13 +534,24 @@ class HyperStudy(ExperimentMixin):
             pandas DataFrame with recording metadata plus ``local_path``
             and ``download_status`` columns.
         """
-        recordings = self.get_recordings(
-            scope_id,
-            scope=scope,
-            deployment_id=deployment_id,
-            room_id=room_id,
-            output="dict",
-        )
+        try:
+            # Signing per-recording GCS URLs scales with count; 30s default is too short.
+            recordings = self._fetch_data(
+                "recordings",
+                scope_id,
+                scope=scope,
+                deployment_id=deployment_id,
+                room_id=room_id,
+                output="dict",
+                timeout=300,
+            )
+        except Exception as exc:
+            raise HyperStudyError(
+                f"Failed to list recordings for {scope}={scope_id!r}: {exc}",
+                code=getattr(exc, "code", "LIST_RECORDINGS_FAILED"),
+                status_code=getattr(exc, "status_code", None),
+                details=getattr(exc, "details", None),
+            ) from exc
 
         if recording_type:
             recordings = [
@@ -566,7 +578,9 @@ class HyperStudy(ExperimentMixin):
 
             if skip_existing and dest.exists():
                 expected_size = rec.get("fileSize")
-                if expected_size is None or dest.stat().st_size == expected_size:
+                # Require positive size match; missing size → re-download to
+                # avoid trusting a truncated file from a prior failed run.
+                if expected_size is not None and dest.stat().st_size == expected_size:
                     local_paths.append(str(dest.resolve()))
                     statuses.append("skipped")
                     continue
@@ -582,6 +596,13 @@ class HyperStudy(ExperimentMixin):
                     f"Failed to download recording {rec.get('recordingId')}: {exc}"
                 )
 
+        failed_count = sum(1 for s in statuses if s == "failed")
+        if failed_count:
+            warnings.warn(
+                f"{failed_count}/{len(recordings)} recordings failed to download; "
+                "see the 'download_status' column for per-file results."
+            )
+
         df = to_pandas(recordings)
         if not df.empty:
             df["local_path"] = local_paths
@@ -649,6 +670,7 @@ class HyperStudy(ExperimentMixin):
         offset: int = 0,
         output: str = "pandas",
         progress: bool = True,
+        timeout: int | float | None = None,
         **extra_params,
     ):
         """Generic data-fetching logic shared by all ``get_*`` methods."""
@@ -675,11 +697,15 @@ class HyperStudy(ExperimentMixin):
         # Single page or all pages
         if limit is not None:
             params["limit"] = limit
-            body = self._transport.get(path, params=params)
+            body = self._transport.get(path, params=params, timeout=timeout)
             data = body.get("data", [])
         else:
             data, _ = fetch_all_pages(
-                self._transport, path, params=params, progress=progress
+                self._transport,
+                path,
+                params=params,
+                progress=progress,
+                timeout=timeout,
             )
 
         return self._convert_output(data, output)

{hyperstudy-0.2.2.dist-info → hyperstudy-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperstudy
-Version: 0.2.2
+Version: 0.2.3
 Summary: Python SDK for the HyperStudy experiment platform API
 Project-URL: Homepage, https://hyperstudy.io
 Project-URL: Documentation, https://docs.hyperstudy.io/developers/python-sdk

hyperstudy-0.2.3.dist-info/RECORD

@@ -0,0 +1,14 @@
+hyperstudy/__init__.py,sha256=-FMsxfDASUokTGR0-iES4eb-SUEjTuThGMqPIg-Imyw,733
+hyperstudy/_dataframe.py,sha256=8pjuMG8phMv8vwdY7sgZ-aF_6_o0UMgP2XXNp5muZ0s,3620
+hyperstudy/_display.py,sha256=5svr4NFOUJkVif57-xJzrXNOSdA5O9tKesu8UhE_jYo,1896
+hyperstudy/_downloads.py,sha256=MKUjzz9UDfo0V-n22UCdLEH2-tQxBeJMN_JoJ2spIT8,2459
+hyperstudy/_http.py,sha256=_RS0ZaRAcKvFQ2NF9II7PpcojE-lTYT6CslaJNlgfPQ,4569
+hyperstudy/_pagination.py,sha256=2NbvNGW_jFQkYquSTi3SNkAmGgsbKop7J9LkGaZWwWA,1812
+hyperstudy/_types.py,sha256=22JTOlM68eGJ4hg4OH-oT-hmARG2CgbhnERJ5hGxHk8,614
+hyperstudy/client.py,sha256=raAEBNxRsFPr260JWOFlOZNwmFk2SB80L2bsMoojUt0,24376
+hyperstudy/exceptions.py,sha256=z_1Ngiq-RU-1yyDI_JqvQ5CvYm2jQ_lyJ9TUbqUKKv8,1529
+hyperstudy/experiments.py,sha256=BHNf-ctKUA9uzy8Kya11nIyHPrsmOslqC_IJ5t4LxKA,4372
+hyperstudy-0.2.3.dist-info/METADATA,sha256=l1DluLkNI41NjQxrCEUAMeWMfi8jZFhXOUdC9RPb_f0,4833
+hyperstudy-0.2.3.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+hyperstudy-0.2.3.dist-info/licenses/LICENSE,sha256=WJdwRhBIOiiQ6lyK4l95u5IJYuuT-XxzwYRraS31NME,1067
+hyperstudy-0.2.3.dist-info/RECORD,,

hyperstudy-0.2.2.dist-info/RECORD

@@ -1,14 +0,0 @@
-hyperstudy/__init__.py,sha256=E8l_qKgFt0TPaZycaGM9L09ThfDSB30c4RqIS4IgolQ,733
-hyperstudy/_dataframe.py,sha256=8pjuMG8phMv8vwdY7sgZ-aF_6_o0UMgP2XXNp5muZ0s,3620
-hyperstudy/_display.py,sha256=5svr4NFOUJkVif57-xJzrXNOSdA5O9tKesu8UhE_jYo,1896
-hyperstudy/_downloads.py,sha256=nqIig09P84XbYLBlI5uPMWtFHISGt9H4qhsf5415kdg,1563
-hyperstudy/_http.py,sha256=wMVAhgxNcp4-zcrc4lTHU_GHGSqsFiyMFWsrLP5EGFM,4430
-hyperstudy/_pagination.py,sha256=Fdr2nW84r6NtXiYQs5jyJEUOjFvXl-RclpDIi-SPDlM,1738
-hyperstudy/_types.py,sha256=22JTOlM68eGJ4hg4OH-oT-hmARG2CgbhnERJ5hGxHk8,614
-hyperstudy/client.py,sha256=MTw81lv1sP16wDMIpF4QaJ6fsNFJmNy62PaAcUAL69k,23202
-hyperstudy/exceptions.py,sha256=z_1Ngiq-RU-1yyDI_JqvQ5CvYm2jQ_lyJ9TUbqUKKv8,1529
-hyperstudy/experiments.py,sha256=BHNf-ctKUA9uzy8Kya11nIyHPrsmOslqC_IJ5t4LxKA,4372
-hyperstudy-0.2.2.dist-info/METADATA,sha256=RSn3To1gXSS0SHnz-QXrtQc1uamcHNkPeDUuckeGSUI,4833
-hyperstudy-0.2.2.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
-hyperstudy-0.2.2.dist-info/licenses/LICENSE,sha256=WJdwRhBIOiiQ6lyK4l95u5IJYuuT-XxzwYRraS31NME,1067
-hyperstudy-0.2.2.dist-info/RECORD,,