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

fleet/verifiers/sql_differ.py

@@ -0,0 +1,187 @@
+import sqlite3
+from typing import Any
+
+
+class SQLiteDiffer:
+    def __init__(self, before_db: str, after_db: str):
+        self.before_db = before_db
+        self.after_db = after_db
+
+    def get_table_schema(self, db_path: str, table_name: str) -> list[str]:
+        """Get column names for a table"""
+        conn = sqlite3.connect(db_path)
+        cursor = conn.cursor()
+        cursor.execute(f"PRAGMA table_info({table_name})")
+        columns = [row[1] for row in cursor.fetchall()]
+        conn.close()
+        return columns
+
+    def get_primary_key_columns(self, db_path: str, table_name: str) -> list[str]:
+        """Get all primary key columns for a table, ordered by their position"""
+        conn = sqlite3.connect(db_path)
+        cursor = conn.cursor()
+        cursor.execute(f"PRAGMA table_info({table_name})")
+
+        pk_columns = []
+        for row in cursor.fetchall():
+            # row format: (cid, name, type, notnull, dflt_value, pk)
+            if row[5] > 0:  # pk > 0 means it's part of primary key
+                pk_columns.append((row[5], row[1]))  # (pk_position, column_name)
+
+        conn.close()
+
+        # Sort by primary key position and return just the column names
+        pk_columns.sort(key=lambda x: x[0])
+        return [col[1] for col in pk_columns]
+
+    def get_all_tables(self, db_path: str) -> list[str]:
+        """Get all table names from database"""
+        conn = sqlite3.connect(db_path)
+        cursor = conn.cursor()
+        cursor.execute(
+            "SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%'"
+        )
+        tables = [row[0] for row in cursor.fetchall()]
+        conn.close()
+        return tables
+
+    def get_table_data(
+        self,
+        db_path: str,
+        table_name: str,
+        primary_key_columns: list[str] | None = None,
+    ) -> tuple[dict[Any, dict], list[str]]:
+        """Get table data indexed by primary key (single column or composite)"""
+        conn = sqlite3.connect(db_path)
+        conn.row_factory = sqlite3.Row
+        cursor = conn.cursor()
+
+        # If no primary key specified, try to detect it
+        if primary_key_columns is None:
+            primary_key_columns = self.get_primary_key_columns(db_path, table_name)
+
+            # Fallback strategies if no primary key found
+            if not primary_key_columns:
+                columns = self.get_table_schema(db_path, table_name)
+                if "id" in columns:
+                    primary_key_columns = ["id"]
+                else:
+                    primary_key_columns = ["rowid"]
+
+        cursor.execute(f"SELECT rowid, * FROM {table_name}")
+        rows = cursor.fetchall()
+
+        data = {}
+        for row in rows:
+            row_dict = dict(row)
+
+            # Create primary key value (single value or tuple for composite keys)
+            if len(primary_key_columns) == 1:
+                pk_col = primary_key_columns[0]
+                if pk_col == "rowid":
+                    pk_value = row_dict["rowid"]
+                else:
+                    pk_value = row_dict.get(pk_col)
+            else:
+                # Composite primary key - use tuple of values
+                pk_values = []
+                for pk_col in primary_key_columns:
+                    pk_values.append(row_dict.get(pk_col))
+                pk_value = tuple(pk_values)
+
+            if pk_value is not None and (
+                not isinstance(pk_value, tuple) or all(v is not None for v in pk_value)
+            ):
+                data[pk_value] = row_dict
+
+        conn.close()
+        return data, primary_key_columns
+
+    def compare_rows(self, before_row: dict, after_row: dict) -> dict[str, dict]:
+        """Compare two rows field by field"""
+        changes = {}
+
+        all_fields = set(before_row.keys()) | set(after_row.keys())
+
+        for field in all_fields:
+            before_val = before_row.get(field)
+            after_val = after_row.get(field)
+
+            if before_val != after_val:
+                changes[field] = {"before": before_val, "after": after_val}
+
+        return changes
+
+    def diff_table(
+        self, table_name: str, primary_key_columns: list[str] | None = None
+    ) -> dict:
+        """Create comprehensive diff of a table"""
+        before_data, detected_pk = self.get_table_data(
+            self.before_db, table_name, primary_key_columns
+        )
+        after_data, _ = self.get_table_data(
+            self.after_db, table_name, primary_key_columns or detected_pk
+        )
+
+        before_keys = set(before_data.keys())
+        after_keys = set(after_data.keys())
+
+        # Find different types of changes
+        added_keys = after_keys - before_keys
+        removed_keys = before_keys - after_keys
+        common_keys = before_keys & after_keys
+
+        result = {
+            "table_name": table_name,
+            "primary_key": primary_key_columns or detected_pk,
+            "added_rows": [],
+            "removed_rows": [],
+            "modified_rows": [],
+            "unchanged_count": 0,
+            "total_changes": 0,
+        }
+
+        # Added rows
+        for key in added_keys:
+            result["added_rows"].append({"row_id": key, "data": after_data[key]})
+
+        # Removed rows
+        for key in removed_keys:
+            result["removed_rows"].append({"row_id": key, "data": before_data[key]})
+
+        # Check for modifications in existing rows
+        for key in common_keys:
+            field_changes = self.compare_rows(before_data[key], after_data[key])
+
+            if field_changes:
+                result["modified_rows"].append(
+                    {
+                        "row_id": key,
+                        "changes": field_changes,
+                        "before_row": before_data[key],
+                        "after_row": after_data[key],
+                    }
+                )
+            else:
+                result["unchanged_count"] += 1
+
+        result["total_changes"] = (
+            len(result["added_rows"])
+            + len(result["removed_rows"])
+            + len(result["modified_rows"])
+        )
+
+        return result
+
+    def diff_all_tables(self) -> dict:
+        """Diff all tables in the database"""
+        tables = self.get_all_tables(self.before_db)
+        results = {}
+
+        for table in tables:
+            try:
+                results[table] = self.diff_table(table)
+            except Exception as e:
+                results[table] = {"error": str(e)}
+
+        return results

