manyi/script/upgrade/removed/upgrade_20180504_adrecord.py

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

"""
本地化部分数据，减少大量集中db query
"""

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

logger = get_logger(__name__)

init_env(interactive=True)

from apps.web.ad.models import AdRecord
from apps.web.user.models import MyUser

for record in AdRecord.objects():
    logger.info('saving record(id=%s)' % (record.id,))

    #: 标注用户性别

    user = MyUser.objects(openId=record.openId, groupId=record.groupId).first()
    maleFans = int(user.is_male) if (user is not None and record.converted) else 0
    femaleFans = int(user.is_female) if (user is not None and record.converted) else 0
    record.features['sex'] = {'is_male': maleFans, 'is_female': femaleFans}
    record.save()

logger.info('finished!')