kiba-core 0.5.3.dev90__tar.gz → 0.5.3.dev92__tar.gz

{kiba_core-0.5.3.dev90 → kiba_core-0.5.3.dev92}/Dockerfile

@@ -1,4 +1,4 @@
-FROM python:3.14.8-slim
+FROM python:3.14.9-slim
 
 RUN apt-get update && apt-get install --yes --no-install-recommends make git
 

{kiba_core-0.5.3.dev90 → kiba_core-0.5.3.dev92}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kiba-core
-Version: 0.5.3.dev90
+Version: 0.5.3.dev92
 Summary: Kiba Labs' python utilities
 Project-URL: repository, https://github.com/kibalabs/core-py
 Requires-Python: <4,>=3.12
@@ -9,36 +9,36 @@ Requires-Dist: aiofiles>=25.1.0
 Requires-Dist: pydantic>=2.14.1
 Requires-Dist: orjson>=3.13.0
 Provides-Extra: api
-Requires-Dist: fastapi>=0.146.0; extra == "api"
+Requires-Dist: fastapi>=0.162.0; extra == "api"
 Requires-Dist: uvicorn[standard]>=0.54.0; extra == "api"
 Provides-Extra: core-api
-Requires-Dist: starlette>=1.9.0; extra == "core-api"
+Requires-Dist: starlette>=1.10.0; extra == "core-api"
 Requires-Dist: uvicorn[standard]>=0.54.0; extra == "core-api"
 Requires-Dist: python-multipart>=0.0.32; extra == "core-api"
 Provides-Extra: storage
-Requires-Dist: aiobotocore>=3.9.2; extra == "storage"
+Requires-Dist: aiobotocore>=3.10.0; extra == "storage"
 Provides-Extra: queues
-Requires-Dist: aiobotocore>=3.9.2; extra == "queues"
+Requires-Dist: aiobotocore>=3.10.0; extra == "queues"
 Provides-Extra: queue-sqs
-Requires-Dist: aiobotocore>=3.9.2; extra == "queue-sqs"
+Requires-Dist: aiobotocore>=3.10.0; extra == "queue-sqs"
 Provides-Extra: queue-aqs
 Requires-Dist: azure-storage-queue[aio]>=12.18.0; extra == "queue-aqs"
 Provides-Extra: queue-cosmos
 Requires-Dist: azure-cosmos>=4.17.1; extra == "queue-cosmos"
 Provides-Extra: database-psql
-Requires-Dist: sqlalchemy[asyncio]>=2.1.5; extra == "database-psql"
+Requires-Dist: sqlalchemy[asyncio]>=2.1.6; extra == "database-psql"
 Requires-Dist: asyncpg>=0.32.0; extra == "database-psql"
 Provides-Extra: database-sqlite
-Requires-Dist: sqlalchemy[asyncio]>=2.1.5; extra == "database-sqlite"
+Requires-Dist: sqlalchemy[asyncio]>=2.1.6; extra == "database-sqlite"
 Requires-Dist: aiosqlite>=0.22.1; extra == "database-sqlite"
 Provides-Extra: requester
-Requires-Dist: httpx2>=2.14.0; extra == "requester"
+Requires-Dist: httpx2>=2.16.0; extra == "requester"
 Provides-Extra: web3
 Requires-Dist: web3<8.0.0,>=7.16.0; extra == "web3"
 Provides-Extra: cron
 Requires-Dist: apscheduler<4,>=3.11.3; extra == "cron"
 Provides-Extra: types
-Requires-Dist: types-aiobotocore[essential]>=3.9.2; extra == "types"
+Requires-Dist: types-aiobotocore[essential]>=3.10.0; extra == "types"
 Requires-Dist: types-aiofiles>=25.1.0.20260518; extra == "types"
 
 # Core-Py

{kiba_core-0.5.3.dev90 → kiba_core-0.5.3.dev92}/core/store/entity_repository.py

@@ -9,6 +9,7 @@ from sqlalchemy.dialects import sqlite as sqlalchemy_sqlite
 from sqlalchemy.engine import Result
 from sqlalchemy.engine import RowMapping
 from sqlalchemy.sql import Select
+from sqlalchemy.sql.base import ReadOnlyColumnCollection
 
 from core.exceptions import KibaException
 from core.exceptions import NotFoundException
@@ -25,6 +26,7 @@ from core.store.retriever import uuid_from_value
 from core.util import date_util
 
 EntityType = typing.TypeVar('EntityType', bound=BaseModel)
+UpsertUpdateWhere = typing.Callable[[ReadOnlyColumnCollection[str, typing.Any]], sqlalchemy.ColumnElement[bool]]  # type: ignore[explicit-any]
 
 
 class EntityRepository(typing.Generic[EntityType]):  # noqa: UP046
