Merge pull request #46 from Ed-XCF/feature/dict-support-pagination

add _asdict to support fastapi-pagination

bali/db/connection.pyi

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Dict, Optional, Any, Type, TypeVar, Union, List
+from typing import Dict, Optional, Any, Type, TypeVar, Union, List, Tuple
 
 from sqlalchemy.orm.query import Query
 from sqlalchemy.orm.session import Session
@@ -17,8 +17,12 @@ class BaseModel:
     updated_time: _DateTimeField
     is_active: _BooleanField
 
+    def _asdict(self) -> Dict[str, Any]: ...
+
     def to_dict(self) -> Dict[str, Any]: ...
 
+    def dict(self) -> Dict[str, Any]: ...
+
     @classmethod
     def exists(cls: Type[_M], **attrs) -> bool: ...
 
@@ -51,6 +55,12 @@ class BaseModel:
     @classmethod
     def get_fields(cls: Type[_M]) -> List[str]: ...
 
+    @classmethod
+    def get_or_create(cls: Type[_M], defaults: Dict[str, Any], **attrs) -> Tuple[_M, bool]: ...
+
+    @classmethod
+    def update_or_create(cls: Type[_M], defaults: Dict[str, Any], **attrs) -> Tuple[_M, bool]: ...
+
 
 class DB(SQLAlchemy):
     BaseModel: BaseModel

bali/db/models.py

@@ -95,11 +95,10 @@ def delete(self):
             db.session.delete(self)
             db.session.commit() if context_auto_commit.get() else db.session.flush()
 
-        def to_dict(self):
+        def _asdict(self):
             return {c.name: getattr(self, c.name, None) for c in self.__table__.columns}
 
-        def dict(self):
-            return self.to_dict()
+        dict = to_dict = _asdict
 
         @classmethod
         def count(cls, **attrs) -> int: