def post(self):
"""
任务同步
"""
done_status = [TaskStatus.DONE, TaskStatus.STOP, TaskStatus.ERROR]
args = self.parse_args(sync_task_fields)
task_id = args.pop('task_id')
scope_id = args.pop('scope_id')
query = {'_id': ObjectId(task_id)}
task_data = utils.conn_db('task').find_one(query)
if not task_data:
return utils.build_ret(ErrorMsg.NotFoundTask, {"task_id": task_id})
asset_scope_data = utils.conn_db('asset_scope').find_one(
{'_id': ObjectId(scope_id)})
if not asset_scope_data:
return utils.build_ret(ErrorMsg.NotFoundScopeID,
{"task_id": task_id})
if task_data.get("type") != "domain":
return utils.build_ret(ErrorMsg.TaskTypeIsNotDomain,
{"task_id": task_id})
if not utils.is_in_scopes(task_data["target"],
asset_scope_data["scope_array"]):
return utils.build_ret(ErrorMsg.TaskTargetNotInScope,
{"task_id": task_id})
if task_data["status"] not in done_status:
return utils.build_ret(ErrorMsg.TaskIsRunning,
{"task_id": task_id})
task_sync_status = task_data.get("sync_status", TaskSyncStatus.DEFAULT)
if task_sync_status not in [
TaskSyncStatus.DEFAULT, TaskSyncStatus.ERROR
]:
return utils.build_ret(ErrorMsg.TaskSyncDealing,
{"task_id": task_id})
task_data["sync_status"] = TaskSyncStatus.WAITING
options = {
"celery_action": CeleryAction.DOMAIN_TASK_SYNC_TASK,
"data": {
"task_id": task_id,
"scope_id": scope_id
}
}
celerytask.arl_task.delay(options=options)
conn('task').find_one_and_replace(query, task_data)
return utils.build_ret(ErrorMsg.Success, {"task_id": task_id})
def get(self):
"""
任务同步资产组查询
"""
args = self.parser.parse_args()
target = args.pop("target")
if not utils.is_valid_domain(target):
return utils.build_ret(ErrorMsg.DomainInvalid, {"target": target})
args["scope_array"] = utils.get_fld(target)
args["size"] = 100
args["order"] = "_id"
data = self.build_data(args=args, collection='asset_scope')
ret = []
for item in data["items"]:
if utils.is_in_scopes(target, item["scope_array"]):
ret.append(item)
data["items"] = ret
data["total"] = len(ret)
return data