api/encrypt_views.py

@@ -3,6 +3,7 @@

                 from __future__ import division
              

                 
              

                 import random
              

                +import re
              

                 
              

                 from django.conf import settings
              

                 from django.db import transaction
              
@@ -204,18 +205,20 @@ def decrypt2(request):

                     code = request.POST.get('code', '')
              

                     user_id = request.POST.get('user_id', '')
              

                 
              

                -    try:
              

                -        user = UserInfo.objects.get(user_id=user_id)
              

                -    except UserInfo.DoesNotExist:
              

                -        return response(UserStatusCode.USER_NOT_FOUND)
              

                +    if code_ticket and user_id:
              

                +        code_ticket = re.sub(r'http://|https://', '', code_ticket)
              

                +        try:
              

                +            user = UserInfo.objects.get(user_id=user_id)
              

                +        except UserInfo.DoesNotExist:
              

                +            return response(UserStatusCode.USER_NOT_FOUND)
              

                 
              

                -    wxcfg = WECHAT.get('JSAPI', {})
              

                +        wxcfg = WECHAT.get('JSAPI', {})
              

                 
              

                -    appid = wxcfg.get('appID')
              

                -    secret = wxcfg.get('appsecret')
              

                +        appid = wxcfg.get('appID')
              

                +        secret = wxcfg.get('appsecret')
              

                 
              

                -    if code_ticket:
              

                -        code_info = tickettocode(code_ticket=code_ticket, openid=user.openid_miniapp, appid=appid, secret=secret, token=None, storage=RedisStorage(r))
              

                +        code_info = tickettocode(code_ticket=code_ticket, openid=user.openid_miniapp, appid=appid, secret=secret,
              

                +                                 token=None, storage=RedisStorage(r))
              

                 
              

                         code = code_info.get('code', '')
              

                 
              
@@ -228,7 +231,6 @@ def decrypt2(request):

                             mieli = MchInfoEncryptLogInfo.objects.get(code_url=code)
              

                         except MchInfoEncryptLogInfo.DoesNotExist:
              

                             return response()
              

                -    
              

                 
              

                     plaintext = mieli.plaintext