@@ -126,36 +126,17 @@ def job_create_v2(job_type, oss_path, input_data, program_id=None, group_id=None
126126 }
127127 if program_id is not None :
128128 post_data ["program_id" ] = program_id
129- if input_data .get ('job_group_id' ) is not None :
130- post_data ["job_group_id" ] = input_data .get ('job_group_id' )
131129 if group_id is not None :
132- post_data ["job_group_id" ] = group_id
133- if input_data .get ('job_name' ) is not None :
134- post_data ["job_name" ] = input_data .get ('job_name' )
135- if input_data .get ('rerun' ) is not None :
136- post_data ["rerun" ] = input_data .get ('rerun' )
137- if input_data .get ('image_name' ) is not None :
138- post_data ["image_name" ] = input_data .get ('image_name' )
139- if input_data .get ('disk_size' ) is not None :
140- post_data ["disk_size" ] = input_data .get ('disk_size' )
141- if input_data .get ('scass_type' ) is not None :
142- post_data ["scass_type" ] = input_data .get ('scass_type' )
143- if input_data .get ('instance_group_id' ) is not None :
144- post_data ["instance_group_id" ] = input_data .get ('instance_group_id' )
130+ post_data ["job_group_id" ] = group_id
145131 if input_data .get ('command' ) is not None :
146132 post_data ["cmd" ] = input_data .get ('command' )
147- if input_data .get ('log_file' ) is not None :
148- post_data ["log_file" ] = input_data .get ('log_file' )
149133 if input_data .get ('backward_files' ) is not None :
150134 post_data ["out_files" ] = input_data .get ('backward_files' )
151- if input_data .get ('platform' ) is not None :
152- post_data ["platform" ] = input_data .get ('platform' )
153- if input_data .get ('region' ) is not None :
154- post_data ["region" ] = input_data .get ('region' )
155- if input_data .get ('zone' ) is not None :
156- post_data ["zone" ] = input_data .get ('zone' )
157- if input_data .get ('on_demand' ) is not None :
158- post_data ["on_demand" ] = input_data .get ('on_demand' )
135+ input_keys = ['job_group_id' , 'job_name' , 'rerun' , 'image_name' , 'disk_size' , 'scass_type' ,
136+ 'instance_group_id' , 'log_file' , 'platform' , 'region' , 'zone' , 'on_demand' ]
137+ for key in input_keys :
138+ if key in input_data :
139+ post_data [key ] = input_data [key ]
159140 ret = post ('/data/v2/insert_job' , post_data )
160141 group_id = ret .get ('job_group_id' )
161142 return ret ['job_id' ], group_id
0 commit comments