lint

Author: euri10

docs/examples/usage/usage_framework_integrations_1.py

@@ -5,6 +5,9 @@
 from sqlspec.adapters.asyncpg import AsyncpgConfig
 from sqlspec.extensions.litestar import SQLSpecPlugin
 
+__all__ = ("index", "test_app_exists" )
+
+
 # Configure database and create plugin
 spec = SQLSpec()
 db = spec.add_config(AsyncpgConfig(pool_config={"dsn": "postgresql://localhost/mydb", "min_size": 10, "max_size": 20}))

docs/examples/usage/usage_framework_integrations_10.py

@@ -1,21 +1,18 @@
 # start-example
-from fastapi import FastAPI, Depends
 from contextlib import asynccontextmanager
+
+from fastapi import FastAPI
+
 from sqlspec import SQLSpec
 from sqlspec.adapters.asyncpg import AsyncpgConfig
-from sqlspec.driver import AsyncDriverAdapterBase
+
+__all__ = ("lifespan", "test_stub" )
+
 
 # Configure database
 spec = SQLSpec()
-db = spec.add_config(
-    AsyncpgConfig(
-        pool_config={
-            "dsn": "postgresql://localhost/mydb",
-            "min_size": 10,
-            "max_size": 20,
-        }
-    )
-)
+db = spec.add_config(AsyncpgConfig(pool_config={"dsn": "postgresql://localhost/mydb", "min_size": 10, "max_size": 20}))
+
 
 # Lifespan context manager
 @asynccontextmanager
@@ -25,8 +22,10 @@ async def lifespan(app: FastAPI):
     # Shutdown
     await spec.close_all_pools()
 
+
 app = FastAPI(lifespan=lifespan)
 # end-example
 
-def test_stub():
+
+def test_stub() -> None:
     assert app is not None

docs/examples/usage/usage_framework_integrations_11.py

@@ -1,22 +1,23 @@
 # start-example
-from typing import AsyncGenerator
+from collections.abc import AsyncGenerator
+
+__all__ = ("get_db_session", "get_user", "test_stub" )
+
 
 async def get_db_session() -> AsyncGenerator[AsyncDriverAdapterBase, None]:
     async with spec.provide_session(config) as session:
         yield session
 
+
 # Use in route handlers
 @app.get("/users/{user_id}")
-async def get_user(
-    user_id: int,
-    db: AsyncDriverAdapterBase = Depends(get_db_session)
-) -> dict:
-    result = await db.execute(
-        "SELECT id, name, email FROM users WHERE id = $1",
-        user_id
-    )
+async def get_user(user_id: int, db: AsyncDriverAdapterBase = Depends(get_db_session)) -> dict:
+    result = await db.execute("SELECT id, name, email FROM users WHERE id = $1", user_id)
     return result.one()
+
+
 # end-example
 
-def test_stub():
+
+def test_stub() -> None:
     assert True

docs/examples/usage/usage_framework_integrations_12.py

@@ -1,27 +1,24 @@
 # start-example
+__all__ = ("create_user", "test_stub" )
+
+
 @app.post("/users")
-async def create_user(
-    user_data: dict,
-    db: AsyncDriverAdapterBase = Depends(get_db_session)
-) -> dict:
+async def create_user(user_data: dict, db: AsyncDriverAdapterBase = Depends(get_db_session)) -> dict:
     async with db.begin_transaction():
         result = await db.execute(
-            "INSERT INTO users (name, email) VALUES ($1, $2) RETURNING id",
-            user_data["name"],
-            user_data["email"]
+            "INSERT INTO users (name, email) VALUES ($1, $2) RETURNING id", user_data["name"], user_data["email"]
         )
 
         user_id = result.scalar()
 
         # Additional operations in same transaction
-        await db.execute(
-            "INSERT INTO audit_log (action, user_id) VALUES ($1, $2)",
-            "user_created",
-            user_id
-        )
+        await db.execute("INSERT INTO audit_log (action, user_id) VALUES ($1, $2)", "user_created", user_id)
 
         return result.one()
+
+
 # end-example
 
-def test_stub():
+
+def test_stub() -> None:
     assert True

docs/examples/usage/usage_framework_integrations_13.py

@@ -1,32 +1,38 @@
 # start-example
+__all__ = ("generate_report", "get_analytics_db", "get_main_db", "test_stub" )
+
+
 # Main database
 main_db = spec.add_config(AsyncpgConfig(pool_config={"dsn": "postgresql://localhost/main"}))
 
 # Analytics database
 analytics_db = spec.add_config(AsyncpgConfig(pool_config={"dsn": "postgresql://localhost/analytics"}))
 
+
 # Dependency functions
 async def get_main_db():
     async with spec.provide_session(main_db) as session:
         yield session
 
+
 async def get_analytics_db():
     async with spec.provide_session(analytics_db) as session:
         yield session
 
