Move SortDirection in common.py

Author: febus982

sqlalchemy_bind_manager/_repository/__init__.py

@@ -1,4 +1,8 @@
 from .async_ import SQLAlchemyAsyncRepository
-from .base_repository import SortDirection
-from .common import CursorPaginatedResult, CursorReference, PaginatedResult
+from .common import (
+    CursorPaginatedResult,
+    CursorReference,
+    PaginatedResult,
+    SortDirection,
+)
 from .sync import SQLAlchemyRepository

sqlalchemy_bind_manager/_repository/async_.py

@@ -19,14 +19,14 @@
 from ..exceptions import InvalidConfig, ModelNotFound
 from .base_repository import (
     BaseRepository,
-    SortDirection,
 )
 from .common import (
     MODEL,
     PRIMARY_KEY,
     CursorPaginatedResult,
     CursorReference,
     PaginatedResult,
+    SortDirection,
 )
 from .result_presenters import CursorPaginatedResultPresenter, PaginatedResultPresenter
 

sqlalchemy_bind_manager/_repository/base_repository.py

@@ -1,6 +1,4 @@
 from abc import ABC
-from enum import Enum
-from functools import partial
 from typing import (
     Any,
     Generic,
@@ -11,7 +9,7 @@
     Union,
 )
 
-from sqlalchemy import asc, desc, func, inspect, select
+from sqlalchemy import asc, func, inspect, select
 from sqlalchemy.orm import Mapper, aliased, class_mapper, lazyload
 from sqlalchemy.orm.exc import UnmappedClassError
 from sqlalchemy.sql import Select
@@ -21,14 +19,10 @@
 from .common import (
     MODEL,
     CursorReference,
+    SortDirection,
 )
 
 
-class SortDirection(Enum):
-    ASC = partial(asc)
-    DESC = partial(desc)
-
-
 class BaseRepository(Generic[MODEL], ABC):
     _max_query_limit: int = 50
     _model: Type[MODEL]

sqlalchemy_bind_manager/_repository/common.py

@@ -1,7 +1,10 @@
+from enum import Enum
+from functools import partial
 from typing import Generic, List, TypeVar, Union
 
 from pydantic import BaseModel, StrictInt, StrictStr
 from pydantic.generics import GenericModel
+from sqlalchemy import asc, desc
 
 MODEL = TypeVar("MODEL")
 PRIMARY_KEY = Union[str, int, tuple, dict]
@@ -38,3 +41,8 @@ class CursorPageInfo(BaseModel):
 class CursorPaginatedResult(GenericModel, Generic[MODEL]):
     items: List[MODEL]
     page_info: CursorPageInfo
+
+
+class SortDirection(Enum):
+    ASC = partial(asc)
+    DESC = partial(desc)

sqlalchemy_bind_manager/_repository/sync.py

@@ -19,14 +19,14 @@
 from ..exceptions import InvalidConfig, ModelNotFound
 from .base_repository import (
     BaseRepository,
-    SortDirection,
 )
 from .common import (
     MODEL,
     PRIMARY_KEY,
     CursorPaginatedResult,
     CursorReference,
     PaginatedResult,
+    SortDirection,
 )
 from .result_presenters import CursorPaginatedResultPresenter, PaginatedResultPresenter
 

sqlalchemy_bind_manager/protocols.py

@@ -9,6 +9,7 @@
     runtime_checkable,
 )
 
+from sqlalchemy_bind_manager._repository import SortDirection
 from sqlalchemy_bind_manager._repository.common import (
     MODEL,
     PRIMARY_KEY,
@@ -17,7 +18,6 @@
     CursorPaginatedResult,
     CursorReference,
     PaginatedResult,
-    SortDirection,
 )