فهرست منبع

修改阈值分数为0.4, 增加分页处理

罗俊辉 11 ماه پیش
والد
کامیت
7b216526bd
1فایلهای تغییر یافته به همراه10 افزوده شده و 12 حذف شده
  1. 10 12
      deal/videos_deal.py

+ 10 - 12
deal/videos_deal.py

@@ -14,7 +14,8 @@ class VideoDeal(object):
         self.start_dt = params['start_date']
         self.end_dt = params['end_date']
         self.cate = params['cate']
-        self.topN = params['topN'] if params.get("topN") else 1
+        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):
@@ -33,7 +34,7 @@ class VideoDeal(object):
             for index, obj in enumerate(temp_list):
                 temp = {
                     "video_id": obj[0],
-                    "title": obj[1],
+                    "title": obj[1].replace('"', ''),
                     "view": obj[2],
                     "return": obj[3],
                     "video_url": obj[4],
@@ -49,7 +50,7 @@ class VideoDeal(object):
             "cate": self.cate,
             "start_date": self.start_dt,
             "end_dt": self.end_dt,
-            "topN": self.topN,
+            # "topN": self.topN,
             "data": L
         }
         return response
@@ -74,20 +75,17 @@ class VideoDeal(object):
         :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.topN};
+            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)
-        # print(score_list)
         res = list(zip(score_list, title_list))
-        s_r = sorted(res, key=lambda x:x[0], reverse=True)
-        for i in s_r:
-            print(i)
         result_list_final = []
         for index, item in enumerate(result_list):
             if score_list[index] > 0.4: