Procházet zdrojové kódy

在回滚的版本上增加过滤的兜底:修改规则为list

zhangbo před 1 rokem
rodič
revize
3e4ab59b2d
1 změnil soubory, kde provedl 17 přidání a 11 odebrání
  1. 17 11
      utils.py

+ 17 - 11
utils.py

@@ -22,15 +22,15 @@ from parameter_update import param_update_risk_filter_flag
 config_ = set_config()
 log_ = Log()
 
-FESTIVAL = {
-    "春节": [2024020900, 2024021000],
-    "初一": [2024021010, 2024021100],
-    "初二": [2024021110, 2024021200],
-    "初三": [2024021210, 2024021300],
-    "初四": [2024021310, 2024021400],
-    "初五": [2024021410, 2024021500],
-    "情人节": [2024021410, 2024021500]
-}
+FESTIVAL = [
+    ["春节", 2024020900, 2024021000],
+    ["初一", 2024021010, 2024021100],
+    ["初二", 2024021110, 2024021200],
+    ["初三", 2024021210, 2024021300],
+    ["初四", 2024021310, 2024021400],
+    ["初五", 2024021410, 2024021500],
+    ["情人节", 2024021410, 2024021500]
+]
 
 def send_msg_to_feishu(msg_text):
     """发送消息到飞书"""
@@ -347,7 +347,11 @@ class FilterVideos(object):
         # todo: 添加app和region的风险过滤。
         st_viewed = time.time()
         videos_filtered = self.filter_videos_with_risk_video(self.video_ids, self.app_type, region_code)
+        # videos_filtered.append(18562889)
+        # videos_filtered.append(18613648)
+        # videos_filtered.append(18608478)
         videos_filtered = self.filter_videos_with_festival(videos_filtered)
+        # print(str(videos_filtered))
         # log_.info({
         #     'logTimestamp': int(time.time() * 1000),
         #     'pool_type': "zhangbo-filter-pool_type",
@@ -911,8 +915,8 @@ class FilterVideos(object):
         now_dt = datetime.strftime(now_date, '%Y%m%d%H')
         now_dt_int = int(now_dt)
         filter_fes = []
-        for k, v in FESTIVAL.items():
-            if now_dt_int >= v[0] and now_dt_int < v[1]:
+        for k, v1, v2 in FESTIVAL:
+            if now_dt_int >= v1 and now_dt_int < v2:
                 filter_fes.append(k)
         if len(filter_fes) == 0:
             return video_ids
@@ -920,6 +924,8 @@ class FilterVideos(object):
         redis_keys = ["alg_recsys_video_tags_" + str(id) for id in video_ids]
         redis_helper = RedisHelper()
         redis_values = redis_helper.get_batch_key(redis_keys)
+        # print(str(video_ids))
+        # print(str(redis_values))
         if redis_values and len(redis_values) > 0 and len(redis_values) == len(redis_keys):
             video_ids_new = []
             for id, tags in zip(video_ids, redis_values):