fleet-python 0.2.1__py3-none-any.whl → 0.2.3__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.1.dist-info → fleet_python-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fleet-python
-Version: 0.2.1
+Version: 0.2.3
 Summary: Python SDK for Fleet environments
 Author-email: Fleet AI <nic@fleet.so>
 License: Apache-2.0
@@ -31,6 +31,8 @@ Requires-Dist: black>=22.0.0; extra == "dev"
 Requires-Dist: isort>=5.0.0; extra == "dev"
 Requires-Dist: mypy>=1.0.0; extra == "dev"
 Requires-Dist: ruff>=0.1.0; extra == "dev"
+Provides-Extra: playwright
+Requires-Dist: playwright>=1.40.0; extra == "playwright"
 Dynamic: license-file
 
 # Fleet SDK

fleet_python-0.2.3.dist-info/RECORD

@@ -0,0 +1,31 @@
+examples/dsl_example.py,sha256=W-g0RR7qOTGUSHYiFhcwL3MsrHB-C6t3UHQg6XTxxss,5259
+examples/example.py,sha256=G_UC4fD55QKPPMAKINtsJs0U6b0wg-GbpaeIGwLnhWw,933
+examples/json_tasks_example.py,sha256=fz1hGaJ_VukH2gwJFvhQmyWRO8ocN9ptFnplTJSOrbA,2348
+examples/nova_act_example.py,sha256=EwTivGRpJ4ZGicJw0eK46lvjgjN0_x3FJavafzdVkfc,777
+examples/openai_example.py,sha256=ebaGIs1OFhohUdtQ0fHcMUqvmCd0vJa45_FUo58eTh4,8486
+examples/openai_simple_example.py,sha256=16u_g9vD8mHNTZhta6Qdq_qAQs-wU6FaPVkhGUPnSIs,1788
+examples/quickstart.py,sha256=lVRzbnWIweU9ioe7uk6R2Rm7oSpt4mt8Jq_VUUp1zKg,4696
+fleet/__init__.py,sha256=AjkbRVYzfIUYEw1PvCceGhHBCPOu_c6Qx5KweoNmAQM,1747
+fleet/base.py,sha256=5JnzMMyT1Ey8SrYs4Ydka3bUoYRRA-wxHx1yCURAM48,2009
+fleet/client.py,sha256=RFOEhxJa_de3N6JP48p8xMYhsGVRv5V4U4OWINKpQ8Y,7392
+fleet/exceptions.py,sha256=yG3QWprCw1OnF-vdFBFJWE4m3ftBLBng31Dr__VbjI4,2249
+fleet/models.py,sha256=Jf6Zmk689TPXhTSnVENK_VCw0VsujWzEWsN3T29MQ0k,3713
+fleet/playwright.py,sha256=5MLFPE5P_-MpzAQ3EJ6GsLthuJiWwYkNuvhPE_rwe_E,8914
+fleet/env/__init__.py,sha256=_zcEf-sukzVblsTfzQbc9ixDC47bsTts2fHlKyu3OMc,81
+fleet/env/client.py,sha256=I2ja8upwdRcBnehn4GRtkRgWPCLz0cJpbSvQmSuKVrc,423
+fleet/instance/__init__.py,sha256=pKuX6zPpTKmJD0c-Qg_hRdnQuiWNohQdYhlHElxilQE,562
+fleet/instance/base.py,sha256=bm6BGd81TnTDqE_qG6S50Xhikf9DwNqEEk1uKFY7dEk,1540
+fleet/instance/client.py,sha256=Tybjm0jbxM_lnTvQ-Gtm2e8u2qvk9vjDfBwns-DpQhw,9872
+fleet/instance/models.py,sha256=ZTiue0YOuhuwX8jYfJAoCzGfqjLqqXRLqK1LVFhq6rQ,4183
+fleet/resources/base.py,sha256=203gD54NP1IvjuSqFo-f7FvrkhtjChggtzrxJK7xf2E,667
+fleet/resources/browser.py,sha256=x11y4aKHogIEv83FByHtExerjV-cDWI3U62349Guq_Q,1368
+fleet/resources/sqlite.py,sha256=sRiII_qJ8X6-FSemlBsXThz4ZPjkNy9wDT8g5UAz2XM,1501
+fleet/verifiers/__init__.py,sha256=sNlURjry8FYFa-0qFy6boRCcNjXAtMPFaAfpaRqpOPk,394
+fleet/verifiers/code.py,sha256=YB96SPncu9emO-udQh7zYf_UR1aPz7nBjPtKf5O6fD8,4356
+fleet/verifiers/db.py,sha256=tssmvJjDHuBIy8qlL_P5-UdmEFUw2DZcqLsWZ8ot3Xw,27766
+fleet/verifiers/sql_differ.py,sha256=dmiGCFXVMEMbAX519OjhVqgA8ZvhnvdmC1BVpL7QCF0,6490
+fleet_python-0.2.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+fleet_python-0.2.3.dist-info/METADATA,sha256=jdYK3MqjugV1mQ4Bl0Uw-ZH1HQL6Vzr0vTFcnP2xATw,3153
+fleet_python-0.2.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fleet_python-0.2.3.dist-info/top_level.txt,sha256=AOyXOrBXUjPcH4BumElz_D95kiWKNIpUbUPFP_9gCLk,15
+fleet_python-0.2.3.dist-info/RECORD,,