@@ -135,7 +137,8 @@ class EntityRepository(typing.Generic[EntityType]):  # noqa: UP046
         result = await database.execute(query=self.table.update().where(self.idColumn == idValue).values(updateValues).returning(*self.list_select_columns()), connection=connection)
         return self.force_from_result(result=result)
 
-    async def upsert(self, database: Database, constraintColumnNames: list[str], connection: DatabaseConnection | None = None, **kwargs) -> EntityType:  # type: ignore[no-untyped-def]  # noqa: ANN003
+    # NOTE(krishan711): updateWhere gets the conflicting row's new values (`excluded`) and only updates when it is true; when it is false nothing is written and the stored row is returned
+    async def upsert(self, database: Database, constraintColumnNames: list[str], connection: DatabaseConnection | None = None, updateWhere: UpsertUpdateWhere | None = None, **kwargs) -> EntityType:  # type: ignore[no-untyped-def]  # noqa: ANN003
         updateValues = self._create_values(kwargs=kwargs, shouldAddUpdatedDate=True, shouldAddCreatedDate=False)
         insertValues = self._create_values(kwargs=kwargs, shouldAddUpdatedDate=True, shouldAddCreatedDate=True)
         self._add_generated_id(values=insertValues)
@@ -144,9 +147,16 @@ class EntityRepository(typing.Generic[EntityType]):  # noqa: UP046
         doUpdateStatement = insertStatement.on_conflict_do_update(
             index_elements=constraintColumns,
             set_=updateValues,
+            where=updateWhere(insertStatement.excluded) if updateWhere is not None else None,
         )
         result = await database.execute(query=doUpdateStatement.returning(*self.list_select_columns()), connection=connection)
-        return self.force_from_result(result=result)
+        row = result.mappings().first()
+        if row is None and updateWhere is not None:
+            storedQuery = self.build_select().where(*[column == insertValues[column] for column in constraintColumns])
+            row = (await database.execute(query=storedQuery, connection=connection)).mappings().first()
+        if row is None:
+            raise NotFoundException
+        return self.from_row(row=row)
 
     async def upsert_many(self, database: Database, constraintColumnNames: list[str], rowDicts: list[dict[str, typing.Any]], connection: DatabaseConnection | None = None) -> list[EntityType]:  # type: ignore[explicit-any]
         if not rowDicts:

{kiba_core-0.5.3.dev90 → kiba_core-0.5.3.dev92}/kiba_core.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kiba-core
-Version: 0.5.3.dev90
+Version: 0.5.3.dev92
 Summary: Kiba Labs' python utilities
 Project-URL: repository, https://github.com/kibalabs/core-py
 Requires-Python: <4,>=3.12
@@ -9,36 +9,36 @@ Requires-Dist: aiofiles>=25.1.0
 Requires-Dist: pydantic>=2.14.1
 Requires-Dist: orjson>=3.13.0
 Provides-Extra: api
-Requires-Dist: fastapi>=0.146.0; extra == "api"
+Requires-Dist: fastapi>=0.162.0; extra == "api"
 Requires-Dist: uvicorn[standard]>=0.54.0; extra == "api"
 Provides-Extra: core-api
-Requires-Dist: starlette>=1.9.0; extra == "core-api"
+Requires-Dist: starlette>=1.10.0; extra == "core-api"
 Requires-Dist: uvicorn[standard]>=0.54.0; extra == "core-api"
 Requires-Dist: python-multipart>=0.0.32; extra == "core-api"
 Provides-Extra: storage
-Requires-Dist: aiobotocore>=3.9.2; extra == "storage"
+Requires-Dist: aiobotocore>=3.10.0; extra == "storage"
 Provides-Extra: queues
-Requires-Dist: aiobotocore>=3.9.2; extra == "queues"
+Requires-Dist: aiobotocore>=3.10.0; extra == "queues"
 Provides-Extra: queue-sqs
-Requires-Dist: aiobotocore>=3.9.2; extra == "queue-sqs"
+Requires-Dist: aiobotocore>=3.10.0; extra == "queue-sqs"
 Provides-Extra: queue-aqs
 Requires-Dist: azure-storage-queue[aio]>=12.18.0; extra == "queue-aqs"
 Provides-Extra: queue-cosmos
 Requires-Dist: azure-cosmos>=4.17.1; extra == "queue-cosmos"
 Provides-Extra: database-psql
