can_send

message/views.py

@@ -12,7 +12,9 @@ from adapter.old_system import bind_prefix
 from adapter.rpcd.exceptions import RPCPermanentError
 from api.models.message import ConversationUserStatus
 from rpc import gaia_client
-from rpc.gaia_client import get_doctor_basic_info, get_single_doctor_basic_info
+from rpc.gaia_client import get_doctor_basic_info, \
+    get_single_doctor_basic_info, get_user_info_by_user_id, \
+    get_fans_relationship, is_kefu_user
 from rpc_framework import exceptions
 from rpc_framework.context import RPCViewContext
 from rpc_framework.decorators import rpc_view, interceptor_classes
@@ -243,26 +245,21 @@ def message_conversation_list_v3(user_ids: List[int],
 @interceptor_classes([CalleeParametersInterceptor, SessionUserInterceptor])
 def check_can_send_message(context: RPCViewContext, target_uid: str) -> Dict:
     user = context.user
-    # doctor_info = get_single_doctor_basic_info(user.id)
-    #
-    # target_user_id = doctor_info.get('user_id')
-    # print('target_user_id:', target_user_id)
+    doctor_info = get_single_doctor_basic_info(user.id)
 
-    # raise exceptions.NotFoundException
+    target_user_info = get_user_info_by_user_id(target_uid)
+    if not target_user_info:
+        raise exceptions.NotFoundException
 
-    # if not target_user_id:
-    #     raise 26872687
-    # target_doctor = get_doctor_by_user_id(target_user.id)
-    #
-    # if doctor or target_doctor:
-    #     return {'can_send': True}
-    #
-    # user_follow = user.fans.filter(follow=target_user, bond=True).exists()
-    # target_follow = target_user.fans.filter(follow=user, bond=True).exists()
-    #
-    # if user_follow and target_follow:
-    #     return {'can_send': True}
-    # if target_user.person.id.hex == settings.KEFU_PERSION_ID:
-    #     return {'can_send': True}
+    target_doctor_info = get_single_doctor_basic_info(target_uid)
+    if doctor_info or target_doctor_info:
+        return {'can_send': True}
+
+    user_follow, target_follow = get_fans_relationship(user.id, target_uid)
+    target_user_id = (doctor_info or {}).get('user_id')
+    if user_follow and target_follow:
+        return {'can_send': True}
+    if is_kefu_user(target_user_id):
+        return {'can_send': True}
 
     return {'can_send': False}

rpc/gaia_client.py

 # -*- coding: utf-8 -*-
-from typing import List, Dict, Optional, Set
+from typing import List, Dict, Optional, Set, Tuple
 
 from rpc import rpc_invoke_error_handle_decorator
 from rpc.context import get_rpc_remote_invoker
@@ -45,9 +45,23 @@ def get_doctor_basic_info(doctor_ids: List[str]) -> List[Dict]:
 
 
 def get_single_doctor_basic_info(doctor_id: str) -> Dict:
-    return get_rpc_remote_invoker()['api/doctor/basic_info'](
+    res = get_rpc_remote_invoker()['api/doctor/basic_info'](
         doctor_ids=[doctor_id]
-    ).unwrap()[0]
+    ).unwrap()
+    if res:
+        return res[0]
+
+def get_fans_relationship(user_id1:str, user_id2:str) -> Tuple[bool, bool]:
+    # response:
+    # value1: user2 是否 user1 的粉丝
+    # value2: user1 是否 user2 的粉丝
+    try:
+        return get_rpc_remote_invoker()['api/get_fans_relationship'](
+            user_id1=user_id1,
+            user_id2=user_id2
+        ).unwrap()
+    except:
+        return [False, False]
 
 
 def batch_get_doctor_info_by_user_ids(user_ids: List[int]) -> Dict:
@@ -55,3 +69,15 @@ def batch_get_doctor_info_by_user_ids(user_ids: List[int]) -> Dict:
         user_ids=user_ids,
         with_fields=['meta']
     ).unwrap()
+
+def get_user_info_by_user_id(user_id):
+    return get_rpc_remote_invoker()['hera/user/getinfoby_userid'](user_id=user_id).unwrap()
+
+
+def is_kefu_user(user_id):
+    try:
+        get_rpc_remote_invoker()['api/is_kefu_user'](user_id=user_id).unwrap()
+    except:
+        return False
+
+

rpc_framework/exceptions.py

@@ -55,7 +55,7 @@ class AuthenticationFailed(RPCViewBaseException):
     message = Error.getDesc(Error.AUTH_FAILED)
 
 class NotFoundException(RPCViewBaseException):
-    code = 1404
+    code = 1404 # gaia RPCNotFoundException
     message = "Not Found"
     # default_message = "Not Found"
 

rpc_framework/views.py

 # -*- coding: utf-8 -*-
+import traceback
+
 from django.db import connection, transaction
 
 from rpc_framework import exceptions
@@ -19,7 +21,8 @@ def exception_handler(exc, wrapped_context) -> None:
         raise exc
     elif issubclass(exc.__class__, exceptions.RPCViewBaseException):
         raise exc.as_rpc_view_exception()
-    return None
+    elif isinstance(exc, Exception):
+        print(exc, exc.args[0] if exc.args else '', traceback.format_exc())
 
 
 class RPCView(RPCAbstractView):