fix unnecessary magic method

Author: h0rn3t

fastapi_async_sqlalchemy/exceptions.py

@@ -8,7 +8,7 @@ def __init__(self):
         a context manager e.g.:
 
         with db():
-            db.session.query(User).all()
+            await db.session.execute(foo.select()).fetchall()
         """
 
         super().__init__(msg)

fastapi_async_sqlalchemy/middleware.py

@@ -74,29 +74,19 @@ async def __anext__(self):
     async def __aenter__(self):
         if not isinstance(_Session, sessionmaker):
             raise SessionNotInitialisedError
+
         self.token = _session.set(_Session(**self.session_args))
         return self.token
 
     async def __aexit__(self, exc_type, exc_value, traceback):
-        sess = _session.get()
-        if exc_type is not None:
-            await sess.rollback()
-
-        if self.commit_on_exit:
-            await sess.commit()
-
-        await sess.close()
-        _session.reset(self.token)
-
-    def __exit__(self, exc_type, exc_value, traceback):
-        sess = _session.get()
+        session = _session.get()
         if exc_type is not None:
-            sess.rollback()
+            await session.rollback()
 
         if self.commit_on_exit:
-            sess.commit()
+            await session.commit()
 
-        sess.close()
+        await session.close()
         _session.reset(self.token)