stackmachine 0.3.0__tar.gz → 0.3.2__tar.gz

{stackmachine-0.3.0 → stackmachine-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stackmachine
-Version: 0.3.0
+Version: 0.3.2
 Summary: Python SDK for StackMachine.
 Project-URL: Homepage, https://github.com/stackmachine/sdks/tree/main/python
 Project-URL: Repository, https://github.com/stackmachine/sdks
@@ -15,6 +15,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Typing :: Typed
 Requires-Python: >=3.9
 Requires-Dist: httpx<1,>=0.27
+Requires-Dist: typing-extensions<5,>=4.10
 Requires-Dist: websocket-client<2,>=1.8
 Requires-Dist: websockets<16,>=12
 Description-Content-Type: text/markdown
@@ -60,15 +61,13 @@ stackmachine = StackMachine("sk_stackmachine_...")
 async_stackmachine = AsyncStackMachine("sk_stackmachine_...")
 ```
 
-Both clients accept JavaScript-style aliases during initialization:
+Both clients accept configuration options during initialization:
 
 ```python
-stackmachine = StackMachine.init(
-    {
-        "token": "sk_stackmachine_...",
-        "apiUrl": "https://api.stackmachine.com/graphql",
-        "maxNetworkRetries": 2,
-    }
+stackmachine = StackMachine(
+    "sk_stackmachine_...",
+    apiUrl="https://api.stackmachine.com/graphql",
+    maxNetworkRetries=2,
 )
 ```
 
@@ -100,10 +99,8 @@ async for app in apps:
 
 ```python
 deployment = stackmachine.deployments.create(
-    {
-        "file": "https://example.com/app.zip",
-        "name": "my-app",
-    }
+    upload_url="https://example.com/app.zip",
+    app_name="my-app",
 )
 
 version = deployment.wait()
@@ -111,10 +108,8 @@ version = deployment.wait()
 
 ```python
 deployment = stackmachine.apps.autobuild(
-    {
-        "file": "https://example.com/app.zip",
-        "name": "my-app",
-    }
+    upload_url="https://example.com/app.zip",
+    app_name="my-app",
 )
 ```
 
@@ -161,13 +156,15 @@ key = stackmachine.apps.ssh.users.authorized_keys.create(
 Most methods accept `request_options` for per-request configuration:
 
 ```python
+from stackmachine import RequestOptions
+
 app = stackmachine.apps.retrieve(
     "app_id",
-    request_options={
-        "api_key": "sk_stackmachine_other",
-        "timeout": 30,
-        "idempotency_key": "deploy-123",
-    },
+    request_options=RequestOptions(
+        api_key="sk_stackmachine_other",
+        timeout=30,
+        idempotency_key="deploy-123",
+    ),
 )
 ```
 
@@ -177,6 +174,7 @@ app = stackmachine.apps.retrieve(
 cd python
 uv sync --dev
 uv run ruff check src examples tests
+uv run mypy
 uv run pytest
 uv build --no-sources
 ```

{stackmachine-0.3.0 → stackmachine-0.3.2}/README.md

@@ -39,15 +39,13 @@ stackmachine = StackMachine("sk_stackmachine_...")
 async_stackmachine = AsyncStackMachine("sk_stackmachine_...")
 ```
 
-Both clients accept JavaScript-style aliases during initialization:
+Both clients accept configuration options during initialization:
 
 ```python
-stackmachine = StackMachine.init(
-    {
-        "token": "sk_stackmachine_...",
-        "apiUrl": "https://api.stackmachine.com/graphql",
-        "maxNetworkRetries": 2,
-    }
+stackmachine = StackMachine(
+    "sk_stackmachine_...",
+    apiUrl="https://api.stackmachine.com/graphql",
+    maxNetworkRetries=2,
 )
 ```
 
@@ -79,10 +77,8 @@ async for app in apps:
 
 ```python
 deployment = stackmachine.deployments.create(
-    {
-        "file": "https://example.com/app.zip",
-        "name": "my-app",
-    }
+    upload_url="https://example.com/app.zip",
+    app_name="my-app",
 )
 
 version = deployment.wait()
@@ -90,10 +86,8 @@ version = deployment.wait()
 
 ```python
 deployment = stackmachine.apps.autobuild(
-    {
-        "file": "https://example.com/app.zip",
-        "name": "my-app",
-    }
+    upload_url="https://example.com/app.zip",
+    app_name="my-app",
 )
 ```
 
@@ -140,13 +134,15 @@ key = stackmachine.apps.ssh.users.authorized_keys.create(
 Most methods accept `request_options` for per-request configuration:
 
 ```python
+from stackmachine import RequestOptions
+
 app = stackmachine.apps.retrieve(
     "app_id",
-    request_options={
-        "api_key": "sk_stackmachine_other",
-        "timeout": 30,
-        "idempotency_key": "deploy-123",
-    },
+    request_options=RequestOptions(
+        api_key="sk_stackmachine_other",
+        timeout=30,
+        idempotency_key="deploy-123",
+    ),
 )
 ```
 
@@ -156,6 +152,7 @@ app = stackmachine.apps.retrieve(
 cd python
 uv sync --dev
 uv run ruff check src examples tests
+uv run mypy
 uv run pytest
 uv build --no-sources
 ```

{stackmachine-0.3.0 → stackmachine-0.3.2}/pyproject.toml

@@ -4,12 +4,13 @@ build-backend = "hatchling.build"
 
 [project]
 name = "stackmachine"
-version = "0.3.0"
+version = "0.3.2"
 description = "Python SDK for StackMachine."
 readme = "README.md"
 requires-python = ">=3.9"
 dependencies = [
   "httpx>=0.27,<1",
+  "typing-extensions>=4.10,<5",
   "websocket-client>=1.8,<2",
   "websockets>=12,<16",
 ]
@@ -31,6 +32,7 @@ Issues = "https://github.com/stackmachine/sdks/issues"
 
 [dependency-groups]
 dev = [
+  "mypy>=1.13,<2",
   "pytest>=8.0",
   "pytest-asyncio>=0.24",
   "ruff>=0.8",
@@ -43,6 +45,17 @@ target-version = "py39"
 [tool.ruff.lint]
 select = ["E", "F", "I", "B"]
 
+[tool.mypy]
+python_version = "3.9"
+files = ["src", "examples", "tests"]
+pretty = true
+show_error_codes = true
+warn_unused_ignores = true
+
+[[tool.mypy.overrides]]
+module = ["pytest", "pytest.*"]
+follow_imports = "skip"
+
 [tool.pytest.ini_options]
 testpaths = ["tests"]
 asyncio_mode = "auto"

stackmachine-0.3.2/src/stackmachine/__init__.py

@@ -0,0 +1,144 @@
+"""StackMachine Python SDK."""
+
+from importlib.metadata import PackageNotFoundError, version
+
+from ._async_client import AsyncStackMachine
+from ._client import StackMachine
+from ._config import RequestOptions
+from ._errors import (
+    StackMachineAPIError,
+    StackMachineAuthenticationError,
+    StackMachineConnectionError,
+    StackMachineError,
+    StackMachineGraphQLError,
+    StackMachineInvalidRequestError,
+    StackMachinePermissionError,
+    StackMachineRateLimitError,
+    StackMachineValidationError,
+    is_stackmachine_error,
+)
+from ._models import (
+    AppAlias,
+    AppSshServer,
+    DeployApp,
+    DeployAppKindWordPress,
+    DeployAppVersion,
+    DeploymentProgress,
+    ExpectedDNSRecord,
+    Log,
+    SshAuthorizedKey,
+    SshUser,
+    UploadProgress,
+    Viewer,
+)
+from ._pagination import AsyncStackMachineList, StackMachineList
+from ._types import (
+    AppAliasSortBy,
+    AppsDomainsCreateInput,
+    AppsDomainsListInput,
+    AppsSshAuthorizedKeysCreateInput,
+    AppsSshAuthorizedKeysDeleteInput,
+    AppsSshAuthorizedKeysListInput,
+    AppsSshServerUpdateInput,
+    AppsSshUsersListInput,
+    AppsSshUsersUpdateInput,
+    AppsVersionsListInput,
+    AppsVersionsLogsListInput,
+    AsyncStackMachineInitSettings,
+    CreateZipFiles,
+    DeployAppAutobuildExtraData,
+    DeployAppAutobuildInput,
+    DeployAppEnvVarInput,
+    DeployAppJobDefinitionInput,
+    DeployAppsListInput,
+    DeployAppsSortBy,
+    DeployAppVersionsSortBy,
+    DeployAppWordPressExtraData,
+    DeploymentProgressCallback,
+    FileInput,
+    Headers,
+    LogStream,
+    PaginationOptions,
+    Readable,
+    RequestOptionsInput,
+    RequestOptionsLike,
+    SshAuthenticationMethod,
+    SshTokenCreateResult,
+    SshUserPasswordRevealResult,
+    SshUserPasswordRotateResult,
+    StackMachineInitSettings,
+    UploadProgressCallback,
+)
+from ._uploads import create_zip
+
+try:
+    __version__ = version("stackmachine")
+except PackageNotFoundError:
+    __version__ = "0.3.2"
+
+__all__ = [
+    "AppAlias",
+    "AppAliasSortBy",
+    "AppSshServer",
+    "AppsDomainsCreateInput",
+    "AppsDomainsListInput",
+    "AppsSshAuthorizedKeysCreateInput",
+    "AppsSshAuthorizedKeysDeleteInput",
+    "AppsSshAuthorizedKeysListInput",
+    "AppsSshServerUpdateInput",
+    "AppsSshUsersListInput",
+    "AppsSshUsersUpdateInput",
+    "AppsVersionsListInput",
+    "AppsVersionsLogsListInput",
+    "AsyncStackMachine",
+    "AsyncStackMachineInitSettings",
+    "AsyncStackMachineList",
+    "CreateZipFiles",
+    "DeployApp",
+    "DeployAppAutobuildExtraData",
+    "DeployAppAutobuildInput",
+    "DeployAppEnvVarInput",
+    "DeployAppJobDefinitionInput",
+    "DeployAppKindWordPress",
+    "DeployAppVersion",
+    "DeployAppVersionsSortBy",
+    "DeployAppWordPressExtraData",
+    "DeployAppsListInput",
+    "DeployAppsSortBy",
+    "DeploymentProgress",
+    "DeploymentProgressCallback",
+    "ExpectedDNSRecord",
+    "FileInput",
+    "Headers",
+    "Log",
+    "LogStream",
+    "PaginationOptions",
+    "Readable",
+    "RequestOptions",
+    "RequestOptionsInput",
+    "RequestOptionsLike",
+    "SshAuthenticationMethod",
+    "SshAuthorizedKey",
+    "SshTokenCreateResult",
+    "SshUser",
+    "SshUserPasswordRevealResult",
+    "SshUserPasswordRotateResult",
+    "StackMachine",
+    "StackMachineAPIError",
+    "StackMachineAuthenticationError",
+    "StackMachineConnectionError",
+    "StackMachineError",
+    "StackMachineGraphQLError",
+    "StackMachineInvalidRequestError",
+    "StackMachineList",
+    "StackMachineInitSettings",
+    "StackMachinePermissionError",
+    "StackMachineRateLimitError",
+    "StackMachineValidationError",
+    "UploadProgress",
+    "UploadProgressCallback",
+    "Viewer",
+    "__version__",
+    "create_zip",
+    "is_stackmachine_error",
+]

stackmachine-0.3.2/src/stackmachine/_async_client.py

@@ -0,0 +1,198 @@
+from __future__ import annotations
+
+from typing import Any, Mapping, Optional
+
+import httpx
+
+from ._config import (
+    DEFAULT_API_URL,
+    DEFAULT_MAX_NETWORK_RETRIES,
+    DEFAULT_TIMEOUT,
+    ClientConfig,
+)
+from ._graphql import operations as gql
+from ._models import Viewer
+from ._transport import AsyncTransport
+from ._types import AsyncStackMachineInitSettings, Headers, RequestOptionsLike
+from ._uploads import AsyncUploader
+from .resources.apps import AsyncDeployAppsResource
+from .resources.deployments import AsyncDeploymentsResource
+from .resources.files import AsyncFilesResource
+
+
+class AsyncStackMachine:
+    def __init__(
+        self,
+        api_key: str,
+        *,
+        api_url: Optional[str] = None,
+        apiUrl: Optional[str] = None,
+        headers: Optional[Headers] = None,
+        timeout: float = DEFAULT_TIMEOUT,
+        max_network_retries: Optional[int] = None,
+        maxNetworkRetries: Optional[int] = None,
+        http_client: Optional[httpx.AsyncClient] = None,
+        http_transport: Optional[httpx.AsyncBaseTransport] = None,
+    ) -> None:
+        resolved_api_url = (
+            api_url if api_url is not None else apiUrl or DEFAULT_API_URL
+        )
+        resolved_max_retries = (
+            max_network_retries
+            if max_network_retries is not None
+            else maxNetworkRetries
+            if maxNetworkRetries is not None
+            else DEFAULT_MAX_NETWORK_RETRIES
+        )
+        self.api_key = api_key
+        self.api_url = resolved_api_url
+        self.apiUrl = resolved_api_url
+        self.timeout = timeout
+        self.max_network_retries = resolved_max_retries
+        self.maxNetworkRetries = resolved_max_retries
+        self._config = ClientConfig(
+            api_url=resolved_api_url,
+            headers=headers,
+            timeout=timeout,
+            max_network_retries=resolved_max_retries,
+        )
+        self._transport = AsyncTransport(
+            api_key,
+            self._config,
+            http_client=http_client,
+            http_transport=http_transport,
+        )
+        self.deployments = AsyncDeploymentsResource(self)
+        self.apps = AsyncDeployAppsResource(self, self.deployments)
+        self.files = AsyncFilesResource(self, AsyncUploader(self._transport))
+
+    @classmethod
+    def init(
+        cls,
+        settings: Optional[AsyncStackMachineInitSettings] = None,
+        *,
+        api_key: Optional[str] = None,
+        apiKey: Optional[str] = None,
+        token: Optional[str] = None,
+        api_url: Optional[str] = None,
+        apiUrl: Optional[str] = None,
+        headers: Optional[Headers] = None,
+        timeout: Optional[float] = None,
+        max_network_retries: Optional[int] = None,
+        maxNetworkRetries: Optional[int] = None,
+        http_client: Optional[httpx.AsyncClient] = None,
+        http_transport: Optional[httpx.AsyncBaseTransport] = None,
+    ) -> "AsyncStackMachine":
+        values: AsyncStackMachineInitSettings = {} if settings is None else settings
+        resolved_api_key = (
+            api_key
+            if api_key is not None
+            else apiKey
+            if apiKey is not None
+            else token
+            if token is not None
+            else values.get("api_key")
+            or values.get("apiKey")
+            or values.get("token")
+            or ""
+        )
+        settings_timeout = values.get("timeout")
+        resolved_timeout = (
+            timeout
+            if timeout is not None
+            else settings_timeout
+            if settings_timeout is not None
+            else DEFAULT_TIMEOUT
+        )
+        resolved_api_url = (
+            api_url
+            if api_url is not None
+            else apiUrl
+            if apiUrl is not None
+            else values.get("api_url")
+            or values.get("apiUrl")
+            or DEFAULT_API_URL
+        )
+        settings_max_retries = (
+            values.get("max_network_retries")
+            if values.get("max_network_retries") is not None
+            else values.get("maxNetworkRetries")
+        )
+        resolved_max_retries = (
+            max_network_retries
+            if max_network_retries is not None
+            else maxNetworkRetries
+            if maxNetworkRetries is not None
+            else settings_max_retries
+            if settings_max_retries is not None
+            else DEFAULT_MAX_NETWORK_RETRIES
+        )
+        return cls(
+            resolved_api_key,
+            api_url=resolved_api_url,
+            headers=headers if headers is not None else values.get("headers"),
+            timeout=resolved_timeout,
+            max_network_retries=resolved_max_retries,
+            http_client=(
+                http_client if http_client is not None else values.get("http_client")
+            ),
+            http_transport=(
+                http_transport
+                if http_transport is not None
+                else values.get("http_transport")
+            ),
+        )
+
+    async def close(self) -> None:
+        await self._transport.close()
+
+    async def __aenter__(self) -> "AsyncStackMachine":
+        return self
+
+    async def __aexit__(self, *exc_info: object) -> None:
+        await self.close()
+
+    async def _query(
+        self,
+        query: str,
+        variables: Optional[Mapping[str, Any]] = None,
+        *,
+        request_options: Optional[RequestOptionsLike] = None,
+    ) -> Any:
+        return await self._transport.execute(
+            query,
+            variables,
+            request_options=request_options,
+        )
+
+    async def _mutation(
+        self,
+        query: str,
+        variables: Optional[Mapping[str, Any]] = None,
+        *,
+        request_options: Optional[RequestOptionsLike] = None,
+    ) -> Any:
+        return await self._transport.execute(
+            query,
+            variables,
+            request_options=request_options,
+            mutation=True,
+        )
+
+    def _subscribe_deployment(
+        self, build_id: str, request_options: Optional[RequestOptionsLike] = None
+    ):
+        return self._transport.subscribe(
+            gql.AUTOBUILD_SUBSCRIPTION,
+            {"buildId": build_id},
+            request_options=request_options,
+        )
+
+    async def viewer(
+        self, *, request_options: Optional[RequestOptionsLike] = None
+    ) -> Optional[Viewer]:
+        response = await self._query(
+            gql.VIEWER_QUERY, {}, request_options=request_options
+        )
+        viewer = response.get("viewer") if response else None
+        return Viewer(username=viewer["username"]) if viewer else None