fastapi-extra 0.3.6__tar.gz → 0.4.0__tar.gz

{fastapi_extra-0.3.6 → fastapi_extra-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastapi-extra
-Version: 0.3.6
+Version: 0.4.0
 Summary: extra package for fastapi.
 Author-email: Ziyan Yin <408856732@qq.com>
 License: BSD-3-Clause

{fastapi_extra-0.3.6 → fastapi_extra-0.4.0}/fastapi_extra/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.3.6"
+__version__ = "0.4.0"
 
 
 from fastapi import FastAPI

{fastapi_extra-0.3.6 → fastapi_extra-0.4.0}/fastapi_extra/database/service.py

@@ -2,13 +2,18 @@ __author__ = "ziyan.yin"
 __date__ = "2025-01-12"
 
 from contextvars import ContextVar
-from typing import Any, Generic, TypeVar
+from typing import Any, Generic, Sequence, TypeVar
+
+from sqlalchemy import ColumnExpressionArgument
+from sqlmodel import select
 
 from fastapi_extra.database.model import SQLModel
 from fastapi_extra.database.session import AsyncSession, DefaultSession
 from fastapi_extra.dependency import AbstractService
 
 Model = TypeVar("Model", bound=SQLModel)
+ID = int | str
+PK = ID | tuple[ID] | dict[str, ID]
 
 
 class ModelService(AbstractService, Generic[Model], abstract=True):
@@ -40,14 +45,24 @@ class ModelService(AbstractService, Generic[Model], abstract=True):
         assert _session is not None, "Session is not initialized"
         return _session
 
-    async def get(self, ident: int | str, **kwargs: Any) -> Model | None:
+    async def get(self, ident: PK, **kwargs: Any) -> Model | None:
         return await self.session.get(self.__model__, ident, **kwargs)
+    
+    async def get_list(self, *clause: ColumnExpressionArgument[bool] | bool) -> Sequence[Model]:
+        return (await self.session.exec(select(self.__model__).where(*clause))).all()
 
     async def create_model(self, **kwargs: Any) -> Model:
         model = self.__model__.model_validate(kwargs)
         self.session.add(model)
         await self.session.flush()
         return model
+    
+    async def update_model(self, model: Model, **kwargs: Any) -> Model:
+        for key, value in kwargs.items():
+            if key in model.__fields__:
+                setattr(model, key, value)
+        await self.session.flush()
+        return model
 
     async def delete(self, model: Model) -> None:
         return await self.session.delete(model)

{fastapi_extra-0.3.6 → fastapi_extra-0.4.0}/fastapi_extra/response.py

@@ -223,7 +223,7 @@ class APIResponse(JSONResponse):
         
     def init_headers(self, headers: Mapping[str, str] | None = None) -> None:
         self.raw_headers = [
-            (b"content-length", str(len(self.body)).encode("latin-1")),
+            (b"content-length", f"{len(self.body)}".encode("latin-1")),
             (b"content-type", b"application/json; charset=utf-8"),
         ]
         if headers:
@@ -234,8 +234,6 @@ class APIResponse(JSONResponse):
             self.raw_headers.extend(raw_headers)
 
 
-
-
 class APIError(Exception):
     __slots__ = ("code", "message")
 

{fastapi_extra-0.3.6 → fastapi_extra-0.4.0}/fastapi_extra.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastapi-extra
-Version: 0.3.6
+Version: 0.4.0
 Summary: extra package for fastapi.
 Author-email: Ziyan Yin <408856732@qq.com>
 License: BSD-3-Clause