fleet_python-0.2.1.dist-info/RECORD

@@ -1,21 +0,0 @@
-examples/example.py,sha256=nF-XyXXYn3eNQQe00DGpagI-bQkh3WRhNPIpQLHfUrc,1502
-examples/nova_act_example.py,sha256=Z-LtSpAP4QCRg4zh-e-MQw5L8bdfZ6ev3fB5ExYToao,6443
-examples/openai_example.py,sha256=NJZo3nA6_56pYVINcfSOg9OH5JX13SNqzvIG-48SLls,10873
-examples/quickstart.py,sha256=AnLlLQYRfrP7y2J69d1HZRlZsjJT-KeBu897MoNfqYM,4671
-fleet/__init__.py,sha256=ckXsUDKRzDs4YOFvEhId_wjYBVbFmmj4wuSweE-Rdtk,1314
-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=h46DwQKJZHqPivXTlwE6JIjwJocwXJQsxe4B2M6xUvo,455
-fleet/env/base.py,sha256=bm6BGd81TnTDqE_qG6S50Xhikf9DwNqEEk1uKFY7dEk,1540
-fleet/env/client.py,sha256=gOfdjaDNilJ2g0snZnR3nDIepD7d8m6QHBQMZNMhsWA,8600
-fleet/env/models.py,sha256=oSWZ893tRhPTLyPybrk1c5AOYduWEGIKEGZjUQa1vGw,3910
-fleet/resources/base.py,sha256=rzFXBoFqtkM0HZpwqN9NHiwQbmeKOGOYh7G4REaHKGc,553
-fleet/resources/browser.py,sha256=3BN8AQXzdd1wFoVfDps_Gyx7QvN2UC4cbjU2ZlnFbIA,1002
-fleet/resources/sqlite.py,sha256=I3TnCxU5_WNarNPpZbfppHk2OeDbgp82JlBYGW6DtRs,1486
-fleet_python-0.2.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-fleet_python-0.2.1.dist-info/METADATA,sha256=W1NcdBg98RWQNnedwyUQ_CdUc0BCDxtO1xBcAtu4Sik,3069
-fleet_python-0.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-fleet_python-0.2.1.dist-info/top_level.txt,sha256=AOyXOrBXUjPcH4BumElz_D95kiWKNIpUbUPFP_9gCLk,15
-fleet_python-0.2.1.dist-info/RECORD,,