fix: linting & test tweaks

Author: cofin

.pre-commit-config.yaml

@@ -17,7 +17,7 @@ repos:
       - id: mixed-line-ending
       - id: trailing-whitespace
   - repo: https://github.com/charliermarsh/ruff-pre-commit
-    rev: "v0.14.9"
+    rev: "v0.14.10"
     hooks:
       - id: ruff
         args: ["--fix"]

docs/examples/usage/usage_query_builder_10.py

@@ -26,7 +26,8 @@ def test_example_10(tmp_path: Path) -> None:
         # start-example
         # Multiple value sets
         query = (
-            sql.insert("users")
+            sql
+            .insert("users")
             .columns("name", "email")
             .values("Alice", "[email protected]")
             .values("Bob", "[email protected]")

docs/examples/usage/usage_query_builder_13.py

@@ -30,7 +30,8 @@ def test_example_13(tmp_path: Path) -> None:
         from sqlglot import exp
 
         query = (
-            sql.update("users")
+            sql
+            .update("users")
             .set("name", "New Name")
             .set("email", "[email protected]")
             .set("updated_at", exp.CurrentTimestamp())

docs/examples/usage/usage_query_builder_17.py

@@ -22,7 +22,8 @@ def test_example_17(tmp_path: Path) -> None:
         # start-example
         # Create table
         query = (
-            sql.create_table("users")
+            sql
+            .create_table("users")
             .if_not_exists()
             .column("id", "INTEGER PRIMARY KEY")
             .column("name", "TEXT NOT NULL")

docs/examples/usage/usage_query_builder_22.py

@@ -31,7 +31,8 @@ def test_example_22(tmp_path: Path) -> None:
         cte = sql.select("user_id", "COUNT(*) as order_count").from_("orders").group_by("user_id")
 
         query = (
-            sql.select("users.name", "user_orders.order_count")
+            sql
+            .select("users.name", "user_orders.order_count")
             .with_("user_orders", cte)
             .from_("users")
             .join("user_orders", "users.id = user_orders.user_id")

docs/examples/usage/usage_query_builder_32.py

@@ -48,7 +48,8 @@ def test_example_32(tmp_path: Path) -> None:
 
         # After: Query Builder
         query = (
-            sql.select("users.id", "users.name", "COUNT(orders.id) as order_count")
+            sql
+            .select("users.id", "users.name", "COUNT(orders.id) as order_count")
             .from_("users")
             .left_join("orders", "users.id = orders.user_id")
             .where("users.status = 'active'")

docs/examples/usage/usage_query_builder_5.py

@@ -33,7 +33,8 @@ def test_example_5(tmp_path: Path) -> None:
 
         # LEFT JOIN
         query = (
-            sql.select("u.id", "u.name", "COUNT(o.id) as order_count")
+            sql
+            .select("u.id", "u.name", "COUNT(o.id) as order_count")
             .from_("users u")
             .left_join("orders o", "u.id = o.user_id")
             .group_by("u.id", "u.name")
@@ -42,7 +43,8 @@ def test_example_5(tmp_path: Path) -> None:
 
         # Multiple JOINs
         query = (
-            sql.select("u.name", "o.id", "p.name as product")
+            sql
+            .select("u.name", "o.id", "p.name as product")
             .from_("users u")
             .join("orders o", "u.id = o.user_id")
             .join("order_items oi", "o.id = oi.order_id")

docs/examples/usage/usage_query_builder_7.py

@@ -47,7 +47,8 @@ def test_example_7(tmp_path: Path) -> None:
 
         # Multiple aggregations
         query = (
-            sql.select(sql.column("created_at").as_("date"), "COUNT(*) as orders", "SUM(total) as revenue")
+            sql
+            .select(sql.column("created_at").as_("date"), "COUNT(*) as orders", "SUM(total) as revenue")
             .from_("orders")
             .group_by("date")
         )

sqlspec/adapters/oracledb/migrations.py

@@ -51,7 +51,8 @@ def _get_create_table_sql(self) -> CreateTable:
             SQL builder object for Oracle table creation.
         """
         return (
-            sql.create_table(self.version_table)
+            sql
+            .create_table(self.version_table)
             .column("version_num", "VARCHAR2(32)", primary_key=True)
             .column("version_type", "VARCHAR2(16)")
             .column("execution_sequence", "INTEGER")
@@ -73,7 +74,8 @@ def _get_current_version_sql(self) -> Select:
             SQL builder object for version query.
         """
         return (
-            sql.select('VERSION_NUM AS "version_num"')
+            sql
+            .select('VERSION_NUM AS "version_num"')
             .from_(self.version_table)
             .order_by("EXECUTION_SEQUENCE DESC")
             .limit(1)
@@ -90,7 +92,8 @@ def _get_applied_migrations_sql(self) -> Select:
             SQL builder object for migrations query.
         """
         return (
-            sql.select(
+            sql
+            .select(
                 'VERSION_NUM AS "version_num"',
                 'VERSION_TYPE AS "version_type"',
                 'EXECUTION_SEQUENCE AS "execution_sequence"',

sqlspec/adapters/psqlpy/config.py

@@ -275,4 +275,4 @@ def get_signature_namespace(self) -> "dict[str, Any]":
     def get_event_runtime_hints(self) -> "EventRuntimeHints":
         """Return LISTEN/NOTIFY defaults for Psqlpy adapters."""
 
-        return EventRuntimeHints(poll_interval=0.5, select_for_update=True, skip_locked=True, json_passthrough=True)
+        return EventRuntimeHints(poll_interval=0.5, select_for_update=True, skip_locked=True)