Fix type checking errors across codebase

- Fix invalid parameter defaults: dict = None → dict | None = None
- Fix invalid type forms: str or int → str | int for union types
- Fix write_row signature: Type[SQLModel] → SQLModel (should accept instances)
- Fix logger type mismatch: add Any annotation for optional loguru dependency
- Fix dynamic import type error: add type: ignore for SQLAlchemy dialect import
- Replace deprecated session.execute() with session.exec() in bulk_upsert
- Add type: ignore comments for SQLModel Field.in_() method in tests

All type checks now pass with ty check.

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>

Author: fsecada01

sqlmodel_crud_utils/a_sync.py

@@ -1,7 +1,5 @@
 """ """
 
-from typing import Type
-
 from dateutil.parser import parse as date_parse
 from dotenv import load_dotenv
 from sqlalchemy.exc import MultipleResultsFound
@@ -46,7 +44,7 @@ async def get_result_from_query(query: SelectOfScalar, session: AsyncSession):
 async def get_one_or_create(
     session_inst: AsyncSession,
     model: type[SQLModel],
-    create_method_kwargs: dict = None,
+    create_method_kwargs: dict | None = None,
     selectin: bool = False,
     select_in_key: str | None = None,
     **kwargs,
@@ -92,12 +90,12 @@ async def _get_entry(sqlmodel, **key_args):
         return created, False
 
 
-async def write_row(data_row: Type[SQLModel], session_inst: AsyncSession):
+async def write_row(data_row: SQLModel, session_inst: AsyncSession):
     """
     Writes a new instance of an SQLModel ORM model to the database, with an
     exception catch that rolls back the session in the event of failure.
 
-    :param data_row: Type[SQLModel]
+    :param data_row: SQLModel
     :param session_inst: AsyncSession
     :return: Tuple[bool, ScalarResult]
     """
@@ -158,7 +156,7 @@ async def insert_data_rows(data_rows, session_inst: AsyncSession):
 
 
 async def get_row(
-    id_str: str or int,
+    id_str: str | int,
     session_inst: AsyncSession,
     model: type[SQLModel],
     selectin: bool = False,
@@ -441,7 +439,7 @@ async def get_rows_within_id_list(
 
 
 async def delete_row(
-    id_str: str or int,
+    id_str: str | int,
     session_inst: AsyncSession,
     model: type[SQLModel],
     pk_field: str = "id",
@@ -498,7 +496,7 @@ async def bulk_upsert_mappings(
         index_elements=[getattr(model, x) for x in pk_fields],
         set_={k: getattr(stmnt.excluded, k) for k in payload[0].keys()},
     )
-    await session_inst.execute(stmnt)
+    await session_inst.exec(stmnt)
 
     results = await session_inst.scalars(
         stmnt.returning(model), execution_options={"populate_existing": True}

sqlmodel_crud_utils/sync.py

@@ -1,7 +1,5 @@
 """ """
 
-from typing import Type
-
 from dateutil.parser import parse as date_parse
 from dotenv import load_dotenv
 from sqlalchemy.exc import MultipleResultsFound
@@ -45,7 +43,7 @@ def get_result_from_query(query: SelectOfScalar, session: Session):
 def get_one_or_create(
     session_inst: Session,
     model: type[SQLModel],
-    create_method_kwargs: dict = None,
+    create_method_kwargs: dict | None = None,
     selectin: bool = False,
     select_in_key: str | None = None,
     **kwargs,
@@ -91,12 +89,12 @@ def _get_entry(sqlmodel, **key_args):
         return created, False
 
 
-def write_row(data_row: Type[SQLModel], session_inst: Session):
+def write_row(data_row: SQLModel, session_inst: Session):
     """
     Writes a new instance of an SQLModel ORM model to the database, with an
     exception catch that rolls back the session in the event of failure.
 
-    :param data_row: Type[SQLModel]
+    :param data_row: SQLModel
     :param session_inst: Session
     :return: Tuple[bool, ScalarResult]
     """
@@ -155,7 +153,7 @@ def insert_data_rows(data_rows, session_inst: Session):
 
 
 def get_row(
-    id_str: str or int,
+    id_str: str | int,
     session_inst: Session,
     model: type[SQLModel],
     selectin: bool = False,
@@ -442,7 +440,7 @@ def get_rows_within_id_list(
 
 
 def delete_row(
-    id_str: str or int,
+    id_str: str | int,
     session_inst: Session,
     model: type[SQLModel],
     pk_field: str = "id",

sqlmodel_crud_utils/utils.py

@@ -1,13 +1,14 @@
 import importlib
 import logging
 import os
+from typing import Any
 
 from dateutil.parser import parse as date_parse
 
 try:
     from loguru import logger
 except ImportError:
-    logger = logging.getLogger("sqlmodel_crud_utils")
+    logger: Any = logging.getLogger("sqlmodel_crud_utils")  # type: ignore[assignment]
 
 
 def get_val(val: str):
@@ -31,7 +32,7 @@ def get_sql_dialect_import(dialect: str):
 
     :return: func
     """
-    return importlib.import_module(f"sqlalchemy.dialects" f".{dialect}").insert
+    return importlib.import_module(f"sqlalchemy.dialects" f".{dialect}").insert  # type: ignore[attr-defined]
 
 
 def is_date(val: str, fuzzy: bool = False):

tests/test_async_utils.py

@@ -275,7 +275,7 @@ async def test_async_insert_data_rows_success(async_session: AsyncSession):
 
     # Assert: Verify they exist in the DB
     ids = [row.id for row in result]
-    statement = select(MockModel).where(MockModel.id.in_(ids))
+    statement = select(MockModel).where(MockModel.id.in_(ids))  # type: ignore[union-attr]
     # FIX: Remove redundant .scalars()
     fetched_rows_result = await async_session.exec(statement)
     fetched_rows = fetched_rows_result.all()

tests/test_sync_utils.py

@@ -220,7 +220,7 @@ def test_sync_insert_data_rows_success(sync_session: Session):
 
     # Assert: Verify they exist in the DB
     ids = [row.id for row in result]
-    statement = select(MockModel).where(MockModel.id.in_(ids))
+    statement = select(MockModel).where(MockModel.id.in_(ids))  # type: ignore[union-attr]
     fetched_rows = sync_session.exec(statement).all()
     assert len(fetched_rows) == len(data_rows)
     names_after = {row.name for row in fetched_rows}