Added order by functions

src/cpl_query/extension/iterable.py

@@ -0,0 +1,53 @@
+from typing import Optional, Callable
+
+from .ordered_iterable import OrderedIterable
+from .ordered_iterable_abc import OrderedIterableABC
+from .._query.any_query import any_query
+from .._query.first_query import first_or_default_query, first_query
+from .._query.for_each_query import for_each_query
+from .._query.order_by import order_by_query, order_by_descending_query
+from .._query.single_query import single_query, single_or_default_query
+from .._query.where_query import where_query
+from cpl_query.extension.iterable_abc import IterableABC
+
+
+class Iterable(IterableABC):
+
+    def __init__(self):
+        IterableABC.__init__(self)
+
+    def any(self, func: str) -> bool:
+        return any_query(self, func)
+
+    def first(self) -> any:
+        return first_query(self)
+
+    def first_or_default(self) -> Optional[any]:
+        return first_or_default_query(self)
+
+    def for_each(self, func: Callable):
+        for_each_query(self, func)
+
+    def order_by(self, func: Callable) -> OrderedIterableABC:
+        res = order_by_query(self, func)
+        res.__class__ = OrderedIterable
+        return res
+
+    def order_by_descending(self, func: Callable) -> OrderedIterableABC:
+        res = order_by_descending_query(self, func)
+        res.__class__ = OrderedIterable
+        return res
+
+    def single(self) -> any:
+        return single_query(self)
+
+    def single_or_default(self) -> Optional[any]:
+        return single_or_default_query(self)
+
+    def where(self, func: str) -> IterableABC:
+        res = where_query(self, func)
+        res.__class__ = Iterable
+        return res
+
+    def to_list(self) -> list:
+        return list(self)

src/cpl_query/extension/iterable_abc.py

@@ -20,6 +20,12 @@ class IterableABC(ABC, list):
     @abstractmethod
     def for_each(self, func: Callable): pass
 
+    @abstractmethod
+    def order_by(self, func: Callable): pass
+
+    @abstractmethod
+    def order_by_descending(self, func: Callable): pass
+
     @abstractmethod
     def single(self): pass
 

src/cpl_query/extension/list.py

@@ -1,40 +1,12 @@
-from typing import Optional, Callable
-
-from .._query.any_query import any_query
-from .._query.first_query import first_or_default_query, first_query
-from .._query.for_each_query import for_each_query
-from .._query.single_query import single_query, single_or_default_query
-from .._query.where_query import where_query
-from cpl_query.extension.iterable_abc import IterableABC
+from cpl_query.extension.iterable import Iterable
 
 
-class List(IterableABC):
+class List(Iterable):
 
-    def __init__(self):
-        IterableABC.__init__(self)
+    def __init__(self, t: type = None, values: list = None):
+        Iterable.__init__(self)
 
-    def any(self, func: str) -> bool:
-        return any_query(self, func)
+        self._type = t
 
-    def first(self) -> any:
-        return first_query(self)
-
-    def first_or_default(self) -> Optional[any]:
-        return first_or_default_query(self)
-
-    def for_each(self, func: Callable):
-        for_each_query(self, func)
-
-    def single(self) -> any:
-        return single_query(self)
-
-    def single_or_default(self) -> Optional[any]:
-        return single_or_default_query(self)
-
-    def where(self, func: str) -> IterableABC:
-        res = where_query(self, func)
-        res.__class__ = List
-        return res
-
-    def to_list(self) -> list:
-        return list(self)
+        if values is not None:
+            self.extend(values)

src/cpl_query/extension/ordered_iterable.py

@@ -0,0 +1,19 @@
+from abc import ABC
+from collections import Callable
+
+from .._query.order_by import then_by_query, then_by_descending_query
+from cpl_query.extension.ordered_iterable_abc import OrderedIterableABC
+
+
+class OrderedIterable(OrderedIterableABC, ABC):
+
+    def __init__(self):
+        OrderedIterableABC.__init__(self)
+
+    def then_by(self, _func: Callable) -> OrderedIterableABC:
+        self._funcs.append(_func)
+        return then_by_query(self, lambda *args: [f(*args) for f in self._funcs])
+
+    def then_by_descending(self, _func: Callable) -> OrderedIterableABC:
+        self._funcs.append(_func)
+        return then_by_descending_query(self, lambda *args: [f(*args) for f in self._funcs])

src/cpl_query/extension/ordered_iterable_abc.py

@@ -0,0 +1,20 @@
+from abc import abstractmethod
+from collections import Callable
+
+from cpl_query.extension.iterable_abc import IterableABC
+
+
+class OrderedIterableABC(IterableABC):
+
+    @abstractmethod
+    def __init__(self, _func: Callable = None):
+        IterableABC.__init__(self)
+        self._funcs: list[Callable] = []
+        if _func is not None:
+            self._funcs.append(_func)
+
+    @abstractmethod
+    def then_by(self, func: Callable) -> 'OrderedIterableABC': pass
+
+    @abstractmethod
+    def then_by_descending(self, func: Callable) -> 'OrderedIterableABC': pass