liqian hace 3 años
padre
commit
98b9e3ebeb
Se han modificado 2 ficheros con 4 adiciones y 4 borrados
  1. 3 3
      utils.py
  2. 1 1
      video_recall.py

+ 3 - 3
utils.py

@@ -145,7 +145,7 @@ class FilterVideos(object):
         if not filtered_viewed_result:
             return None
         else:
-            return filtered_viewed_result
+            return [int(video_id) for video_id in filtered_viewed_result]
 
     def filter_video_previewed(self, video_ids):
         """
@@ -160,7 +160,7 @@ class FilterVideos(object):
         pe_videos_list = redis_helper.get_data_from_set(key_name)
         if not pe_videos_list:
             return video_ids
-        pe_videos = [eval(video) for video in pe_videos_list]
+        pe_videos = [int(video) for video in pe_videos_list]
         filtered_videos = [video_id for video_id in video_ids if video_id not in pe_videos]
         return filtered_videos
 
@@ -195,7 +195,7 @@ class FilterVideos(object):
 
         hologres_helper = HologresHelper()
         data = hologres_helper.get_data(sql=sql)
-        filtered_videos = [temp[0] for temp in data]
+        filtered_videos = [int(temp[0]) for temp in data]
         return filtered_videos
 
     def filter_video_viewed(self, video_ids, types=(1,)):

+ 1 - 1
video_recall.py

@@ -113,7 +113,7 @@ class PoolRecall(object):
             video_score = {}
             for value in data:
                 video_id, flow_pool = value[0].split('-')
-                video_id = eval(video_id)
+                video_id = int(video_id)
                 if video_id not in video_ids:
                     video_ids.append(video_id)
                     video_score[video_id] = value[1]