code

controller/index.py

@@ -93,6 +93,7 @@ class loginHandleView(TemplateView):
             if not user_qs.exists():
                 user_qs = UserModel.objects.create(userID=userID, token=token, addTime=nowTime, updTime=nowTime)
                 userID = user_qs.userID
+                user_qs = UserModel.objects.filter(userID=userID)
             else:
                 userID = user_qs[0].userID
             querysetlist = []
@@ -111,6 +112,7 @@ class loginHandleView(TemplateView):
                 querysetlist.append(UidRtspModel(**create_data))
             UidRtspModel.objects.bulk_create(querysetlist)
             code = CommonService.encrypt_data(32)
+            user_qs.update(code=code)
             redirect_uri = redirect_uri + '?code=' + code + '&state=' + state
             return response.json(0, res=redirect_uri)
         else: