add test score

search/utils/topic.py

@@ -93,7 +93,7 @@ class TopicUtils(object):
             return []
 
     @classmethod
-    def get_recommend_topic_ids(cls,user_id,offset,size,query=None,query_type=TopicPageType.FIND_PAGE,filter_topic_id_list=[]):
+    def get_recommend_topic_ids(cls,user_id,offset,size,query=None,query_type=TopicPageType.FIND_PAGE,filter_topic_id_list=[],test_score=False):
         """
         :需增加打散逻辑
         :remark:获取首页推荐帖子列表
@@ -254,11 +254,19 @@ class TopicUtils(object):
             result_dict = ESPerform.get_search_results(ESPerform.get_cli(), sub_index_name="topic", query_body=q,
                                                        offset=offset, size=size)
 
-            topic_id_dict = dict()
-            for item in result_dict["hits"]:
-                topic_id_dict[item["_source"]["id"]] = [item["_source"]["group_id"],item["_source"]["user_id"]]
+            if not test_score:
+                topic_id_dict = dict()
+                for item in result_dict["hits"]:
+                    topic_id_dict[item["_source"]["id"]] = [item["_source"]["group_id"],item["_source"]["user_id"]]
+
+                return topic_id_dict
+            else:
+                topic_id_dict = dict()
+                for item in result_dict["hits"]:
+                    topic_id_dict[item["_source"]["id"]] = [item["_source"]["group_id"],item["_source"]["user_id"],item["_score"]]
+
+                return topic_id_dict
 
-            return topic_id_dict
         except:
             logging.error("catch exception,err_msg:%s" % traceback.format_exc())
             return dict()

trans2es/management/commands/trans2es_data2es_parallel.py

@@ -127,7 +127,7 @@ class Command(BaseCommand):
 
     def generate_topic_score_detail(self):
         try:
-            topic_id_dict = TopicUtils.get_recommend_topic_ids(-1, 0, 500)
+            topic_id_dict = TopicUtils.get_recommend_topic_ids(-1, 0, 500,test_score=True)
             for topic_id in topic_id_dict:
 
                 offline_score = 0.0
@@ -182,8 +182,8 @@ class Command(BaseCommand):
                 if user_is_shadow:
                     offline_score = offline_score * 0.5
 
-                logging.info("test_score,offline_score:%f,user_is_recommend:%f,group_is_recommend:%f,topic_level_score:%f,exposure_score:%f,uv_score:%f"
-                             % (offline_score,user_is_recommend,group_is_recommend,topic_level_score,exposure_score,uv_score))
+                logging.info("test_score######score:%f,offline_score:%f,user_is_recommend:%f,group_is_recommend:%f,topic_level_score:%f,exposure_score:%f,uv_score:%f"
+                             % (topic_id_dict[topic_id][2],offline_score,user_is_recommend,group_is_recommend,topic_level_score,exposure_score,uv_score))
         except:
             logging.error("catch exception,err_msg:%s" % traceback.format_exc())