fastapi-async-sqlalchemy 0.7.0.dev2__py3-none-any.whl → 0.7.0.dev4__py3-none-any.whl

fastapi_async_sqlalchemy/__init__.py

@@ -1,5 +1,9 @@
-from fastapi_async_sqlalchemy.middleware import SQLAlchemyMiddleware, db
+from fastapi_async_sqlalchemy.middleware import (
+    SQLAlchemyMiddleware,
+    db,
+    create_middleware_and_session_proxy,
+)
 
-__all__ = ["db", "SQLAlchemyMiddleware"]
+__all__ = ["db", "SQLAlchemyMiddleware", "create_middleware_and_session_proxy"]
 
-__version__ = "0.7.0.dev2"
+__version__ = "0.7.0.dev4"

fastapi_async_sqlalchemy/middleware.py

@@ -21,9 +21,6 @@ def create_middleware_and_session_proxy():
     _Session: Optional[async_sessionmaker] = None
     _session: ContextVar[Optional[AsyncSession]] = ContextVar("_session", default=None)
     _multi_sessions_ctx: ContextVar[bool] = ContextVar("_multi_sessions_context", default=False)
-    _task_session_ctx: ContextVar[Optional[AsyncSession]] = ContextVar(
-        "_task_session_ctx", default=None
-    )
     _commit_on_exit_ctx: ContextVar[bool] = ContextVar("_commit_on_exit_ctx", default=False)
     # Usage of context vars inside closures is not recommended, since they are not properly
     # garbage collected, but in our use case context var is created on program startup and
@@ -92,25 +89,22 @@ def create_middleware_and_session_proxy():
                 ```
                 """
                 commit_on_exit = _commit_on_exit_ctx.get()
-                session = _task_session_ctx.get()
-                if session is None:
-                    session = _Session()
-                    _task_session_ctx.set(session)
-
-                    async def cleanup():
-                        try:
-                            if commit_on_exit:
-                                await session.commit()
-                        except Exception:
-                            await session.rollback()
-                            raise
-                        finally:
-                            await session.close()
-                            _task_session_ctx.set(None)
-
-                    task = asyncio.current_task()
-                    if task is not None:
-                        task.add_done_callback(lambda t: asyncio.create_task(cleanup()))
+                # Always create a new session for each access when multi_sessions=True
+                session = _Session()
+
+                async def cleanup():
+                    try:
+                        if commit_on_exit:
+                            await session.commit()
+                    except Exception:
+                        await session.rollback()
+                        raise
+                    finally:
+                        await session.close()
+
+                task = asyncio.current_task()
+                if task is not None:
+                    task.add_done_callback(lambda t: asyncio.create_task(cleanup()))
                 return session
             else:
                 session = _session.get()
@@ -126,7 +120,6 @@ def create_middleware_and_session_proxy():
             multi_sessions: bool = False,
         ):
             self.token = None
-            self.multi_sessions_token = None
             self.commit_on_exit_token = None
             self.session_args = session_args or {}
             self.commit_on_exit = commit_on_exit

{fastapi_async_sqlalchemy-0.7.0.dev2.dist-info → fastapi_async_sqlalchemy-0.7.0.dev4.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: fastapi-async-sqlalchemy
-Version: 0.7.0.dev2
+Version: 0.7.0.dev4
 Summary: SQLAlchemy middleware for FastAPI
 Home-page: https://github.com/h0rn3t/fastapi-async-sqlalchemy.git
 Author: Eugene Shershen
@@ -28,8 +28,19 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: starlette >=0.13.6
-Requires-Dist: SQLAlchemy >=1.4.19
+Requires-Dist: starlette>=0.13.6
+Requires-Dist: SQLAlchemy>=1.4.19
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: project-url
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 # SQLAlchemy FastAPI middleware
 
@@ -48,7 +59,7 @@ Provides SQLAlchemy middleware for FastAPI using AsyncSession and async engine.
 ### Install
 
 ```bash
-  pip install fastapi-async-sqlalchemy
+      pip install fastapi-async-sqlalchemy
 ```
 
 

fastapi_async_sqlalchemy-0.7.0.dev4.dist-info/RECORD

@@ -0,0 +1,9 @@
+fastapi_async_sqlalchemy/__init__.py,sha256=7Ac5_oyQGoV5gcLGyEx83lWaT0kgJDYAxx9nlEGiZPU,236
+fastapi_async_sqlalchemy/exceptions.py,sha256=dH3xjPE7B6kgq8L3LvxK-MsGZ_ZnW2WH2drLkK2oD-4,887
+fastapi_async_sqlalchemy/middleware.py,sha256=0jrZrrH9YsimOqsx7hFtO-BvKiNLFeDuDcqYLSS7E_g,6577
+fastapi_async_sqlalchemy/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+fastapi_async_sqlalchemy-0.7.0.dev4.dist-info/LICENSE,sha256=OhSfEMY0WqZhq41M6mLmkuFloAG7ZJYKfryF2jyxkfY,1108
+fastapi_async_sqlalchemy-0.7.0.dev4.dist-info/METADATA,sha256=CGyu0yOvcj3JUJ_TsbUa8UiwxaYOxzWXmuBesxjOS0I,7068
+fastapi_async_sqlalchemy-0.7.0.dev4.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+fastapi_async_sqlalchemy-0.7.0.dev4.dist-info/top_level.txt,sha256=po-P5Tif35GmAaUR0mgZr5uBwN91iAgszaKI_r4ZTZA,25
+fastapi_async_sqlalchemy-0.7.0.dev4.dist-info/RECORD,,

{fastapi_async_sqlalchemy-0.7.0.dev2.dist-info → fastapi_async_sqlalchemy-0.7.0.dev4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.2.0)
+Generator: setuptools (75.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

fastapi_async_sqlalchemy-0.7.0.dev2.dist-info/RECORD

@@ -1,9 +0,0 @@
-fastapi_async_sqlalchemy/__init__.py,sha256=uBiMzBBYVuNC52ekhbIasBTLsIhSvVyXY2KXD8nOWuQ,143
-fastapi_async_sqlalchemy/exceptions.py,sha256=dH3xjPE7B6kgq8L3LvxK-MsGZ_ZnW2WH2drLkK2oD-4,887
-fastapi_async_sqlalchemy/middleware.py,sha256=z3lPmnIihKyRfy34qE2z-38KuxFSXnVVcy_rg_S6IwM,6900
-fastapi_async_sqlalchemy/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-fastapi_async_sqlalchemy-0.7.0.dev2.dist-info/LICENSE,sha256=OhSfEMY0WqZhq41M6mLmkuFloAG7ZJYKfryF2jyxkfY,1108
-fastapi_async_sqlalchemy-0.7.0.dev2.dist-info/METADATA,sha256=ANbR70mznxKwilOJ4OmYPVQ0OjmntfTdXr-SFcbL2eU,6831
-fastapi_async_sqlalchemy-0.7.0.dev2.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-fastapi_async_sqlalchemy-0.7.0.dev2.dist-info/top_level.txt,sha256=po-P5Tif35GmAaUR0mgZr5uBwN91iAgszaKI_r4ZTZA,25
-fastapi_async_sqlalchemy-0.7.0.dev2.dist-info/RECORD,,