-Requires-Dist: sqlalchemy[asyncio]>=2.1.5; extra == "database-psql"
+Requires-Dist: sqlalchemy[asyncio]>=2.1.6; extra == "database-psql"
 Requires-Dist: asyncpg>=0.32.0; extra == "database-psql"
 Provides-Extra: database-sqlite
-Requires-Dist: sqlalchemy[asyncio]>=2.1.5; extra == "database-sqlite"
+Requires-Dist: sqlalchemy[asyncio]>=2.1.6; extra == "database-sqlite"
 Requires-Dist: aiosqlite>=0.22.1; extra == "database-sqlite"
 Provides-Extra: requester
-Requires-Dist: httpx2>=2.14.0; extra == "requester"
+Requires-Dist: httpx2>=2.16.0; extra == "requester"
 Provides-Extra: web3
 Requires-Dist: web3<8.0.0,>=7.16.0; extra == "web3"
 Provides-Extra: cron
 Requires-Dist: apscheduler<4,>=3.11.3; extra == "cron"
 Provides-Extra: types
-Requires-Dist: types-aiobotocore[essential]>=3.9.2; extra == "types"
+Requires-Dist: types-aiobotocore[essential]>=3.10.0; extra == "types"
 Requires-Dist: types-aiofiles>=25.1.0.20260518; extra == "types"
 
 # Core-Py

{kiba_core-0.5.3.dev90 → kiba_core-0.5.3.dev92}/kiba_core.egg-info/requires.txt

@@ -3,11 +3,11 @@ pydantic>=2.14.1
 orjson>=3.13.0
 
 [api]
-fastapi>=0.146.0
+fastapi>=0.162.0
 uvicorn[standard]>=0.54.0
 
 [core-api]
-starlette>=1.9.0
+starlette>=1.10.0
 uvicorn[standard]>=0.54.0
 python-multipart>=0.0.32
 
@@ -15,11 +15,11 @@ python-multipart>=0.0.32
 apscheduler<4,>=3.11.3
 
 [database-psql]
-sqlalchemy[asyncio]>=2.1.5
+sqlalchemy[asyncio]>=2.1.6
 asyncpg>=0.32.0
 
 [database-sqlite]
-sqlalchemy[asyncio]>=2.1.5
+sqlalchemy[asyncio]>=2.1.6
 aiosqlite>=0.22.1
 
 [queue-aqs]
@@ -29,19 +29,19 @@ azure-storage-queue[aio]>=12.18.0
 azure-cosmos>=4.17.1
 
 [queue-sqs]
-aiobotocore>=3.9.2
+aiobotocore>=3.10.0
 
 [queues]
-aiobotocore>=3.9.2
+aiobotocore>=3.10.0
 
 [requester]
-httpx2>=2.14.0
+httpx2>=2.16.0
 
 [storage]
-aiobotocore>=3.9.2
+aiobotocore>=3.10.0
 
 [types]
-types-aiobotocore[essential]>=3.9.2
+types-aiobotocore[essential]>=3.10.0
 types-aiofiles>=25.1.0.20260518
 
 [web3]

{kiba_core-0.5.3.dev90 → kiba_core-0.5.3.dev92}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kiba-core"
-version = "0.5.3.dev90"
+version = "0.5.3.dev92"
 description = "Kiba Labs' python utilities"
 readme = "README.md"
 requires-python = ">=3.12, <4"