+
 # Use in handlers
 @app.get("/report")
 async def generate_report(
     main_db: AsyncDriverAdapterBase = Depends(get_main_db),
-    analytics_db: AsyncDriverAdapterBase = Depends(get_analytics_db)
+    analytics_db: AsyncDriverAdapterBase = Depends(get_analytics_db),
 ) -> dict:
     users = await main_db.execute("SELECT COUNT(*) FROM users")
     events = await analytics_db.execute("SELECT COUNT(*) FROM events")
-    return {
-        "users": users.scalar(),
-        "events": events.scalar()
-    }
+    return {"users": users.scalar(), "events": events.scalar()}
+
+
 # end-example
 
-def test_stub():
+
+def test_stub() -> None:
     assert True

docs/examples/usage/usage_framework_integrations_14.py

@@ -1,8 +1,12 @@
 # start-example
-from sanic import Sanic, Request, json
+from sanic import Sanic
+
 from sqlspec import SQLSpec
 from sqlspec.adapters.asyncpg import AsyncpgConfig
 
+__all__ = ("close_db", "test_stub" )
+
+
 app = Sanic("MyApp")
 
 # Initialize SQLSpec
@@ -13,11 +17,15 @@
 app.ctx.sqlspec = spec
 app.ctx.db_config = db
 
+
 # Cleanup on shutdown
 @app.before_server_stop
-async def close_db(app, loop):
+async def close_db(app, loop) -> None:
     await app.ctx.sqlspec.close_all_pools()
+
+
 # end-example
 
-def test_stub():
+
+def test_stub() -> None:
     assert app is not None

docs/examples/usage/usage_framework_integrations_15.py

@@ -1,13 +1,16 @@
 # start-example
+__all__ = ("get_user", "test_stub" )
+
+
 @app.get("/users/<user_id:int>")
 async def get_user(request: Request, user_id: int):
     async with request.app.ctx.sqlspec.provide_session(request.app.ctx.db_config) as db:
-        result = await db.execute(
-            "SELECT id, name, email FROM users WHERE id = $1",
-            user_id
-        )
+        result = await db.execute("SELECT id, name, email FROM users WHERE id = $1", user_id)
         return json(result.one())
+
+
 # end-example
 
-def test_stub():
+
+def test_stub() -> None:
     assert True

docs/examples/usage/usage_framework_integrations_16.py

@@ -1,21 +1,27 @@
 # start-example
+__all__ = ("add_db_session", "cleanup_db_session", "list_users", "test_stub" )
+
+
 @app.middleware("request")
-async def add_db_session(request):
-    request.ctx.db = await request.app.ctx.sqlspec.provide_session(
-        request.app.ctx.db_config
-    ).__aenter__()
+async def add_db_session(request) -> None:
+    request.ctx.db = await request.app.ctx.sqlspec.provide_session(request.app.ctx.db_config).__aenter__()
+
 
 @app.middleware("response")
-async def cleanup_db_session(request, response):
+async def cleanup_db_session(request, response) -> None:
     if hasattr(request.ctx, "db"):
         await request.ctx.db.__aexit__(None, None, None)
 
+
 # Use in handlers
 @app.get("/users")
 async def list_users(request: Request):
     result = await request.ctx.db.execute("SELECT * FROM users")
     return json(result.rows)
+
+
 # end-example
 
-def test_stub():
+
+def test_stub() -> None:
     assert True

docs/examples/usage/usage_framework_integrations_17.py

@@ -1,14 +1,19 @@
 # start-example
-from flask import Flask, g
+from flask import Flask
+
 from sqlspec import SQLSpec
 from sqlspec.adapters.sqlite import SqliteConfig
 
+__all__ = ("test_stub", )
+
+
 app = Flask(__name__)
 
 # Initialize SQLSpec
 spec = SQLSpec()
 db = spec.add_config(SqliteConfig(pool_config={"database": "app.db"}))
 # end-example
 
-def test_stub():
+
+def test_stub() -> None:
     assert app is not None

docs/examples/usage/usage_framework_integrations_18.py

@@ -1,22 +1,30 @@
 # start-example
+__all__ = ("close_db", "get_db", "get_user", "test_stub" )
+
+
 def get_db():
-    if 'db' not in g:
+    if "db" not in g:
         g.db = spec.provide_session(db).__enter__()
     return g.db
 
+
 @app.teardown_appcontext
-def close_db(error):
-    db = g.pop('db', None)
+def close_db(error) -> None:
+    db = g.pop("db", None)
     if db is not None:
         db.__exit__(None, None, None)
 
+
 # Use in routes
-@app.route('/users/<int:user_id>')
+@app.route("/users/<int:user_id>")
 def get_user(user_id):
     db = get_db()
     result = db.execute("SELECT * FROM users WHERE id = ?", user_id)
     return result.one()
+
+
 # end-example
 
-def test_stub():
+
+def test_stub() -> None:
     assert True