{fleet_python-0.2.0.dist-info → fleet_python-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fleet-python
-Version: 0.2.0
+Version: 0.2.2
 Summary: Python SDK for Fleet environments
 Author-email: Fleet AI <nic@fleet.so>
 License: Apache-2.0

fleet_python-0.2.2.dist-info/RECORD

@@ -0,0 +1,27 @@
+examples/dsl_example.py,sha256=LKqm2f2BHZh3Z3MWhpi4L87l5qt2TR0Ol1Mdb4mQ3bA,3297
+examples/example.py,sha256=G_UC4fD55QKPPMAKINtsJs0U6b0wg-GbpaeIGwLnhWw,933
+examples/nova_act_example.py,sha256=Z-LtSpAP4QCRg4zh-e-MQw5L8bdfZ6ev3fB5ExYToao,6443
+examples/openai_example.py,sha256=BWiDQrQCcyKgveVBXmyD7V2xypoPnumF5T5WvuOCpJ0,15258
+examples/quickstart.py,sha256=m5i5gJ0RdHJgBUlOeUSkLPHjj2GVhNvm7BP6g_A9mzA,4691
+fleet/__init__.py,sha256=D8aNOSGc-kUGrgPTVGtRhmz_Lov5yyqSX9-iBQv6Y4A,1361
+fleet/base.py,sha256=5JnzMMyT1Ey8SrYs4Ydka3bUoYRRA-wxHx1yCURAM48,2009
+fleet/client.py,sha256=5QKsCSmVhR10lcEIky5heL2NFZbmSvej19VXtfVuq14,6980
+fleet/exceptions.py,sha256=yG3QWprCw1OnF-vdFBFJWE4m3ftBLBng31Dr__VbjI4,2249
+fleet/models.py,sha256=Jf6Zmk689TPXhTSnVENK_VCw0VsujWzEWsN3T29MQ0k,3713
+fleet/env/__init__.py,sha256=_zcEf-sukzVblsTfzQbc9ixDC47bsTts2fHlKyu3OMc,81
+fleet/env/client.py,sha256=I2ja8upwdRcBnehn4GRtkRgWPCLz0cJpbSvQmSuKVrc,423
+fleet/manager/__init__.py,sha256=KXs3cz8_524XwG-MwizeEwsaX65HL8ueZieXFb7RItw,467
+fleet/manager/base.py,sha256=bm6BGd81TnTDqE_qG6S50Xhikf9DwNqEEk1uKFY7dEk,1540
+fleet/manager/client.py,sha256=s0EZcFolQyrTscbWVnvcVJpovMmWQFBR-NOBNFjgSBY,8571
+fleet/manager/models.py,sha256=oSWZ893tRhPTLyPybrk1c5AOYduWEGIKEGZjUQa1vGw,3910
+fleet/resources/base.py,sha256=eFo3wnzfPt6UoMxZSLdklvnPTDgqvjS870u8RcOEAvU,666
+fleet/resources/browser.py,sha256=H8wgabP4lxSoN2q3iFQ35qYU3nwSG-gOVQcQ8ot3DqU,1522
+fleet/resources/sqlite.py,sha256=Y30EcelvRGip0jyFEr9wxt3_EfwWS_Edt3O_qiha-pU,1498
+fleet/verifiers/__init__.py,sha256=CLw9-6bshvX7PgLcQ5KYCG9hmCLx83pi3ZS4FZPauAU,163
+fleet/verifiers/database_snapshot.py,sha256=8-DkYVEvHCwBEhKKAfMJBQmE2146z2aZvIaRtZRCk1s,26147
+fleet/verifiers/sql_differ.py,sha256=4o6Gt9472x8rcbhA58OMXdaczJiWjpYwR7LeB32D4QE,6489
+fleet_python-0.2.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+fleet_python-0.2.2.dist-info/METADATA,sha256=qwWJ_66gEeJeWSDcQt8SGkCNJfB0-6Y6dqFXyDDRu-0,3069
+fleet_python-0.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fleet_python-0.2.2.dist-info/top_level.txt,sha256=AOyXOrBXUjPcH4BumElz_D95kiWKNIpUbUPFP_9gCLk,15
+fleet_python-0.2.2.dist-info/RECORD,,

examples/browser_control_example.py

@@ -1,51 +0,0 @@
-#!/usr/bin/env python3
-"""Example demonstrating browser control with Fleet Manager Client."""
-
-import asyncio
-import fleet as flt
-
-
-async def main():
-    fleet = flt.AsyncFleet()
-
-    environments = await fleet.list_envs()
-    print("Environments:", len(environments))
-
-    instances = await fleet.instances(status="running")
-    print("Instances:", len(instances))
-
-    instance = await fleet.instance("16fdbc96")
-    print("Instance:", instance.instance_id)
-    print("Instance Environment:", instance.env_key)
-
-    environment = await fleet.environment(instance.env_key)
-    print("Environment Default Version:", environment.default_version)
-
-    response = await instance.env.reset()
-    print("Reset response:", response)
-
-    print(await instance.env.resources())
-
-    sqlite = instance.env.sqlite("current")
-    print("SQLite:", await sqlite.describe())
-
-    print("Query:", await sqlite.query("SELECT * FROM users"))
-
-    sqlite = await instance.env.state("sqlite://current").describe()
-    print("SQLite:", sqlite)
-
-    browser = await instance.env.browser("cdp").describe()
-    print("CDP URL:", browser.url)
-    print("CDP Devtools URL:", browser.devtools_url)
-
-    # Create a new instance
-    instance = await fleet.make(flt.InstanceRequest(env_key=instance.env_key))
-    print("New Instance:", instance.instance_id)
-
-    # Delete the instance
-    instance = await fleet.delete(instance.instance_id)
-    print("Instance deleted:", instance.terminated_at)
-
-
-if __name__ == "__main__":
-    asyncio.run(main())

fleet_python-0.2.0.dist-info/RECORD

@@ -1,19 +0,0 @@
-examples/browser_control_example.py,sha256=2nEzUc83bliqR9jYTenkrLTGoKSkWahjP4WpiaHIMxw,1520
-examples/quickstart.py,sha256=AnLlLQYRfrP7y2J69d1HZRlZsjJT-KeBu897MoNfqYM,4671
-fleet/__init__.py,sha256=F8esCp6DcdAc4JAhWlCd_G3pBKRBzuLuVZnElWv5GGQ,987
-fleet/base.py,sha256=lgKhPZhLotP4Iqn7Z4nQTfCvuD3bT37MoucTrSRE2zs,2006
-fleet/client.py,sha256=xMyKc49YwTAh1F5Wbk5hdaB4F6wlrl41a3EjyBdjFUk,5660
-fleet/exceptions.py,sha256=yG3QWprCw1OnF-vdFBFJWE4m3ftBLBng31Dr__VbjI4,2249
-fleet/models.py,sha256=Jf6Zmk689TPXhTSnVENK_VCw0VsujWzEWsN3T29MQ0k,3713
-fleet/env/__init__.py,sha256=ScTHui5BAFTTKKZ6TxS71D5r-JYMxxFEkcT_AcAXDsM,145
-fleet/env/base.py,sha256=bm6BGd81TnTDqE_qG6S50Xhikf9DwNqEEk1uKFY7dEk,1540
-fleet/env/client.py,sha256=yVS4cblGlARWeoZC69Uz_KR2wlmYC19d-BsQalLB9kw,8043
-fleet/env/models.py,sha256=mxRoQEzswMEjwdEN-gn2ylG9Z8IZH-Gqs6hVG0fkSVs,3883
-fleet/resources/base.py,sha256=rzFXBoFqtkM0HZpwqN9NHiwQbmeKOGOYh7G4REaHKGc,553
-fleet/resources/browser.py,sha256=mC1g7xpyvEK8vDGXNn0MXGc8TGE83PIW3KOOj9i3rIA,591
-fleet/resources/sqlite.py,sha256=hZA2qcd7SqjsACJEkKzXQwlJCtvNZ9yom2HCjKy7lOs,1484
-fleet_python-0.2.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-fleet_python-0.2.0.dist-info/METADATA,sha256=qrpy0oFm5wfUvW7VVqbrdZX566B-CB1iJSp4spH9gmg,3069
-fleet_python-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-fleet_python-0.2.0.dist-info/top_level.txt,sha256=AOyXOrBXUjPcH4BumElz_D95kiWKNIpUbUPFP_9gCLk,15
-fleet_python-0.2.0.dist-info/RECORD,,