feat: optional cleanup for sqlalchemy lifespans (#67)

Author: simba-git

README.md

@@ -64,7 +64,10 @@ class Order(Base):
     user: Mapped[User] = relationship(back_populates="orders")
 
 # That's it! Create your MCP app
-app = EnrichMCP("E-commerce Data", lifespan=sqlalchemy_lifespan(Base, engine))
+app = EnrichMCP(
+    "E-commerce Data",
+    lifespan=sqlalchemy_lifespan(Base, engine, cleanup_db_file=True),
+)
 include_sqlalchemy_models(app, Base)
 
 if __name__ == "__main__":

docs/getting-started.md

@@ -239,7 +239,7 @@ class Product(Base):
     price: Mapped[float] = mapped_column()
 
 
-lifespan = sqlalchemy_lifespan(Base, engine)
+lifespan = sqlalchemy_lifespan(Base, engine, cleanup_db_file=True)
 app = EnrichMCP("My ORM API", lifespan=lifespan)
 include_sqlalchemy_models(app, Base)
 app.run()

docs/sqlalchemy.md

@@ -67,3 +67,5 @@ Pagination parameters `page` and `page_size` are available on the generated
 `sqlalchemy_lifespan` automatically creates tables on startup and yields a
 `session_factory` that resolvers can use. Providing a `seed` function is
 optional and useful only for loading sample data during development or tests.
+If you are using a temporary SQLite file and want it removed on shutdown,
+pass `cleanup_db_file=True`.

examples/README.md

@@ -56,8 +56,9 @@ Both shop examples include the same core functionality but demonstrate different
 A version of the shop example built with SQLAlchemy ORM models. All entities are
 declared using SQLAlchemy and registered through `include_sqlalchemy_models`,
 which automatically creates CRUD endpoints and relationship resolvers. The
-`sqlalchemy_lifespan` helper manages the async engine and seeds the SQLite
-database on first run.
+`sqlalchemy_lifespan` helper manages the async engine, seeds the SQLite
+database on first run, and removes the file on shutdown when using
+`cleanup_db_file=True`.
 
 To run this example:
 
@@ -85,4 +86,4 @@ python app.py
 ```
 
 Stop the background server when finished. The gateway listens on port 8000 and
-provides the same schema-driven interface as the other examples.
+provides the same schema-driven interface as the other examples.

examples/sqlalchemy_shop/README.md

@@ -47,6 +47,8 @@ The app will:
 
 The lifecycle is managed using the `sqlalchemy_lifespan` helper from
 `enrichmcp.sqlalchemy`, which provides a session factory to all resources.
+Passing `cleanup_db_file=True` removes the `shop.db` file when the app shuts
+down.
 
 ## Automatic Endpoints
 

examples/sqlalchemy_shop/app.py

@@ -262,7 +262,8 @@ async def seed_database(session: AsyncSession) -> None:
 
 # Application instance will be created after defining the lifespan
 
-lifespan = sqlalchemy_lifespan(Base, engine, seed=seed_database)
+lifespan = sqlalchemy_lifespan(Base, engine, seed=seed_database, cleanup_db_file=True)
+
 
 app = EnrichMCP(
     title="Shop API (SQLAlchemy)",

src/enrichmcp/sqlalchemy/lifecycle.py

@@ -28,6 +28,7 @@ def sqlalchemy_lifespan(
     *,
     seed: Callable[[AsyncSession], Awaitable[None]] | None = None,
     session_kwargs: dict[str, Any] | None = None,
+    cleanup_db_file: bool = False,
 ) -> Lifespan:
     """Create a lifespan that sets up tables and yields a session factory."""
 
@@ -48,5 +49,14 @@ async def _lifespan(app: EnrichMCP) -> AsyncIterator[dict[str, Any]]:
             yield {"session_factory": session_factory}
         finally:
             await engine.dispose()
+            if (
+                cleanup_db_file
+                and engine.url.database
+                and engine.url.drivername.startswith("sqlite")
+            ):
+                import os
+
+                if os.path.exists(engine.url.database):
+                    os.remove(engine.url.database)
 
     return _lifespan