fix after review

Author: Вадим Козыревский

examples/saga_sqlalchemy_storage.py

@@ -40,8 +40,8 @@
 logger = logging.getLogger(__name__)
 
 # Database Configuration
-# Using SQLite for this example, but can be swapped for PostgreSQL/MySQL
-DB_URL = os.getenv("DATABASE_URL", "mysql+asyncmy://cqrs:cqrs@localhost:3307/test_cqrs")
+# Using SQLite for this example, but can be swapped for PostgreSQL/MySQL via DATABASE_URL
+DB_URL = os.getenv("DATABASE_URL", "sqlite+aiosqlite:///./test.db")
 
 
 # ============================================================================

src/cqrs/saga/saga.py

@@ -168,8 +168,12 @@ async def __aiter__(
             run_cm = None
 
         if run_cm is not None:
-            async with run_cm as run:
-                async for step_result in self._execute(run):
+            try:
+                async with run_cm as run:
+                    async for step_result in self._execute(run):
+                        yield step_result
+            except NotImplementedError:
+                async for step_result in self._execute(None):
                     yield step_result
         else:
             async for step_result in self._execute(None):

src/cqrs/saga/storage/sqlalchemy.py

@@ -238,7 +238,7 @@ async def update_status(
                 version=SagaExecutionModel.version + 1,
             ),
         )
-        if result.rowcount == 0:
+        if result.rowcount == 0:  # pyright: ignore[reportAttributeAccessIssue]
             raise SagaConcurrencyError(
                 f"Saga {saga_id} does not exist or was modified concurrently",
             )