manyi/script/utils/translate_user_phone.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.user.models import MyUser, UniqueUser

productAgentId = '5ca5b9af0030487811eee0f0'

skip = 0
limit = 2000
count = 0

while True:
    users = MyUser.objects(__raw__ = {
        'phoneNumber': {'$nin': [None, '']}, 'productAgentId': productAgentId
    }).skip(skip).limit(limit)
    count = 0

    for user in users:  # type: MyUser
        count = count + 1

        uniqueUser = UniqueUser.get_or_create(user.openId)
        if not uniqueUser.phone:
            uniqueUser.phone = user.phoneNumber
            uniqueUser.save()

            print '{}: {}: {}'.format(str(user), str(uniqueUser), uniqueUser.phone)

    print("next {}".format(skip))

    skip = skip + limit

    if count == 0:
        break