py_script/task_clean/update_first_type.py

#!/usr/bin/env python
#coding=utf-8

"""电视台收视率统计

"""
import sys

from fty_util.common import Mysql

reload(sys)
sys.setdefaultencoding('utf8')

conn = Mysql.createOfflineConn()

sql = """
    select tv_id, tv_name, first_type from odl.ad_tv_lib where is_use = 1 and decade is null
"""
rows = Mysql.getAll(sql, conn=conn)

for row in rows:
    tv_id = row['tv_id']
    tv_name = row['tv_name']
    first_type = row['first_type']

    sql = """
        select name, theme from odl.ad_tv_record_distribution where name = '%s'
    """
    sql = sql % (tv_name)
    row = Mysql.getOne(sql, conn=conn)
    if row is not None:
        theme = row[1]
        if theme is not None and len(theme) > 0:
            if first_type is None or len(first_type) == 0:
                first_type = theme
            else:
                decade = theme[:2]
                update_sql =  """
                    update odl.ad_tv_lib set decade = '%s' where tv_id = '%s'
                """
                update_sql = update_sql % (decade, tv_id)
                Mysql.update(update_sql, conn=conn)
                # first_type = theme[:2] + first_type
    # if first_type is not None:
    #     # update_sql =  """
    #     #     update odl.ad_tv_lib set first_type = '%s' where tv_id = '%s'
    #     # """
    #     update_sql =  """
    #         update odl.ad_tv_lib set decade = '%s' where tv_id = '%s'
    #     """
    #     update_sql = update_sql % (first_type, tv_id)
    #     Mysql.update(update_sql, conn=conn)
Mysql.close(conn)