Precommit.

Author: simonvb00

stac_fastapi/elasticsearch/stac_fastapi/elasticsearch/database_logic.py

@@ -27,7 +27,6 @@
     PartialItem,
     PatchOperation,
 )
-
 from stac_fastapi.sfeos_helpers import filter
 from stac_fastapi.sfeos_helpers.database import (
     apply_free_text_filter_shared,
@@ -44,6 +43,10 @@
     return_date,
     validate_refresh,
 )
+from stac_fastapi.sfeos_helpers.database.query import (
+    ES_MAX_URL_LENGTH,
+    add_collections_to_body,
+)
 from stac_fastapi.sfeos_helpers.database.utils import (
     merge_to_operations,
     operations_to_script,
@@ -61,9 +64,6 @@
 from stac_fastapi.types.rfc3339 import DateTimeType
 from stac_fastapi.types.stac import Collection, Item
 
-from stac_fastapi.sfeos_helpers.database.query import add_collections_to_body, \
-    ES_MAX_URL_LENGTH
-
 logger = logging.getLogger(__name__)
 
 

stac_fastapi/opensearch/stac_fastapi/opensearch/database_logic.py

@@ -42,6 +42,10 @@
     return_date,
     validate_refresh,
 )
+from stac_fastapi.sfeos_helpers.database.query import (
+    ES_MAX_URL_LENGTH,
+    add_collections_to_body,
+)
 from stac_fastapi.sfeos_helpers.database.utils import (
     merge_to_operations,
     operations_to_script,
@@ -62,9 +66,6 @@
 from stac_fastapi.types.rfc3339 import DateTimeType
 from stac_fastapi.types.stac import Collection, Item
 
-from stac_fastapi.sfeos_helpers.database.query import add_collections_to_body, \
-    ES_MAX_URL_LENGTH
-
 logger = logging.getLogger(__name__)
 
 

stac_fastapi/sfeos_helpers/stac_fastapi/sfeos_helpers/database/query.py

@@ -9,6 +9,7 @@
 
 ES_MAX_URL_LENGTH = 4096
 
+
 def apply_free_text_filter_shared(
     search: Any, free_text_queries: Optional[List[str]]
 ) -> Any:
@@ -86,8 +87,10 @@ def populate_sort_shared(sortby: List) -> Optional[Dict[str, Dict[str, str]]]:
         return None
 
 
-def add_collections_to_body(collection_ids: List[str], query: Optional[Dict[str, Any]]) -> Dict[str, Any]:
-    """Adds a list of collection ids to the body of a query.
+def add_collections_to_body(
+    collection_ids: List[str], query: Optional[Dict[str, Any]]
+) -> Dict[str, Any]:
+    """Add a list of collection ids to the body of a query.
 
     Args:
         collection_ids (List[str]): A list of collections ids.
@@ -111,4 +114,4 @@ def add_collections_to_body(collection_ids: List[str], query: Optional[Dict[str,
     filters = query["bool"]["filter"]
     if index_filter not in filters:
         filters.append(index_filter)
-    return query
+    return query