# -*- coding: utf-8 -*- # !/usr/bin/env python """ 将原来的离线任务都加入厂商ID, 目前都为rsd的 """ import os import sys from bson.objectid import ObjectId #: current_dir - 2 PROJECT_ROOT = os.path.join(os.path.abspath(os.path.split(os.path.realpath(__file__))[0] + "/.."), '..') sys.path.insert(0, PROJECT_ROOT) from script.base import init_env, get_logger logger = get_logger(__name__) init_env(interactive=True) from apps.web.management.models import OfflineTask rsd_manager_id = ObjectId("5abc5f5c4864d0265c654cb0") if __name__ == '__main__': for task in OfflineTask.objects(): updated = task.update(managerId=rsd_manager_id) if not updated: logger.error('updated failed, taskid = %s' % (str(task.id))) logger.info('finished!')