implement sorting

NINAnor · Nov 10, 2023 · 9c7af04 · 9c7af04
1 parent 95969a3
commit 9c7af04
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/metadata_catalogue/datasets/managers.py b/metadata_catalogue/datasets/managers.py
@@ -4,6 +4,10 @@
 from .libs.csw_mapping import CSWMapping
 from .libs.csw_query import Group
 
+SORT_CONFIG = {
+    "title": "metadata__title",
+}
+
 
 class DatasetQuerySet(models.QuerySet):
     def as_csw(self, *args, warn=True, **kwargs):
@@ -21,6 +25,13 @@ def csw_filter(self, filter):
                 return q
         return self
 
+    def csw_sort(self, sort):
+        try:
+            return self.order_by(f'{"-" if sort["order"] == "DESC" else ""}{SORT_CONFIG[sort["propertyname"]]}')
+        except KeyError:
+            logger.warn(f"Not implemented! {sort.propertyname}")
+        return self
+
 
 class DatasetManager(models.Manager):
     def get_queryset(self):

diff --git a/metadata_catalogue/datasets/repository.py b/metadata_catalogue/datasets/repository.py
@@ -76,7 +76,8 @@ def query(self, constraint, sortby=None, typenames=None, maxrecords=10, startpos
         limit = int(maxrecords)
         offset = int(startposition)
         query = Dataset.objects.csw_filter(constraint)
-        # TODO: sort
+        if sortby:
+            query = query.csw_sort(sortby)
         return [str(query.count()), query[offset : offset + limit].as_csw()]
 
         # # run the raw query and get total