fix:fix pylint

opennaslab · May 21, 2024 · 64859fb · 64859fb
1 parent ba2b4f1
commit 64859fb
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 20 deletions.
diff --git a/kubespider/core/period_manager.py b/kubespider/core/period_manager.py
@@ -112,7 +112,7 @@ def __add_task(self, task: PeriodTask):
         elif task.task_type == "scheduler":
             kwargs = copy.deepcopy(task.arguments)
             kwargs['binding_ids'] = task.binding_ids
-            self.scheduler.add_job(id=task.name, func=self.period_scheduler_task, trigger='tigger', kwargs=kwargs)
+            self.scheduler.add_job(id=task.name, func=self.period_scheduler_task, trigger=tigger, kwargs=kwargs)
 
     def __remove_task(self, task_name: str):
         try:

diff --git a/kubespider/core/plugin/search.py b/kubespider/core/plugin/search.py
@@ -43,9 +43,8 @@ def search(self, event: SearchEvent, providers: list[SearchProvider] = None) ->
                          event.keyword, ",".join([str(p) for p in providers]))
             for provider in providers:
                 result = provider.search(event)
-                for item in result:
-                    item["plugin"] = provider.plugin_instance.definition.name
-                resource += result
+                result["plugin"] = provider.plugin_instance.definition.name
+                resource.append(result)
         return resource
 
 

diff --git a/kubespider/plugin_provider/search.py b/kubespider/plugin_provider/search.py
@@ -32,24 +32,14 @@ def get_download_provider_type(self):
     def get_prefer_download_provider(self):
         pass
 
-    def search(self, event: SearchEvent) -> list:
+    def search(self, event: SearchEvent) -> dict:
         # Return the download resources for the parser provider
         result = self.plugin_instance.call_api("search", **self.get_search_param(event))
-        # result:
-        # {"code": 200, "msg": "Success", "data": {"page": 1, "page_size": 100, "next_page": True, "data": []}}
-        # [{"code": 200, "msg": "Success", "data": {"page": 1, "page_size": 100, "next_page": True, "data": []}}]
-        if isinstance(result.get("data"), dict):
-            resource = result.get("data", {"page": 1, "page_size": 100, "next_page": False, "data": []})
-            resource["data"] = [Resource(**item).data for item in resource.get("data", [])]
-            # resource: {"page": 1, "page_size": 100, "next_page": True, "data": []}
-            return [resource, ]
-        else:
-            resources = []
-            for item in result.get("data", []):
-                item["data"] = [Resource(**item).data for item in item.get("data", [])]
-                # resource: {"page": 1, "page_size": 100, "next_page": True, "data": []}
-                resources.append(item)
-            return resources
+        # result: {"code": 200, "msg": "Success", "data": {"page": 1, "page_size": 100, "next_page": True, "data": []}}
+        resource = result.get("data", {"page": 1, "page_size": 100, "next_page": False, "data": []})
+        resource["data"] = [Resource(**item).data for item in resource.get("data", [])]
+        # resource: {"page": 1, "page_size": 100, "next_page": True, "data": []}
+        return resource
 
     def __repr__(self):
         return f"<SearchProvider[bind:{self.bind.name} plugin:{self.plugin_instance.definition.name}]>"