@@ -12,23 +12,23 @@ dependencies = [
 
 [project.optional-dependencies]
 api = [
-    "fastapi>=0.146.0",
+    "fastapi>=0.162.0",
     "uvicorn[standard]>=0.54.0",
 ]
 core-api = [
-    "starlette>=1.9.0",
+    "starlette>=1.10.0",
     "uvicorn[standard]>=0.54.0",
     "python-multipart>=0.0.32",
 ]
 storage = [
-    "aiobotocore>=3.9.2",
+    "aiobotocore>=3.10.0",
 ]
 # TODO(krishan711): remove this in next major update
 queues = [
-    "aiobotocore>=3.9.2",
+    "aiobotocore>=3.10.0",
 ]
 queue-sqs = [
-    "aiobotocore>=3.9.2",
+    "aiobotocore>=3.10.0",
 ]
 queue-aqs = [
     "azure-storage-queue[aio]>=12.18.0",
@@ -37,15 +37,15 @@ queue-cosmos = [
     "azure-cosmos>=4.17.1",
 ]
 database-psql = [
-    "sqlalchemy[asyncio]>=2.1.5",
+    "sqlalchemy[asyncio]>=2.1.6",
     "asyncpg>=0.32.0",
 ]
 database-sqlite = [
-    "sqlalchemy[asyncio]>=2.1.5",
+    "sqlalchemy[asyncio]>=2.1.6",
     "aiosqlite>=0.22.1",
 ]
 requester = [
-    "httpx2>=2.14.0",
+    "httpx2>=2.16.0",
 ]
 web3 = [
     "web3>=7.16.0,<8.0.0",
@@ -54,13 +54,13 @@ cron = [
     "apscheduler>=3.11.3,<4",
 ]
 types = [
-    "types-aiobotocore[essential]>=3.9.2",
+    "types-aiobotocore[essential]>=3.10.0",
     "types-aiofiles>=25.1.0.20260518",
 ]
 
 [dependency-groups]
 dev = [
-    "kiba-build==0.1.11.dev20",
+    "kiba-build==0.1.11.dev22",
 ]
 
 [project.urls]

{kiba_core-0.5.3.dev90 → kiba_core-0.5.3.dev92}/tests/test_entity_repository.py

@@ -172,6 +172,49 @@ async def test_upsert_inserts_then_updates(database: Database):
     assert upsertedItem.count == 2
 
 
+async def test_upsert_with_false_update_where_keeps_and_returns_the_stored_row(database: Database):
+    item = await _create_item(database=database, name='a', count=1)
+    async with database.create_transaction() as connection:
+        upsertedItem = await ItemsRepository.upsert(database=database, connection=connection, constraintColumnNames=['name'], updateWhere=lambda excluded: excluded.count > ItemsTable.c.count, name='a', count=0, amount=5)
+        storedItem = await ItemsRepository.get(database=database, connection=connection, idValue=item.itemId)
+    assert upsertedItem == item
+    assert storedItem == item
+
+
+async def test_upsert_with_false_update_where_finds_the_stored_row_by_converted_constraint_values(database: Database):
+    repository = UpperNameEntityRepository(table=ItemsTable, modelClass=Item)
+    async with database.create_transaction() as connection:
+        item = await repository.create(database=database, connection=connection, name='a', count=1, amount=1)
+        upsertedItem = await repository.upsert(database=database, connection=connection, constraintColumnNames=['name'], updateWhere=lambda _excluded: sqlalchemy.false(), name='a', count=2, amount=1)
+    assert upsertedItem == item
+
+
+async def test_upsert_with_true_update_where_updates_the_row(database: Database):
+    item = await _create_item(database=database, name='a', count=1)
+    async with database.create_transaction() as connection:
+        upsertedItem = await ItemsRepository.upsert(database=database, connection=connection, constraintColumnNames=['name'], updateWhere=lambda excluded: excluded.count > ItemsTable.c.count, name='a', count=2, amount=5)
+    assert (upsertedItem.itemId, upsertedItem.createdDate, upsertedItem.count, upsertedItem.amount) == (item.itemId, item.createdDate, 2, 5)
+
+
+async def test_upsert_with_update_where_still_inserts_new_rows(database: Database):
+    async with database.create_transaction() as connection:
+        upsertedItem = await ItemsRepository.upsert(database=database, connection=connection, constraintColumnNames=['name'], updateWhere=lambda _excluded: sqlalchemy.false(), name='a', count=1, amount=1)
+        storedItems = await ItemsRepository.list_many(database=database, connection=connection, fieldFilters=[])
+    assert storedItems == [upsertedItem]
+
+
+async def test_upsert_only_when_values_changed(database: Database):
+    def changed(excluded: typing.Any) -> sqlalchemy.ColumnElement[bool]:  # type: ignore[explicit-any]
+        return sqlalchemy.or_(ItemsTable.c.count.is_distinct_from(excluded.count), ItemsTable.c.expiryDate.is_distinct_from(excluded.expiryDate))
+
+    async with database.create_transaction() as connection:
+        item = await ItemsRepository.upsert(database=database, connection=connection, constraintColumnNames=['name'], updateWhere=changed, name='a', count=1, amount=1, expiryDate=None)
+        unchangedItem = await ItemsRepository.upsert(database=database, connection=connection, constraintColumnNames=['name'], updateWhere=changed, name='a', count=1, amount=1, expiryDate=None)
+        changedItem = await ItemsRepository.upsert(database=database, connection=connection, constraintColumnNames=['name'], updateWhere=changed, name='a', count=1, amount=1, expiryDate=datetime.datetime(2030, 1, 1, tzinfo=datetime.UTC))
+    assert unchangedItem == item
+    assert (changedItem.itemId, changedItem.expiryDate) == (item.itemId, datetime.datetime(2030, 1, 1, tzinfo=datetime.UTC))
+
+
 async def test_upsert_many_inserts_and_updates(database: Database):
     item = await _create_item(database=database, name='a')
     async with database.create_transaction() as connection: