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