manyi/script/upgrade/pending/upgrade_ReportModel_20180718.py

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

"""

"""

import os
import sys

#: 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, setup_logger, get_current_file

setup_logger(get_current_file(), __name__)
logger = get_logger(__name__)

init_env(interactive=True)

from pymongo.collection import Collection

from apps.web.report.models import DevReport, GroupReport, DealerReport

report_models = (DealerReport, GroupReport, DevReport)


def rename(collection, change=None):
    # type:(Collection, dict) -> int
    change = {'generatedTime': 'dateTimeAdded'} if change is None else change
    return collection.update({}, {'$rename': change}, multi=True)

for _ in report_models:
    logger.info('renaming collection %s' % (_.__name__,))
    rename(_.get_collection())