# -*- 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())