Skip to content

Commit f63802c

Browse files
committed
improve job_create_v2
1 parent 15b93a5 commit f63802c

File tree

1 file changed

+6
-25
lines changed
  • dpdispatcher/dpcloudserver

1 file changed

+6
-25
lines changed

dpdispatcher/dpcloudserver/api.py

Lines changed: 6 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)