modify

linucb/views/collect_data.py

@@ -99,9 +99,9 @@ class CollectData(object):
 
                                 tag_list = list()
                                 click_sql_query_results = TopicTag.objects.using(settings.SLAVE_DB_NAME).filter(topic_id=topic_id).values_list("tag_id","is_online")
-                                for sql_item in click_sql_query_results:
-                                    if sql_item.is_online:
-                                        tag_list.append(sql_item.tag_id)
+                                for tag_id,is_online in click_sql_query_results:
+                                    if is_online:
+                                        tag_list.append(tag_id)
 
                                 is_click = 1
                                 is_vote = 0
@@ -136,14 +136,14 @@ class CollectData(object):
 
                                 topic_tag_id_dict = dict()
                                 tag_list = list()
-                                exposure_sql_query_results = TopicTag.objects.using(settings.SLAVE_DB_NAME).filter(topic_id__in=exposure_topic_id_list).values_list("tag_id","is_online")
-                                for sql_item in exposure_sql_query_results:
-                                    if sql_item.is_online:
-                                        tag_list.append(sql_item.tag_id)
-
-                                    if sql_item.topic_id not in topic_tag_id_dict:
-                                        topic_tag_id_dict[sql_item.topic_id] = list()
-                                    topic_tag_id_dict[sql_item.topic_id].append(sql_item.tag_id)
+                                exposure_sql_query_results = TopicTag.objects.using(settings.SLAVE_DB_NAME).filter(topic_id__in=exposure_topic_id_list).values_list("topic_id","tag_id","is_online")
+                                for topic_id,tag_id,is_online in exposure_sql_query_results:
+                                    if is_online:
+                                        tag_list.append(tag_id)
+
+                                    if topic_id not in topic_tag_id_dict:
+                                        topic_tag_id_dict[topic_id] = list()
+                                    topic_tag_id_dict[topic_id].append(tag_id)
 
                                 is_click = 0
                                 is_vote = 0