feat: add job search and archive (#44)

SDK now allows to search jobs by name and archive a list of jobs by given id list.

Co-authored-by: Pablo <pjestradac@gmail.com>

Author: PJEstrada

sdk/diffgram/core/core.py

@@ -48,7 +48,6 @@ def __init__(
         if host is None:
             if self.debug is True:
                 self.host = "http://127.0.0.1:8085"
-                print("Debug", __version__)
             elif self.staging is True:
                 self.host = "https://20200110t142358-dot-walrus-dot-diffgram-001.appspot.com/"
             else:
@@ -79,7 +78,6 @@ def __init__(
 
         if init_default_directory is True:
             self.set_default_directory(directory = self.directory)
-            print("Default directory set:", self.directory_id)
 
         if refresh_local_label_dict is True:
             self.get_label_file_dict()

sdk/diffgram/file/file_constructor.py

@@ -148,7 +148,7 @@ def from_blob_path(self,
         :param frame_packet_map:
         :return:
         """
-        if self.client.default_directory:
+        if not self.client.default_directory:
             raise Exception("Directory not set. call set_default_directory() to set upload directory.")
         directory_id = self.client.directory_id
         name = file_name

sdk/diffgram/file/view.py

@@ -56,6 +56,5 @@ def get_label_file_dict(self, schema_id = None, use_session = True):
     data = response.json()
     if data["log"]["success"] == True:
         self.name_to_file_id = data["name_to_file_id"]
-        print("Loaded schema")
     else:
         raise Exception(data["log"]["errors"])

sdk/diffgram/job/job.py

@@ -239,9 +239,17 @@ def new(self,
         return job
 
     def list(self,
-             limit = 10,
-             status = "All",
+             limit = 100,
+             status = ["All"],
+             name = None,
              tags = []):
+        """
+
+        :param limit:
+        :param status: a list with any of the values: ["All", "draft", "active", "complete", "cancelled"]
+        :param tags:
+        :return:
+        """
 
         # Example usage print(project.job.list().json())
         tag_id_list = None
@@ -259,6 +267,7 @@ def list(self,
             {
                 "limit": limit,
                 "data_mode": "with_tags",
+                "search": name,
                 "status": status,
                 "tag_list": tag_id_list,
                 "project_string_id": self.client.project_string_id
@@ -348,6 +357,23 @@ def guide_update(
 
         return False
 
+    def archive_jobs(self, id_list: list):
+        job_list = [{'id': elm} for elm in id_list]
+        endpoint = "/api/v1/job/cancel"
+
+        payload = {'job_list': job_list, 'mode': 'archive'}
+
+        response = self.client.session.post(self.client.host + endpoint, json = payload)
+
+        self.client.handle_errors(response)
+
+        data = response.json()
+
+        if data["log"]["success"] == True:
+            return True
+
+        return False
+
     def get_by_id(
             self,
             id: int):