job download log interface

Author: dingzhaohan

src/bohrium/resources/job/job.py

@@ -94,6 +94,7 @@ def detail(self, job_id):
         response = self._client.get(f"/openapi/v1/job/{job_id}")
         pprint(response.request)
         print(response.json())
+        return response.json().get("data")
 
     def create_job(
         self,
@@ -140,6 +141,11 @@ def uploadr(self, work_dir, store_path, token):
                 self.upload(full_path, object_key, token)
 
 
+    def download(self, job_id, save_path):
+        detail = self.detail(job_id)
+        tiefblue = Tiefblue()
+        tiefblue.download_from_url(detail["resultUrl"], save_path)
+        
 class AsyncJob(AsyncAPIResource):
     pass
 

tests/job.py

@@ -5,13 +5,12 @@
 if __name__ == "__main__":
 
     bohrium = Bohrium(
-        access_key="7d300756a875455a9faff8eaad2337bf",
+        access_key="7bbf9f82c3904a1c909644e0f7e5ee58",
         project_id="154",
         timeout=httpx.Timeout(40.0),
     )
 
     data = {
-        "job_type": "container",
         "job_name": "test-name",
         "project_id": 154,
         "machine_type": "c2_m4_cpu",
@@ -25,18 +24,20 @@
         "result": "/personal/test/result"
     }
 
-    job = bohrium.job.submit(
-        project_id=data["project_id"],
-        job_name=data["job_name"],
-        machine_type=data["machine_type"],
-        cmd=data["cmd"],
-        image_address=data["image_address"],
-        job_group_id=data["job_group_id"],
-        work_dir=data["work_dir"],
-        result=data["result"],
-        dataset_path=data["dataset_path"],
-        log_files=data["log_files"],
-        out_files=data["out_files"],
-    )
+    # job = bohrium.job.submit(
+    #     project_id=data["project_id"],
+    #     job_name=data["job_name"],
+    #     machine_type=data["machine_type"],
+    #     cmd=data["cmd"],
+    #     image_address=data["image_address"],
+    #     job_group_id=data["job_group_id"],
+    #     work_dir=data["work_dir"],
+    #     result=data["result"],
+    #     dataset_path=data["dataset_path"],
+    #     log_files=data["log_files"],
+    #     out_files=data["out_files"],
+    # )
 
-    job = bohrium.job.detail("12513182")
+    # job = bohrium.job.detail("12513182")
+    # job = bohrium.job.download("12513182", "./output.zip")
+    job = bohrium.job.log("12513182")