rov-server/user2new.py

import traceback
from datetime import date, datetime, timedelta
from db_helper import RedisHelper
from config import set_config
from log import Log

config_ = set_config()
log_ = Log()


def user2new(app_type, mid, uid):
    """
    管理后台指定用户恢复成新用户
    :param app_type: 产品标识
    :param mid: mid
    :param uid: uid
    :return: None
    """
    redis_helper = RedisHelper()
    if app_type is None:
        app_type_list = [value for _, value in config_.APP_TYPE.items()]
    else:
        app_type_list = [app_type]

    for app_type_item in app_type_list:
        # 清除对应的预曝光列表
        preview_key = f"{config_.PREVIEW_KEY_PREFIX}{app_type_item}.{mid}"
        redis_helper.del_keys(key_name=preview_key)

        # 清除对应的小时级数据及数据记录
        h_key_prefix_list = [
            config_.H_WITH_MID_RECALL_KEY_NAME_PREFIX,
            config_.H_WITH_MID_RECORD_KEY_NAME_PREFIX,
            config_.H_WITH_MID_RECALL_KEY_NAME_PREFIX_24H,
            config_.H_WITH_MID_RECORD_KEY_NAME_PREFIX_24H,
            config_.H_WITH_MID_RECALL_KEY_NAME_PREFIX_REGION_24H,
            config_.H_WITH_MID_RECORD_KEY_NAME_PREFIX_REGION_24H
        ]
        for h_key_prefix in h_key_prefix_list:
            h_key_name = f"{h_key_prefix}{app_type_item}.{mid}"
            redis_helper.del_keys(key_name=h_key_name)

        # 清除用户上一次列表中对应的位置记录数据
        last_idx_key_prefix_list = [
            config_.LAST_VIDEO_FROM_ROV_POOL_PREFIX,
            config_.LAST_VIDEO_FROM_RULE_DAY_POOL_PREFIX,
            config_.LAST_VIDEO_FROM_ROV_POOL_PRE_PREFIX,
            config_.LAST_VIDEO_FROM_ROV_POOL_NOW_PREFIX,
            f"{config_.LAST_VIDEO_FROM_REGION_DUP_PREFIX}{1}.",
            f"{config_.LAST_VIDEO_FROM_REGION_DUP_PREFIX}{2}.",
            config_.LAST_VIDEO_FROM_WHOLE_MOVIES_PREFIX,
        ]
        # 天级
        now_date = date.today().strftime('%Y%m%d')
        pre_date = (date.today() - timedelta(days=1)).strftime('%Y%m%d')
        date_list = [now_date, pre_date]
        for dt in date_list:
            for last_idx_key_prefix in last_idx_key_prefix_list:
                h_key_name = f"{last_idx_key_prefix}{app_type_item}.{mid}.{dt}"
                redis_helper.del_keys(key_name=h_key_name)
        # 小时级
        h = datetime.now().hour
        h_list = [h]
        if h == 0:
            h_list.append(23)
        else:
            h_list.append(h-1)
        for redis_h in h_list:
            for last_idx_key_prefix in last_idx_key_prefix_list:
                h_key_name = f"{last_idx_key_prefix}{app_type_item}.{mid}.{redis_h}"
                redis_helper.del_keys(key_name=h_key_name)