manyi/script/upgrade/pending/upgrade_20180514_agent_income_report.py

# -*- coding: utf-8 -*-
#!/usr/bin/env python

"""
更改agentIncomeReport 字段名，为了与其他model统一使用query 对象
"""

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.agent.models import AgentIncomeReport

agentIncomeReportCollection = AgentIncomeReport.get_collection()

reports = agentIncomeReportCollection.find({'createdTime': {'$ne': ''}})

if not reports: print 'no reports to operate on'

for report in reports:
    print 'updating report=%s' % (str(report['_id']),)

    updated0 = agentIncomeReportCollection.update({'_id': ObjectId(report['_id'])},
                                                  {'$set': {'dateTimeAdded': report['createdTime']}})
    assert updated0, 'updated0 failed, report=%s' % (str(report['_id']),)

    updated1 = agentIncomeReportCollection.update({'_id': ObjectId(report['_id'])},
                                                  {'$unset': {'createdTime': ''}})
    assert updated1, 'updated1 failed, report=%s' % (str(report['_id']),)

print 'finished!'