manyi/script/bank/update_bank.py

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

import os
import sys

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.common.models import Banks
from apps.provision.bank import BANK_LIST

logger.info('start inserting bank...')

initial_sn = 10000

for item in BANK_LIST:
    logger.info('inserting...bankCode=%s' % (item['bankCode']))

    set_value = {
        'bankCode': item['bankCode'],
        'bankName': item['bankName'],
        'bankName2': item.get('bankName2', item['bankName']),
        'wechatBankCode': item.get('wechatBankCode', ''),
        'sn': initial_sn
    }

    Banks.get_collection().update({'bankCode': item['bankCode']},
                                  {
                                      '$set': set_value
                                  },
                                  multi = False, upsert = True)

    initial_sn = initial_sn + 10

logger.info('inserting bank ends..')