fastapi-extra 0.1.7__tar.gz → 0.1.8__tar.gz

{fastapi_extra-0.1.7 → fastapi_extra-0.1.8}/PKG-INFO

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

{fastapi_extra-0.1.7 → fastapi_extra-0.1.8}/fastapi_extra/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.1.7"
+__version__ = "0.1.8"
 
 
 from fastapi import FastAPI

{fastapi_extra-0.1.7 → fastapi_extra-0.1.8}/fastapi_extra/database/model.py

@@ -15,7 +15,7 @@ from fastapi_extra.utils import get_machine_seed
 
 class AutoPK(SQLModel):
     id: int | None = Field(
-        default_factory=lambda: None, 
+        default=None, 
         title="ID", 
         primary_key=True,
         sa_type=BigInteger,

{fastapi_extra-0.1.7 → fastapi_extra-0.1.8}/fastapi_extra/database/service.py

@@ -12,7 +12,7 @@ Model = TypeVar("Model", bound=SQLModel)
 
 class ModelService(AbstractDependency, Generic[Model], annotated=False):
     __slot__ = ("session", )
-    __model__: SQLModel
+    __model__: Model
     
     @classmethod
     def __class_getitem__(cls, item: type[SQLModel]) -> Self:
@@ -20,8 +20,11 @@ class ModelService(AbstractDependency, Generic[Model], annotated=False):
             raise TypeError(f"type[SQLModel] expected, got {item}")
         if not (table_arg := item.model_config.get("table", None)):
             raise AttributeError(f"True expected for argument {item.__name__}.model_config.table, got {table_arg}")
-        cls.__model__ = item
-        return cls
+        
+        class SubService(ModelService):
+            __model__ = item
+        
+        return SubService
 
     def __init__(self, session: DefaultSession):
         self.session = session
@@ -29,12 +32,11 @@ class ModelService(AbstractDependency, Generic[Model], annotated=False):
     async def get(self, ident: int | str, **kwargs: Any) -> Model | None:
         return await self.session.get(self.__model__, ident, **kwargs)
 
-    async def create(self, model: Model) -> Model:
-        return await self.session.add(model)
-    
     async def create_model(self, **kwargs: Any) -> Model:
         model = self.__model__.model_validate(kwargs)
-        return self.create(model)
+        self.session.add(model)
+        await self.session.flush()
+        return model
     
     async def delete(self, model: Model) -> Model:
         return await self.session.delete(model)

{fastapi_extra-0.1.7 → fastapi_extra-0.1.8}/fastapi_extra/database/session.py

@@ -14,11 +14,13 @@ from fastapi_extra.database.driver import DB, AsyncDB
 async def get_async_session(db: AsyncDB) -> AsyncGenerator[AsyncSession, None]:
     async with db.session as session:
         yield session
+        await session.commit()
 
 
 def get_session(db: DB) -> Generator[Session, None, None]:
     with db.session as session:
         yield session
+        session.commit()
 
 
 DefaultSession = Annotated[AsyncSession, Depends(get_async_session)]

{fastapi_extra-0.1.7 → fastapi_extra-0.1.8}/fastapi_extra.egg-info/PKG-INFO

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

{fastapi_extra-0.1.7 → fastapi_extra-0.1.8}/fastapi_extra.egg-info/SOURCES.txt

@@ -4,6 +4,7 @@ pyproject.toml
 fastapi_extra/__init__.py
 fastapi_extra/dependency.py
 fastapi_extra/form.py
+fastapi_extra/py.typed
 fastapi_extra/response.py
 fastapi_extra/settings.py
 fastapi_extra/types.py