upgrade_offline_tasks_20180919.py 795 B

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