refactor: remove unnecessary type casting in CRUD operations for delete and update methods

This commit is contained in:
Abhimanyu Saharan
2026-02-09 02:24:49 +05:30
parent 9340a74c42
commit 9d91e87190

View File

@@ -1,7 +1,7 @@
from __future__ import annotations
from collections.abc import Iterable, Mapping
from typing import Any, TypeVar, cast
from typing import Any, TypeVar
from sqlalchemy import delete as sql_delete
from sqlalchemy import update as sql_update
@@ -154,10 +154,10 @@ async def delete_where(
*criteria: Any,
commit: bool = False,
) -> int:
stmt = sql_delete(model)
stmt: Any = sql_delete(model)
if criteria:
stmt = stmt.where(*criteria)
result = await session.exec(cast(Any, stmt))
result = await session.exec(stmt)
if commit:
await _commit_or_rollback(session)
rowcount = getattr(result, "rowcount", None)
@@ -190,10 +190,10 @@ async def update_where(
if not values:
return 0
stmt = sql_update(model).values(**values)
stmt: Any = sql_update(model).values(**values)
if criteria:
stmt = stmt.where(*criteria)
result = await session.exec(cast(Any, stmt))
result = await session.exec(stmt)
if commit:
await _commit_or_rollback(session)
rowcount = getattr(result, "rowcount", None)