modify

trans2es/utils/tag_transfer.py

@@ -10,6 +10,9 @@ from libs.tools import tzlc
 from trans2es.models.topic import Topic
 from trans2es.models.tag import TopicTag,CommunityTagType,CommunityTagTypeRelation
 import datetime
+from django.conf import settings
+
+
 
 class TagTransfer(object):
 
@@ -38,20 +41,24 @@ class TagTransfer(object):
             res["is_online"] = instance.is_online
             res["is_deleted"] = instance.is_deleted
 
-            res["near_new_topic_num"] = 0
+            topic_num = 0
+            res["near_new_topic_num"] = topic_num
             if instance.is_online==True and instance.is_deleted==False:
                 topic_id_list = list()
 
-                sql_result_list = TopicTag.objects.filter(tag_id=instance.id).values_list("topic_id",flat=True)
+                sql_result_list = TopicTag.objects.using(settings.SLAVE_DB_NAME).filter(tag_id=instance.id,is_online=True).values_list("topic_id",flat=True)
                 for item_id in sql_result_list:
                     topic_id_list.append(item_id)
 
                 time_base_val = datetime.datetime.strftime(datetime.datetime.now()+datetime.timedelta(-7), "%Y-%m-%d")
-                topic_num = Topic.objects.filter(id__in=topic_id_list,create_time__gte=time_base_val).count()
+
+                for topic_begin_index in range(0,len(topic_id_list),100):
+                    cur_topic_num = Topic.objects.using(settings.SLAVE_DB_NAME).filter(id__in=topic_id_list[topic_begin_index:topic_begin_index+100],create_time__gte=time_base_val).count()
+                    topic_num += cur_topic_num
 
                 res["near_new_topic_num"] = topic_num
 
-            tag_type_sql_list = CommunityTagTypeRelation.objects.filter(tag_id=instance.id).values_list("tag_type_id",flat=True)
+            tag_type_sql_list = CommunityTagTypeRelation.objects.using(settings.SLAVE_DB_NAME).filter(tag_id=instance.id).values_list("tag_type_id",flat=True)
             tag_type_list = list()
             for tag_type_id in tag_type_sql_list:
                 tag_type_list.append(tag_type_id)

trans2es/utils/user_transfer.py

@@ -7,6 +7,7 @@ import traceback
 import time
 from libs.tools import tzlc
 from trans2es.models.user import User
+from django.conf import settings
 
 
 class UserTransfer(object):
@@ -20,7 +21,7 @@ class UserTransfer(object):
 
         follow_user_detail_list = list()
         for i in range(0, len(follow_user_id_list), 1000):
-            sql_data_list = User.objects.filter(user_id__in=follow_user_id_list[i:i + 1000], is_online=True,
+            sql_data_list = User.objects.using(settings.SLAVE_DB_NAME).filter(user_id__in=follow_user_id_list[i:i + 1000], is_online=True,
                                                 is_deleted=False)
             for detail_data in sql_data_list:
                 item = {