Skip to content

Commit 0a00a3d

Browse files
authored
Add auto launch feature to job creation method (#20)
* feat: add auto launch feature to job creation method * remove empty Co-authored-by: Pablo <[email protected]>
1 parent 83d8488 commit 0a00a3d

File tree

1 file changed

+14
-2
lines changed

1 file changed

+14
-2
lines changed

sdk/diffgram/job/job.py

+14-2
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def refresh_from_dict(
5151
def __repr__(self):
5252
return str(self.serialize())
5353

54-
def __add_directory_to_job(self, directory, mode='sync'):
54+
def __add_directory_to_job(self, directory: Directory, mode='sync'):
5555
"""
5656
5757
:param directories: Array of directories Objects
@@ -60,7 +60,7 @@ def __add_directory_to_job(self, directory, mode='sync'):
6060
"""
6161
self.attached_directories.append(
6262
{
63-
'directory_id': directory['id'],
63+
'directory_id': directory.id,
6464
'selected': mode,
6565

6666
}
@@ -156,6 +156,7 @@ def new(self,
156156
sync_directories=[],
157157
single_copy_directories=[],
158158
members_list_ids = [],
159+
auto_launch=True,
159160
):
160161
"""
161162
@@ -231,6 +232,17 @@ def new(self,
231232
if guide:
232233
job.guide_update(guide=guide)
233234

235+
if auto_launch:
236+
endpoint_launch = "/api/v1/job/launch".format(self.client.project_string_id)
237+
response = self.client.session.post(
238+
self.client.host + endpoint_launch,
239+
json = {
240+
'job_id': data['job']['id']
241+
})
242+
self.client.handle_errors(response)
243+
244+
245+
234246
return job
235247

236248
def file_update(

0 commit comments

Comments
 (0)