xenfra-sdk 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

xenfra_sdk/resources/deployments.py

@@ -1,4 +1,6 @@
+import json
 import logging
+from typing import Iterator
 
 # Import Deployment model when it's defined in models.py
 # from ..models import Deployment
@@ -10,7 +12,7 @@ logger = logging.getLogger(__name__)
 
 
 class DeploymentsManager(BaseManager):
-    def create(self, project_name: str, git_repo: str, branch: str, framework: str) -> dict:
+    def create(self, project_name: str, git_repo: str, branch: str, framework: str, region: str = None, size_slug: str = None) -> dict:
         """Creates a new deployment."""
         try:
             payload = {
@@ -19,6 +21,11 @@ class DeploymentsManager(BaseManager):
                 "branch": branch,
                 "framework": framework,
             }
+            if region:
+                payload["region"] = region
+            if size_slug:
+                payload["size_slug"] = size_slug
+            
             response = self._client._request("POST", "/deployments", json=payload)
             # Safe JSON parsing
             return safe_json_parse(response)
@@ -87,3 +94,92 @@ class DeploymentsManager(BaseManager):
             raise  # Re-raise API errors
         except Exception as e:
             raise XenfraError(f"Failed to get logs for deployment {deployment_id}: {e}")
+
+    def create_stream(self, project_name: str, git_repo: str, branch: str, framework: str, region: str = None, size_slug: str = None) -> Iterator[dict]:
+        """
+        Creates a new deployment with real-time SSE log streaming.
+
+        Yields SSE events as dictionaries with 'event' and 'data' keys.
+
+        Args:
+            project_name: Name of the project
+            git_repo: Git repository URL
+            branch: Git branch to deploy
+            framework: Framework type (fastapi, flask, django)
+            region: DigitalOcean region (optional)
+            size_slug: DigitalOcean droplet size (optional)
+
+        Yields:
+            dict: SSE events with 'event' and 'data' fields
+
+        Example:
+            for event in client.deployments.create_stream(...):
+                if event['event'] == 'log':
+                    print(event['data'])
+                elif event['event'] == 'deployment_complete':
+                    print("Done!")
+        """
+        payload = {
+            "project_name": project_name,
+            "git_repo": git_repo,
+            "branch": branch,
+            "framework": framework,
+        }
+        if region:
+            payload["region"] = region
+        if size_slug:
+            payload["size_slug"] = size_slug
+
+        try:
+            # Use httpx to stream the SSE response
+            import httpx
+
+            headers = {
+                "Authorization": f"Bearer {self._client.token}",
+                "Accept": "text/event-stream",
+                "Content-Type": "application/json",
+            }
+
+            url = f"{self._client.api_url}/deployments/stream"
+
+            with httpx.stream(
+                "POST",
+                url,
+                json=payload,
+                headers=headers,
+                timeout=600.0,  # 10 minute timeout for deployments
+            ) as response:
+                if response.status_code not in [200, 201, 202]:
+                    error_text = response.text
+                    raise XenfraAPIError(
+                        status_code=response.status_code,
+                        detail=f"Deployment failed: {error_text}"
+                    )
+
+                # Parse SSE events
+                for line in response.iter_lines():
+                    line = line.strip()
+                    if not line:
+                        continue
+
+                    # SSE format: "event: eventname" or "data: eventdata"
+                    if line.startswith("event:"):
+                        current_event = line[6:].strip()
+                    elif line.startswith("data:"):
+                        data = line[5:].strip()
+                        try:
+                            # Try to parse as JSON
+                            data_parsed = json.loads(data)
+                            yield {"event": current_event if 'current_event' in locals() else "message", "data": data_parsed}
+                        except json.JSONDecodeError:
+                            # If not JSON, yield as plain text
+                            yield {"event": current_event if 'current_event' in locals() else "message", "data": data}
+
+                        # Reset current_event
+                        if 'current_event' in locals():
+                            del current_event
+
+        except httpx.HTTPError as e:
+            raise XenfraError(f"HTTP error during streaming deployment: {e}")
+        except Exception as e:
+            raise XenfraError(f"Failed to create streaming deployment: {e}")

{xenfra_sdk-0.1.3.dist-info → xenfra_sdk-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: xenfra-sdk
-Version: 0.1.3
+Version: 0.1.4
 Summary: Xenfra SDK: Core engine and utilities for the Xenfra platform.
 Author: xenfra-cloud
 Author-email: xenfra-cloud <xenfracloud@gmail.com>

{xenfra_sdk-0.1.3.dist-info → xenfra_sdk-0.1.4.dist-info}/RECORD

@@ -18,7 +18,7 @@ xenfra_sdk/privacy.py,sha256=ksGf5L9PVtRP-xZS3T-Gj7MKfexTqIMgbFLoYkIESOE,5662
 xenfra_sdk/recipes.py,sha256=g_UKQIcdSokYh7zn186mzDTr08P034-KZ1iiDNELyP4,877
 xenfra_sdk/resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 xenfra_sdk/resources/base.py,sha256=C6BuZfhR-oU5ecHSfkGG6ZLU6MHGXUyCWoy1F-yTIf8,84
-xenfra_sdk/resources/deployments.py,sha256=JwlsCUKQUyvUF0lYUv6A3n1EkglXoYsICNVssx8E6ec,3428
+xenfra_sdk/resources/deployments.py,sha256=GmgX3F-mr3mDR_vJzidqDmkOhJKiFA9adQmSiBrSdzM,7272
 xenfra_sdk/resources/intelligence.py,sha256=Y11K6_iXfm2QKTbH1vUmt45MifLoVtZtlHEkqbzmTzs,3418
 xenfra_sdk/resources/projects.py,sha256=EsCVXmqkhWl_Guz_8WDQDi3kAm1Wyg1rjXcyAigPD6E,3712
 xenfra_sdk/security.py,sha256=Px887RRb1BUDXaPUrxmQITJ1mHyOyupCJqEDZ78F7Tk,1240
@@ -26,6 +26,6 @@ xenfra_sdk/templates/Dockerfile.j2,sha256=GXc0JiaF-HsxTQS15Gs2fcvsIhA1EHnwapdFVi
 xenfra_sdk/templates/cloud-init.sh.j2,sha256=NKIwtL9OgnlK2NnYRZI3gWC9aYl6wNPsS6r14g8eHQQ,2290
 xenfra_sdk/templates/docker-compose.yml.j2,sha256=qMHiatuZlxiYZ1pE_g2ag1M798MvQbeq0cVTVK07jkM,893
 xenfra_sdk/utils.py,sha256=d8eCjjV32QwqoJa759CEcETnnsjG5qVKDLQ84yYtlus,3898
-xenfra_sdk-0.1.3.dist-info/WHEEL,sha256=KSLUh82mDPEPk0Bx0ScXlWL64bc8KmzIPNcpQZFV-6E,79
-xenfra_sdk-0.1.3.dist-info/METADATA,sha256=0uvLdk5kr1Zts-HDNDqOTPNHJ0R2DkRFjG0Q64l4T0Q,3980
-xenfra_sdk-0.1.3.dist-info/RECORD,,
+xenfra_sdk-0.1.4.dist-info/WHEEL,sha256=KSLUh82mDPEPk0Bx0ScXlWL64bc8KmzIPNcpQZFV-6E,79
+xenfra_sdk-0.1.4.dist-info/METADATA,sha256=H8i-Gv6K6KGaHbIkdBAqDtS8DXrVl8rD2HFa4IX4vPM,3980
+xenfra_sdk-0.1.4.dist-info/RECORD,,