add

search/views/search_hotword.py

@@ -33,12 +33,23 @@ def search_hotword(device_id=-1):
     """
     try:
         all_tag_name_list = set()
-        results_registr_tag = int(redis_client.get("physical:search_hotword:results_registr_tag"))
-        results_tag = int(redis_client.get("physical:search_hotword:results_tag"))
-        tag_id = random.sample(range(1, results_registr_tag), 6)
-        results_tag_chose = Tag.objects.filter(id__in=tag_id).values_list("name", flat=True)
-        for i in results_tag_chose:
-            all_tag_name_list.add(i)
+        results_registr_tag = json.loads(redis_client.get("physical:search_hotword:results_registr_tag"))
+        results_tag = json.loads(redis_client.get("physical:search_hotword:results_tag"))
+        # 先获取搜索推荐热词
+        for num in range(0, len(results_registr_tag) - 1):
+            tag_id = random.randint(0, len(results_registr_tag) - 1)
+            results_tag_chose = list(
+                set(RegisterShowTag.objects.filter(tag_id=results_registr_tag[tag_id], is_online=True).values_list(
+                    "tag_id", flat=True)))
+            if results_tag_chose:
+                results_tag_recommend = list(
+                    set(Tag.objects.filter(id=results_tag_chose[0], is_online=True).values_list("name",
+                                                                                                flat=True)))
+                if results_tag_recommend:
+                    all_tag_name_list.add(results_tag_recommend[0])
+
+            if len(all_tag_name_list) == 6 or num == results_tag:
+                break
 
         # 获取个性化标签
         linucb_recommend_redis_prefix = "physical:linucb:tag_recommend:device_id:"
@@ -46,19 +57,24 @@ def search_hotword(device_id=-1):
         linucb_recommend_tag_data = redis_client.get(tag_recommend_redis_key)
         linucb_recommend_tag_list = json.loads(linucb_recommend_tag_data) if linucb_recommend_tag_data else []
         for item in linucb_recommend_tag_list:
-            results_tag_recommend = Tag.objects.filter(id=item, is_online=True).values_list("name", flat=True)
+            results_tag_recommend = list(
+                set(Tag.objects.filter(id=item, is_online=True).values_list("name", flat=True)))
             all_tag_name_list.add(results_tag_recommend[0])
             if len(all_tag_name_list) == 12:
                 return {"recommend_tag_name": list(all_tag_name_list)}
+
         # 取不够数则取核心标签
         if len(all_tag_name_list) < 12:
             for i in range(0, 12):
-                tag_id = random.sample(range(1, results_tag), 12 - len(all_tag_name_list))
-                results_tag_hexin = Tag.objects.filter(id__in=tag_id).values_list("name", flat=True)
-                if results_tag_hexin not in all_tag_name_list:
-                    all_tag_name_list.add(results_tag_hexin[0])
-                    if len(all_tag_name_list) >= 12:
-                        return {"recommend_tag_name": list(all_tag_name_list)}
+                tag_id = random.randint(1, len(results_tag))
+                results_tag_hexin = Tag.objects.filter(id=results_tag[tag_id], is_online=True,
+                                                       collection=1).values_list("name",
+                                                                                 flat=True)
+                if results_tag_hexin:
+                    if results_tag_hexin[0] not in all_tag_name_list:
+                        all_tag_name_list.add(results_tag_hexin[0])
+                        if len(all_tag_name_list) >= 12:
+                            return {"recommend_tag_name": list(all_tag_name_list)}
 
     except:
         logging.error("catch exception,err_msg:%s" % traceback.format_exc())