丁云鹏 hace 3 meses
padre
commit
b3b5e440f1
Se han modificado 1 ficheros con 5 adiciones y 4 borrados
  1. 5 4
      flowpool_data_update_with_level_v2.py

+ 5 - 4
flowpool_data_update_with_level_v2.py

@@ -58,7 +58,7 @@ def update_remain_view_count(video_info_list):
     """
     redis_helper = RedisHelper()
     if not video_info_list:
-        return {}
+        return dict()
 
     remain_videos = dict()
     # 每次请求10个
@@ -79,7 +79,7 @@ def update_remain_view_count(video_info_list):
                 continue
             distribute_count = int(item['distributeCount'])
             if distribute_count > 0:
-                remain_videos[item['videoId']].add(distribute_count)
+                remain_videos[item['videoId']]=distribute_count
                 # 将分发数更新到本地记录
                 key_name = f"{config_.LOCAL_DISTRIBUTE_COUNT_PREFIX}{item['videoId']}:{item['flowPool']}"
                 redis_helper.set_data_to_redis(key_name=key_name, value=distribute_count, expire_time=25 * 60)
@@ -173,6 +173,7 @@ def update_flow_pool(flow_pool_id_list):
         for video_id,distribute_count in remain_videos:
             for item in mapping.get(video_id):
                 flow_pool = item['flowPool']
+                level = item['level']
                 # 判断是否为快速曝光流量池视频
                 value = '{}-{}'.format(video_id, flow_pool)
                 flow_pool_id = int(flow_pool.split('#')[0])  # flowPool: 流量池ID#分级ID#级别Level#生命周期ID
@@ -204,7 +205,7 @@ def update_flow_pool(flow_pool_id_list):
                                                expire_time=15 * 60)
 
         # 3. 普通流量池
-        for level, videos in redis_data.items():
+        for level, videos in flow_pool_redis_data.items():
             log_.info(f"level: {level}, videos_count: {len(videos)}")
             flow_pool_key_name = f"flow:pool:level:item:v2:{app_type}:{level}"
             # 如果key已存在,删除key
@@ -221,7 +222,7 @@ def update_flow_pool(flow_pool_id_list):
 
 
         # 4. 写入权重
-        redis_helper.set_data_to_redis(key_name="flow:pool:level:weight:v2", value=json.dump(level_weight), expire_time=24*3600): 
+        redis_helper.set_data_to_redis(key_name="flow:pool:level:weight:v2", value=json.dumps(level_weight), expire_time=24*3600)
 
 
         log_.info('data to redis finished!')