stackshift 0.1.0__py3-none-any.whl

stackshift/__init__.py

@@ -0,0 +1,5 @@
+from .client import StackShift
+from .assets import AssetsClient
+from .projects import ProjectsClient
+
+__all__ = ["StackShift", "AssetsClient", "ProjectsClient"]

stackshift/assets.py

@@ -0,0 +1,89 @@
+from __future__ import annotations
+
+import json
+from pathlib import Path
+from typing import Any, BinaryIO
+
+
+class AssetsClient:
+    def __init__(self, client: Any) -> None:
+        self._client = client
+
+    def upload(
+        self,
+        file: str | Path | BinaryIO,
+        *,
+        bucket: str | None = None,
+        key: str | None = None,
+        folder: str | None = None,
+        visibility: str | None = None,
+        cache_control: str | None = None,
+        metadata: dict[str, Any] | None = None,
+    ) -> dict[str, Any]:
+        return self._multipart("POST", "/assets/upload", file, {
+            "bucket": bucket,
+            "key": key,
+            "folder": folder,
+            "visibility": visibility,
+            "cacheControl": cache_control,
+            "metadata": json.dumps(metadata) if metadata else None,
+        })
+
+    def list(self, **params: Any) -> dict[str, Any]:
+        clean = {key: value for key, value in params.items() if value not in (None, "")}
+        return self._client.request("GET", "/assets", params=clean)
+
+    def get(self, asset_id: str) -> dict[str, Any]:
+        return self._client.request("GET", f"/assets/{asset_id}")
+
+    def delete(self, asset_id: str) -> dict[str, Any]:
+        return self._client.request("DELETE", f"/assets/{asset_id}")
+
+    def replace(
+        self,
+        asset_id: str,
+        file: str | Path | BinaryIO,
+        *,
+        cache_control: str | None = None,
+        metadata: dict[str, Any] | None = None,
+    ) -> dict[str, Any]:
+        return self._multipart("PUT", f"/assets/{asset_id}/replace", file, {
+            "cacheControl": cache_control,
+            "metadata": json.dumps(metadata) if metadata else None,
+        })
+
+    def signed_url(
+        self,
+        asset_id: str,
+        *,
+        expires_in: str = "10m",
+        max_downloads: int | None = None,
+    ) -> dict[str, Any]:
+        return self._client.request("POST", f"/assets/{asset_id}/signed-url", json={
+            "expiresIn": expires_in,
+            "maxDownloads": max_downloads,
+        })
+
+    def signed_upload_url(self, **payload: Any) -> dict[str, Any]:
+        return self.create_upload_session(**payload)
+
+    def create_upload_session(self, **payload: Any) -> dict[str, Any]:
+        clean = {key: value for key, value in payload.items() if value not in (None, "")}
+        return self._client.request("POST", "/assets/upload-sessions", json=clean)
+
+    def _multipart(self, method: str, suffix: str, file: str | Path | BinaryIO, fields: dict[str, Any]) -> dict[str, Any]:
+        should_close = False
+        if isinstance(file, (str, Path)):
+            handle = Path(file).open("rb")
+            filename = Path(file).name
+            should_close = True
+        else:
+            handle = file
+            filename = getattr(file, "name", "asset")
+
+        try:
+            data = {key: value for key, value in fields.items() if value not in (None, "")}
+            return self._client.request(method, suffix, data=data, files={"file": (filename, handle)})
+        finally:
+            if should_close:
+                handle.close()

stackshift/client.py

@@ -0,0 +1,44 @@
+from __future__ import annotations
+
+import os
+from typing import Any
+
+import requests
+
+
+class StackShift:
+    def __init__(
+        self,
+        *,
+        api_key: str | None = None,
+        base_url: str = "https://api.stackshift.cloud/api/v1",
+        session: requests.Session | None = None,
+    ) -> None:
+        api_key = api_key or os.getenv("STACKSHIFT_API_KEY")
+        if not api_key:
+            raise ValueError("StackShift SDK requires an api_key")
+
+        self.api_key = api_key
+        self.base_url = base_url.rstrip("/")
+        self.session = session or requests.Session()
+
+        from .assets import AssetsClient
+        from .projects import ProjectsClient
+
+        self.projects = ProjectsClient(self)
+        self.assets = AssetsClient(self)
+
+    def request(self, method: str, path: str, **kwargs: Any) -> Any:
+        headers = kwargs.pop("headers", {})
+        headers["Authorization"] = f"Bearer {self.api_key}"
+        headers["User-Agent"] = "StackShift-Python/0.1"
+
+        response = self.session.request(method, f"{self.base_url}{path}", headers=headers, **kwargs)
+        payload = response.json() if response.content else None
+        if response.status_code < 200 or response.status_code >= 300:
+            message = payload.get("message") if isinstance(payload, dict) else None
+            raise RuntimeError(message or f"StackShift API request failed with {response.status_code}")
+
+        if isinstance(payload, dict) and "success" in payload and "data" in payload:
+            return payload["data"]
+        return payload

stackshift/projects.py

@@ -0,0 +1,16 @@
+from __future__ import annotations
+
+from typing import Any
+
+
+class ProjectsClient:
+    def __init__(self, client: Any) -> None:
+        self._client = client
+
+    def list(self, *, page: int | None = None, per_page: int | None = None) -> list[dict[str, Any]]:
+        params = {
+            key: value
+            for key, value in {"page": page, "per_page": per_page}.items()
+            if value not in (None, "")
+        }
+        return self._client.request("GET", "/projects/", params=params)

stackshift-0.1.0.dist-info/METADATA

@@ -0,0 +1,6 @@
+Metadata-Version: 2.4
+Name: stackshift
+Version: 0.1.0
+Summary: Official Python SDK for StackShift.
+Requires-Python: >=3.9
+Requires-Dist: requests>=2.31

stackshift-0.1.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+stackshift/__init__.py,sha256=Z0Q6SrKZyIx-YBt5pdIVXgnJGF4TeBYKtUA7vdxKlU4,161
+stackshift/assets.py,sha256=qvu1qXd5wJqeWc4hpdGYcorXXMtZUNEwjlaT61lu7-A,3119
+stackshift/client.py,sha256=YM9xgy0145f37l8x6A-Ecrg9zw5Z1E6DTRKEdiQ0P4g,1566
+stackshift/projects.py,sha256=oYWaE7xNRR6OXuPEwJH-oFFeOmDA9ozq6jJY09ANIHA,498
+stackshift-0.1.0.dist-info/METADATA,sha256=M9M2rhNT9c7bNRMCQkRTQ-Db8mYZ7T_Ccxva3KXCnq8,152
+stackshift-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+stackshift-0.1.0.dist-info/top_level.txt,sha256=qe3uWsu_O9mNtq7bIck9Y8rjTIMrH0kzgAlGFDoRGqs,11
+stackshift-0.1.0.dist-info/RECORD,,

stackshift-0.1.0.dist-info/WHEEL

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

stackshift-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+stackshift