From bafde84864b026406e78c5a8e467a48f5a338f01 Mon Sep 17 00:00:00 2001
From: Valentin Rigal <rigal@teklia.com>
Date: Wed, 23 Feb 2022 16:05:12 +0100
Subject: [PATCH] Update with_classes parameter

---
 arkindex_worker/worker/element.py | 16 ++++------------
 1 file changed, 4 insertions(+), 12 deletions(-)

diff --git a/arkindex_worker/worker/element.py b/arkindex_worker/worker/element.py
index 747af95b..a89528ba 100644
--- a/arkindex_worker/worker/element.py
+++ b/arkindex_worker/worker/element.py
@@ -188,12 +188,11 @@ class ElementMixin(object):
     def list_element_children(
         self,
         element,
-        best_class=None,
         folder=None,
         name=None,
         recursive=None,
         type=None,
-        with_best_classes=None,
+        with_classes=None,
         with_corpus=None,
         with_has_children=None,
         with_zone=None,
@@ -206,11 +205,6 @@ class ElementMixin(object):
             element, (Element, CachedElement)
         ), "element shouldn't be null and should be an Element or CachedElement"
         query_params = {}
-        if best_class is not None:
-            assert isinstance(best_class, str) or isinstance(
-                best_class, bool
-            ), "best_class should be of type str or bool"
-            query_params["best_class"] = best_class
         if folder is not None:
             assert isinstance(folder, bool), "folder should be of type bool"
             query_params["folder"] = folder
@@ -223,11 +217,9 @@ class ElementMixin(object):
         if type:
             assert isinstance(type, str), "type should be of type str"
             query_params["type"] = type
-        if with_best_classes is not None:
-            assert isinstance(
-                with_best_classes, bool
-            ), "with_best_classes should be of type bool"
-            query_params["with_best_classes"] = with_best_classes
+        if with_classes is not None:
+            assert isinstance(with_classes, bool), "with_classes should be of type bool"
+            query_params["with_classes"] = with_classes
         if with_corpus is not None:
             assert isinstance(with_corpus, bool), "with_corpus should be of type bool"
             query_params["with_corpus"] = with_corpus
-- 
GitLab