fix

linucb/views/collect_data.py

@@ -252,6 +252,40 @@ class CollectData(object):
             logging.error("get_device_tag_ctr error!")
             return 0.001
 
+    # 用户打标签加分
+    # 新增四种用户兴趣分行为
+    # 四种日志均为后端埋点日志
+    def transfer_update_recommend_tag_list(self, device_id, user_feature, user_id, tag_list, score_loop=1):
+        if len(tag_list) > 0:
+            is_click = 1
+            is_vote = 0
+            reward = 1 if is_click or is_vote else 0
+            # 移植老用户的lin信息到ctr特征策略
+            self.transfer_old_info2ctr_feature_key(device_id)
+            for i in range(score_loop):
+                for tag_id in tag_list:
+                    self.update_user_linucb_tag_info(reward, device_id, tag_id,
+                                                     user_feature,
+                                                     self.linucb_matrix_redis_prefix)
+                    # 获取tag的ctr信息
+                    device_tag_ctr = self.get_device_tag_ctr(device_id, tag_id)
+                    user_feature_ctr = [device_tag_ctr, device_tag_ctr]
+                    self.update_user_linucb_tag_info(reward, device_id, tag_id,
+                                                     user_feature_ctr, self.ctr_linucb_matrix_redis_prefix)
+            # 更新该用户的推荐tag数据，放在 更新完成user tag行为信息之后
+            self.update_recommend_tag_list(device_id, user_feature, user_id,
+                                           click_topic_tag_list=tag_list,
+                                           linucb_matrix_prefix=self.linucb_matrix_redis_prefix,
+                                           linucb_recommend_tag_prefix=self.linucb_recommend_redis_prefix,
+                                           linucb_topic_ids_prefix=self.linucb_recommend_topic_id_prefix,
+                                           linucb_pictorial_ids_prefix=self.linucb_recommend_pictorial_id_prefix)
+            self.update_recommend_tag_list(device_id, user_feature, user_id,
+                                           click_topic_tag_list=tag_list,
+                                           linucb_matrix_prefix=self.ctr_linucb_matrix_redis_prefix,
+                                           linucb_recommend_tag_prefix=self.ctr_linucb_recommend_redis_prefix,
+                                           linucb_topic_ids_prefix=self.ctr_linucb_recommend_topic_id_prefix,
+                                           linucb_pictorial_ids_prefix=self.ctr_linucb_recommend_pictorial_id_prefix)
+
     def consume_data_from_kafka(self, topic_name=None):
         try:
             user_feature = [1, 1]
@@ -595,37 +629,37 @@ class CollectData(object):
                                                       str(tag_query_results_multi)))
                                 # 首页搜索精准匹配标签关键字进linucb
                                 elif 'SYS' in data and 'APP' in data and 'action' in data['SYS'] and \
-                                        data['SYS']['action'] == "api/v1/cards/topic":
-                                    tag_name = raw_val_dict["APP"].get("query", [])
+                                        "api/v1/cards/topic" in data['SYS']['action']:
+                                    tag_name = data["APP"].get("query", [])
                                     tag_list = list(Tag.objects.using(settings.SLAVE1_DB_NAME).filter(
                                         tag=tag_name).values_list("id"))
-                                    device_id = raw_val_dict["device"]["device_id"]
-                                    user_id = raw_val_dict["user_id"] if "user_id" in raw_val_dict else None
+                                    device_id = data["SYS"]["cl_id"]
+                                    user_id = data['SYS'].get('user_id', None)
                                     self.transfer_update_recommend_tag_list(device_id, user_feature, user_id,
                                                                             tag_list,
                                                                             5)
                                 # (客户端创建回答,后台创建回答或修改回答关联标签关键字进linucb
                                 elif 'SYS' in data and 'APP' in data and 'action' in data['SYS'] and \
-                                        (data['SYS']['action'] in (
-                                                "venus/community/topic/create", "venus/sun/topic/edit")):
-                                    tag_ids = list(raw_val_dict["APP"].get("tag_ids", []))
+                                        ("venus/community/topic/create" in data['SYS'][
+                                            'action'] or "venus/sun/topic/edit" in data['SYS']['action']):
+                                    tag_ids = list(data["APP"].get("tag_ids", []))
                                     tag_list = list(Tag.objects.using(settings.SLAVE1_DB_NAME).filter(
                                         id__in=tag_ids, is_online=True, is_deleted=False,
                                         is_category=False).values_list("id"))
-                                    device_id = raw_val_dict["device"]["device_id"]
-                                    user_id = raw_val_dict["user_id"] if "user_id" in raw_val_dict else None
+                                    device_id = data["SYS"]["cl_id"]
+                                    user_id = data['SYS'].get('user_id', None)
                                     self.transfer_update_recommend_tag_list(device_id, user_feature, user_id,
                                                                             tag_list,
                                                                             10)
                                 # 创建问题关注标签关键字进linucb
                                 elif 'SYS' in data and 'APP' in data and 'action' in data['SYS'] and \
-                                        data['SYS']['action'] == "venus/sun/pictorial/edit":
-                                    tag_ids = list(raw_val_dict["APP"].get("tag_ids", []))
+                                        "venus/sun/pictorial/edit" in data['SYS']['action']:
+                                    tag_ids = list(data["APP"].get("tag_ids", []))
                                     tag_list = list(Tag.objects.using(settings.SLAVE1_DB_NAME).filter(
                                         id__in=tag_ids, is_online=True, is_deleted=False,
                                         is_category=False).values_list("id"))
-                                    device_id = raw_val_dict["device"]["device_id"]
-                                    user_id = raw_val_dict["user_id"] if "user_id" in raw_val_dict else None
+                                    device_id = data["SYS"]["cl_id"]
+                                    user_id = data['SYS'].get('user_id', None)
                                     self.transfer_update_recommend_tag_list(device_id, user_feature, user_id,
                                                                             tag_list,
                                                                             20)
@@ -638,7 +672,7 @@ class CollectData(object):
                             logging.error("catch exception,err_msg:%s" % traceback.format_exc())
                             # 假设数据库连接异常，强制退出程序，supervisor重启linub
                             os._exit(0)
-            return True
+                return True
         except:
             logging_exception()
             logging.error("catch exception,err_msg:%s" % traceback.format_exc())