add

trans2es/utils/user_transfer.py

@@ -8,29 +8,33 @@ import time
 from libs.tools import tzlc
 from trans2es.models.user import User
 
+
 class UserTransfer(object):
 
     @classmethod
-    def get_follow_user_id_list(cls,userInstance):
+    def get_follow_user_id_list(cls, userInstance):
         follow_user_id_list = list()
         user_follows = userInstance.userfollow.filter(is_online=True)
         for user_follow in user_follows:
             follow_user_id_list.append(user_follow.follow_id)
 
         follow_user_detail_list = list()
-        sql_data_list = User.objects.filter(user_id__in=follow_user_id_list,is_online=True,is_deleted=False)
-        for detail_data in sql_data_list:
-            item = {
-                "user_id":detail_data.user_id,
-                "country_id":detail_data.country_id
-            }
-            follow_user_detail_list.append(item)
+        logging.info("get follow_user_id_list :%s "%follow_user_id_list)
+        for i in range(0, len(follow_user_id_list), 1000):
+            logging.info("get follow_user_id_list[i:i + 1000] " % follow_user_id_list[i:i+1000])
+            sql_data_list = User.objects.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 = {
+                    "user_id": detail_data.user_id,
+                    "country_id": detail_data.country_id
+                }
+                follow_user_detail_list.append(item)
 
         return follow_user_detail_list
 
-
     @classmethod
-    def get_user_data(cls,instance):
+    def get_user_data(cls, instance):
         try:
             res = dict()
 
@@ -80,10 +84,9 @@ class UserTransfer(object):
                 res["attention_user_id_list"] = []
                 res["attention_group_id_list"] = []
                 res["pick_user_id_list"] = []
-                res["same_group_user_id_list"] = []            
-
+                res["same_group_user_id_list"] = []
 
             return res
         except:
             logging.error("catch exception,err_msg:%s" % traceback.format_exc())
-            return {}
\ No newline at end of file
+            return {}