fleet-python 0.2.0__py3-none-any.whl → 0.2.2__py3-none-any.whl

fleet/{env → manager}/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
-#   filename:  openapi_2.json
-#   timestamp: 2025-07-08T20:02:11+00:00
+#   filename:  openapi (2).json
+#   timestamp: 2025-07-09T20:11:31+00:00
 
 from __future__ import annotations
 
@@ -10,22 +10,27 @@ from typing import Any, Dict, List, Optional, Union
 from pydantic import BaseModel, Field
 
 
-class BrowserDescribeResponse(BaseModel):
+class CDPDescribeResponse(BaseModel):
     success: bool = Field(..., title="Success")
-    url: str = Field(..., title="Url")
-    devtools_url: str = Field(..., title="Devtools Url")
+    cdp_page_url: str = Field(..., title="Url")
+    cdp_browser_url: str = Field(..., title="Browser Url")
+    cdp_devtools_url: str = Field(..., title="Devtools Url")
 
 
-class BrowserStartRequest(BaseModel):
-    start_page: Optional[str] = Field("about:blank", title="Start Page")
+class ChromeStartRequest(BaseModel):
     resolution: Optional[str] = Field("1920x1080", title="Resolution")
 
 
-class BrowserStartResponse(BaseModel):
+class ChromeStartResponse(BaseModel):
     success: bool = Field(..., title="Success")
     message: str = Field(..., title="Message")
 
 
+class ChromeStatusResponse(BaseModel):
+    running: bool = Field(..., title="Running")
+    message: str = Field(..., title="Message")
+
+
 class CreateSnapshotsResponse(BaseModel):
     success: bool = Field(..., title="Success")
     initial_snapshot_path: Optional[str] = Field(None, title="Initial Snapshot Path")
@@ -69,8 +74,8 @@ class QueryResponse(BaseModel):
 
 
 class ResetRequest(BaseModel):
+    timestamp: Optional[int] = Field(None, title="Timestamp")
     seed: Optional[int] = Field(None, title="Seed")
-    timestamp: Optional[str] = Field(None, title="Timestamp")
 
 
 class ResetResponse(BaseModel):
@@ -84,7 +89,7 @@ class ResourceMode(Enum):
 
 
 class ResourceType(Enum):
-    sqlite = "sqlite"
+    db = "sqlite"
     cdp = "cdp"
 
 
@@ -121,7 +126,3 @@ class Resource(BaseModel):
     type: ResourceType
     mode: ResourceMode
     label: Optional[str] = Field(None, title="Label")
-
-
-class ResourcesResponse(BaseModel):
-    resources: List[Resource] = Field(..., title="Resources")

fleet/resources/base.py

@@ -1,5 +1,5 @@
 from abc import ABC
-from ..env.models import Resource as ResourceModel, ResourceType, ResourceMode
+from ..manager.models import Resource as ResourceModel, ResourceType, ResourceMode
 
 
 class Resource(ABC):
@@ -8,7 +8,7 @@ class Resource(ABC):
 
     @property
     def uri(self) -> str:
-        return f"{self.resource.type}://{self.resource.name}"
+        return f"{self.resource.type.value}://{self.resource.name}"
 
     @property
     def name(self) -> str:
@@ -21,3 +21,6 @@ class Resource(ABC):
     @property
     def mode(self) -> ResourceMode:
         return self.resource.mode
+    
+    def __repr__(self) -> str:
+        return f"Resource(uri={self.uri}, mode={self.mode.value})"

fleet/resources/browser.py

@@ -1,18 +1,44 @@
-from ..env.models import Resource as ResourceModel
-from ..env.models import BrowserDescribeResponse
+from typing import Optional
+from ..manager.models import (
+    Resource as ResourceModel,
+    CDPDescribeResponse,
+    ChromeStartRequest,
+    ChromeStartResponse,
+)
 from .base import Resource
 
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from ..env.base import AsyncWrapper
+    from ..manager.base import AsyncWrapper
 
 
 class AsyncBrowserResource(Resource):
     def __init__(self, resource: ResourceModel, client: "AsyncWrapper"):
         super().__init__(resource)
         self.client = client
+        self._describe: Optional[CDPDescribeResponse] = None
 
-    async def describe(self) -> BrowserDescribeResponse:
-        response = await self.client.request("GET", "/resource/cdp/describe")
-        return BrowserDescribeResponse(**response.json())
+    async def start(self, width: int = 1920, height: int = 1080) -> CDPDescribeResponse:
+        response = await self.client.request(
+            "POST",
+            "/resources/cdp/start",
+            json=ChromeStartRequest(resolution=f"{width},{height}").model_dump(),
+        )
+        ChromeStartResponse(**response.json())
+        return await self.describe()
+
+    async def describe(self) -> CDPDescribeResponse:
+        if self._describe is None:
+            response = await self.client.request("GET", "/resources/cdp/describe")
+            if response.status_code != 200:
+                await self.start()
+                response = await self.client.request("GET", "/resources/cdp/describe")
+            self._describe = CDPDescribeResponse(**response.json())
+        return self._describe
+
+    async def cdp_url(self) -> str:
+        return (await self.describe()).cdp_browser_url
+
+    async def devtools_url(self) -> str:
+        return (await self.describe()).cdp_devtools_url

fleet/resources/sqlite.py

@@ -1,12 +1,12 @@
 from typing import Any, List, Optional
-from ..env.models import Resource as ResourceModel
-from ..env.models import DescribeResponse, QueryRequest, QueryResponse
+from ..manager.models import Resource as ResourceModel
+from ..manager.models import DescribeResponse, QueryRequest, QueryResponse
 from .base import Resource
 
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from ..env.base import AsyncWrapper
+    from ..manager.base import AsyncWrapper
 
 
 class AsyncSQLiteResource(Resource):
@@ -17,7 +17,7 @@ class AsyncSQLiteResource(Resource):
     async def describe(self) -> DescribeResponse:
         """Describe the SQLite database schema."""
         response = await self.client.request(
-            "GET", f"/resource/sqlite/{self.resource.name}/describe"
+            "GET", f"/resources/sqlite/{self.resource.name}/describe"
         )
         return DescribeResponse(**response.json())
 
@@ -35,7 +35,7 @@ class AsyncSQLiteResource(Resource):
         request = QueryRequest(query=query, args=args, read_only=read_only)
         response = await self.client.request(
             "POST",
-            f"/resource/sqlite/{self.resource.name}/query",
+            f"/resources/sqlite/{self.resource.name}/query",
             json=request.model_dump(),
         )
         return QueryResponse(**response.json())

fleet/verifiers/__init__.py

@@ -0,0 +1,4 @@
+from .database_snapshot import QueryBuilder, DatabaseSnapshot
+from .sql_differ import SQLiteDiffer
+
+__all__ = ["QueryBuilder", "DatabaseSnapshot", "SQLiteDiffer"]