title_with_video/server/api/get_off_videos.py

"""
@author: luojunhui
"""
import json
import time

from applications.log import logging
from applications.functions.forward_request import forward_requests


class GetOffVideos(object):
    """
    下架视频
    """

    def __init__(self, params, mysql_client, config):
        self.params = params
        self.mysql_client = mysql_client
        self.article_match_video_table = config.article_match_video_table
        self.get_off_videos = config.get_off_video_table
        self.trace_id = None

    def check_params(self):
        """

        :return:
        """
        try:
            self.trace_id = self.params['traceId']
            logging(
                code="4121",
                info='自动下架视频接口参数校验成功',
                function="get_off_videos",
                data=self.params,
                trace_id=self.trace_id
            )
            return None
        except Exception as e:
            response = {
                "error": "params error",
                "info": str(e),
                "data": self.params
            }
            return response

    async def push_video_into_queue(self):
        """
        将视频id记录到待下架表中
        :return:
        """
        select_sql = f"""
        select response from {self.article_match_video_table} where trace_id = '{self.trace_id}';
        """
        result = await self.mysql_client.async_select(sql=select_sql)
        if result:
            video_list = json.loads(result[0][0])
            info_list = [
                (
                    item['videoId'],
                    int(time.time()),
                    1,
                    self.trace_id
                )
                for item in video_list
            ]
            try:
                insert_sql = f"""
                    INSERT INTO {self.get_off_videos}
                    (video_id, publish_time, video_status, trace_id)
                    values 
                    (%s, %s, %s, %s);"""
                await self.mysql_client.async_insert_many(
                    sql=insert_sql,
                    params_list=info_list
                )
                logging(
                    code="4122",
                    info='自动下架视频接口存储成功',
                    function="get_off_videos",
                    trace_id=self.trace_id
                )
                return {
                    "status": "success",
                    "traceId": self.trace_id
                }
            except Exception as e:
                logging(
                    code="4123",
                    info="自动下架视频处理失败: {}".format(e),
                    function="get_off_videos",
                    trace_id=self.trace_id
                )
                return {
                    "status": "fail",
                    "traceId": self.trace_id,
                    "msg": "insert fails, {}".format(e)
                }
        else:
            return {
                "status": "fail",
                "traceId": self.trace_id,
                "msg": "traceId error, can't find trace_id"
            }

    async def check_trace_id(self):
        """
        check trace id 是否存在与系统中
        """
        select_sql = f"""
            SELECT trace_id
            FROM {self.article_match_video_table}
            WHERE trace_id = '{self.trace_id}';
            """
        response = await self.mysql_client.async_select(select_sql)
        if response:
            return True
        else:
            return False

    async def check_trace_id_startswith(self):
        """
        trace_id是否以search开头
        """
        if self.trace_id.startswith('search'):
            return True
        else:
            return False

    async def deal(self):
        """
        :return:
        """
        params_error = self.check_params()
        if params_error:
            return params_error
        else:
            trace_id_process_flag = await self.check_trace_id_startswith()
            if not trace_id_process_flag:
                return {
                    "traceId": self.trace_id,
                    "msg": "do not process trace id"
                }

            trace_id_exist_flag = await self.check_trace_id()
            if trace_id_exist_flag:
                return await self.push_video_into_queue()
            else:
                # 只需要传trace_id, 老系统接口不穿strategy参数默认strategy_v1
                response = await forward_requests(
                    params={
                        "traceId": self.trace_id
                    },
                    api="get_off_videos"
                )
                return response