add exposure data

linucb/views/collect_data.py

@@ -79,6 +79,8 @@ class CollectData(object):
 
     def consume_data_from_kafka(self,topic_name=None):
         try:
+            user_feature = [1,1]
+
             kafka_consumer_obj = KafkaManager.get_kafka_consumer_ins(topic_name)
             for ori_msg in kafka_consumer_obj:
                 logging.info(ori_msg)
@@ -90,7 +92,7 @@ class CollectData(object):
                         topic_id = raw_val_dict["params"]["business_id"]
                         device_id = raw_val_dict["device"]["device_id"]
 
-                        logging.info("consume topic_id:%s,device_id:%s" % (str(topic_id),str()))
+                        logging.info("consume topic_id:%s,device_id:%s" % (str(topic_id),str(device_id)))
 
                         tag_list = list()
                         sql_query_results = TopicTag.objects.filter(is_online=True,topic_id=topic_id)
@@ -100,7 +102,6 @@ class CollectData(object):
                         is_click = 1
                         is_vote = 0
 
-                        user_feature = [is_click, is_vote]
                         reward = 1 if is_click or is_vote else 0
 
                         for tag_id in tag_list:
@@ -108,6 +109,28 @@ class CollectData(object):
 
                         # 更新该用户的推荐tag数据，放在 更新完成user tag行为信息之后
                         self.update_recommend_tag_list(device_id, user_feature)
+                    elif "type" in raw_val_dict and "page_precise_exposure"==raw_val_dict["type"]:
+                        exposure_cards_list = raw_val_dict["params"]["exposure_cards"]
+                        device_id = raw_val_dict["device"]["device_id"]
+                        for item in exposure_cards_list:
+                            exposure_topic_id = item["card_id"]
+                            logging.info("consume exposure topic_id:%s,device_id:%s" % (str(exposure_topic_id),str(device_id)))
+
+                        tag_list = list()
+                        sql_query_results = TopicTag.objects.filter(is_online=True,topic_id=topic_id)
+                        for sql_item in sql_query_results:
+                            tag_list.append(sql_item.tag_id)
+
+                            is_click = 0
+                            is_vote = 0
+
+                            reward = 1 if is_click or is_vote else 0
+
+                            for tag_id in tag_list:
+                                self.update_user_linucb_tag_info(reward,device_id,tag_id,user_feature)
+
+                            # 更新该用户的推荐tag数据，放在 更新完成user tag行为信息之后
+                            self.update_recommend_tag_list(device_id, user_feature)
 
             return True
         except: