VideoToArticles/deal/videos_deal.py

"""
@author: luojunhui
"""
from applications.functions import chunks, get_info_lists, title_filter
from applications.config import minigram_info


class VideoDeal(object):
    """
    Deal Request from outside
    """

    def __init__(self, params, mysql_client):
        self.start_dt = params['start_date']
        self.end_dt = params['end_date']
        self.cate = params['cate']
        self.pageIndex = params.get("page_index", 1)
        self.pageSize = params.get("page_size", 50)
        self.mysql_client = mysql_client

    def response_obj(self, data_list):
        """
        处理 return 信息
        :param data_list:
        :return:video_id, title, view_, return_, video_url, rov
        """
        L = []
        sub_list = chunks(data_list, 20)
        for temp_list in sub_list:
            vid_list = [i[0] for i in temp_list]
            pq_response = get_info_lists(vid_list=vid_list)
            cover_list = [i['shareImgPath'] for i in pq_response['data']]
            publish_time_list = [i['gmtCreateTimestamp'] for i in pq_response['data']]
            for index, obj in enumerate(temp_list):
                temp = {
                    "video_id": obj[0],
                    "title": obj[1].replace('"', ''),
                    "view": obj[2],
                    "return": obj[3],
                    "video_url": obj[4],
                    "rov": obj[5],
                    "video_text": "",
                    "publish_time": publish_time_list[index],
                    "cover": cover_list[index],
                    "minigram_info": minigram_info
                }
                L.append(temp)
        response = {
            "status": "success",
            "cate": self.cate,
            "start_date": self.start_dt,
            "end_dt": self.end_dt,
            # "topN": self.topN,
            "data": L
        }
        return response

    async def deal(self):
        """
        deal function
        :return:
        """
        if self.cate == "video_return":
            return await self.deal_return()
        elif self.cate == "video_view":
            return await self.deal_view()
        elif self.cate == "video_rov":
            return await self.deal_rov()
        else:
            return {"error": "params is not correct"}

    async def deal_return(self):
        """
        return
        :return:
        """
        sql = f"""
            SELECT video_id, title, view_, return_, video_url, rov
            FROM top_return_daily 
            WHERE dt >= '{self.start_dt}' and dt < '{self.end_dt}' 
            ORDER BY return_ DESC
            LIMIT {self.pageSize}
            OFFSET {self.pageSize * (self.pageIndex - 1)};
        """
        result_list = await self.mysql_client.select(sql)
        title_list = [i[1] for i in result_list]
        score_list = title_filter(title_list)
        res = list(zip(score_list, title_list))
        result_list_final = []
        for index, item in enumerate(result_list):
            if score_list[index] > 0.4:
                result_list_final.append(item)
        return self.response_obj(result_list_final)

    async def deal_view(self):
        """
        view
        :return:
        """
        sql = f"""
        select video_id, title, view_, return_, video_url, rov
        from top_return_daily 
        where dt >= '{self.start_dt}' and dt < '{self.end_dt}' 
        order by view_  DESC
        limit {self.topN}
        """
        result_list = await self.mysql_client.select(sql)
        return self.response_obj(result_list)

    async def deal_rov(self):
        """
        rov
        :return:
        """
        sql = f"""
        select video_id, title, view_, return_, video_url, rov
        from top_return_daily 
        where dt >= '{self.start_dt}' and dt < '{self.end_dt}' 
        order by rov DESC
        limit {self.topN}
        """
        result_list = await self.mysql_client.select(sql)
        return self.response_obj(result_list)