modify

trans2es/utils/user_transfer.py

@@ -6,9 +6,29 @@ import logging
 import traceback
 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):
+        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)
+
+        return follow_user_detail_list
+
+
     @classmethod
     def get_user_data(cls,instance):
         try:
@@ -52,7 +72,7 @@ class UserTransfer(object):
 
             try:
                 res["tag_list"] = instance.get_user_tag_id_list()
-                res["attention_user_id_list"] = instance.get_follow_user_id_list()
+                res["attention_user_id_list"] = cls.get_follow_user_id_list(userInstance=instance)
                 res["attention_group_id_list"] = []
                 res["pick_user_id_list"] = []
                 res["same_group_user_id_list"] = []