From 5afbc2e0a5ec22b059771d7ba02bde7f512d7e7e Mon Sep 17 00:00:00 2001 From: Simon Date: Fri, 2 Aug 2024 18:39:24 +0200 Subject: [PATCH] fix types --- tubearchivist/video/src/query_building.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tubearchivist/video/src/query_building.py b/tubearchivist/video/src/query_building.py index fb525051..a4c5ff65 100644 --- a/tubearchivist/video/src/query_building.py +++ b/tubearchivist/video/src/query_building.py @@ -18,7 +18,7 @@ class QueryBuilder: data = {} data["query"] = self.build_query() if sort := self.parse_sort(): - data["sort"] = sort + data.update(sort) return data @@ -77,7 +77,7 @@ class QueryBuilder: return {"match": {"vid_type": vid_type}} - def parse_sort(self) -> list | None: + def parse_sort(self) -> dict | None: """build sort key""" sort = self.request_params.get("sort") if not sort: @@ -95,6 +95,5 @@ class QueryBuilder: raise ValueError(f"'{order}' not in OrderEnum") order_by = getattr(OrderEnum, order.upper()).value - sort_key = [{sort_field: {"order": order_by}}] - return sort_key + return {"sort": [{sort_field: {"order": order_by}}]}