chenjunkai преди 5 години
родител
ревизия
c33ff7b203
променени са 1 файла, в които са добавени 2 реда и са изтрити 1 реда
  1. 2 1
      controller/index.py

+ 2 - 1
controller/index.py

@@ -90,10 +90,11 @@ class loginHandleView(TemplateView):
             token = tko.encrypt(data=userID)
             nowTime = int(time.time())
             user_qs = UserModel.objects.filter(userID=userID)
-            userID = user_qs[0].userID
             if not user_qs.exists():
                 user_qs = UserModel.objects.create(userID=userID, token=token, addTime=nowTime, updTime=nowTime)
                 userID = user_qs.userID
+            else:
+                userID = user_qs[0].userID
             querysetlist = []
             rtko = tkObject(rank=1)
             UidRtspModel.objects.filter(user_id=userID).delete()