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

"""
广告记录添加逻辑编码
"""

import os
import sys

from bson.objectid import ObjectId

#: 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.device.models import Device

for record in AdRecord.objects():
    dev = Device.get_dev(record.devNo) or {}
    updated = record.update(logicalCode=dev.get('logicalCode', ''))
    assert updated, 'unable to update record(id=%s)' % (str(record.id),)

    print 'record(id=%s) updated' % (str(record.id),)

print 'finished!'