update

linucb/views/collect_data.py

@@ -7,7 +7,7 @@ from libs.cache import redis_client
 import logging
 from linucb.views.linucb import LinUCB
 import json
-from trans2es.models.tag import TopicTag
+from trans2es.models.tag import TopicTag,Tag
 import traceback
 from django.conf import settings
 from libs.es import ESPerform
@@ -138,7 +138,10 @@ class CollectData(object):
                                 click_sql_query_results = TopicTag.objects.using(settings.SLAVE_DB_NAME).filter(topic_id=topic_id).values_list("tag_id","is_online")
                                 for tag_id,is_online in click_sql_query_results:
                                     if is_online:
-                                        tag_list.append(tag_id)
+                                      tag_sql_query_results = Tag.objects.using(settings.SLAVE_DB_NAME).filter(id=tag_id).values_list("id","collection","is_ai")
+                                      for id,collection,is_ai in tag_sql_query_results:
+                                         if collection == 1 or is_ai == 1:
+                                            tag_list.append(id)
 
                                 is_click = 1
                                 is_vote = 0