title_with_video/applications/deal/record.py

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

from uuid import uuid4

from applications.functions.log import logging
from applications.functions.forward import forward_requests
from static.config import db_article


class Record(object):
    """
    搜索接口处理逻辑
    """
    NEW_STRATEGY = "strategy_v2"

    def __init__(self, params, mysql_client):
        self.content_id = None
        self.account_name = None
        self.contents = None
        self.title = None
        self.gh_id = None
        self.strategy = None
        self.params = params
        self.mysql_client = mysql_client
        self.trace_id = "search-{}-{}".format(str(uuid4()), str(int(time.time())))

    def check_params(self):
        """
        检查请求params
        :return:
        """
        try:
            self.gh_id = self.params['ghId']
            self.title = self.params['title'].split("@@")[-1].replace("'", "")
            self.contents = self.params['content'].replace("'", "")
            self.account_name = self.params['accountName'].replace("'", "")
            self.content_id = self.params['articleId']
            self.strategy = self.params['strategy']
            logging(
                code="1001",
                info="搜索视频内容接口请求成功, 参数校验成功",
                port="title_to_search",
                trace_id=self.trace_id,
                data=self.params
            )
            return None
        except Exception as e:
            result = {
                "status": "fail",
                "code": 1,
                "message": str(e),
                "info": "params check error"
            }
            logging(
                code="4001",
                info="搜索视频内容接口请求成功, 参数校验失败",
                port="title_to_search",
                trace_id=self.trace_id,
                data=self.params
            )
            return result

    async def input_queue(self):
        """
        把数据插入待处理队列
        :return:
        """
        request_time = int(time.time())
        insert_sql = f"""
                        INSERT INTO {db_article}
                            (trace_id, content_id, gh_id, account_name, article_title, article_text, content_status, success, request_time_stamp)
                        VALUES 
                            (%s, %s, %s, %s, %s, %s, %s, %s, %s);"""
        await self.mysql_client.async_insert(
            sql=insert_sql,
            params=(self.trace_id, self.content_id, self.gh_id, self.account_name, self.title, self.contents, 0, 0, request_time)
        )
        logging(
            code="1002",
            info="成功记录请求数据到mysql中",
            trace_id=self.trace_id
        )

    async def deal(self):
        """
        deal
        :return:
        """
        params_error = self.check_params()
        if params_error:
            return params_error
        else:
            # 记录
            if self.strategy == self.NEW_STRATEGY:
                response = await forward_requests(
                    params=self.params,
                    api="search_videos"
                )
                return response
            await self.input_queue()
            res = {
                "status": "success input to article queue",
                "code": 0,
                "traceId": self.trace_id
            }
            return res