Bläddra i källkod

增加用户id

peng 11 månader sedan
förälder
incheckning
2cbb6b3709
2 ändrade filer med 4 tillägg och 3 borttagningar
  1. 1 0
      Model/models.py
  2. 3 3
      Object/TokenObject.py

+ 1 - 0
Model/models.py

@@ -5285,6 +5285,7 @@ class CustomCustomerDevice(models.Model):
 
 class TokenRecord(models.Model):
     id = models.AutoField(primary_key=True, verbose_name='自增标记ID')
+    user_id = models.CharField(default='', max_length=32, db_index=True, verbose_name='用户id')
     data = models.JSONField(verbose_name=u'加密信息', null=True)
     access_token = models.TextField(default='', verbose_name='请求token')
     refresh_token = models.TextField(default='', verbose_name='刷新token')

+ 3 - 3
Object/TokenObject.py

@@ -130,15 +130,15 @@ class TokenObject:
                     LOGGER.info('{}登出推送结果:{}'.format(data['userID'], result.json()))
             now_time = int(time.time())
             if old_refresh_token:
-                token_qs = TokenRecord.objects.filter(refresh_token=old_refresh_token)
+                token_qs = TokenRecord.objects.filter(refresh_token=old_refresh_token, user_id=data['userID'])
                 if token_qs.exists():
                     token_qs.update(refresh_token=refresh_token, access_token=access_token, update_time=now_time)
                 else:
                     TokenRecord.objects.create(access_token=access_token, refresh_token=refresh_token, data=data,
-                                               created_time=now_time, updated_time=now_time)
+                                               created_time=now_time, updated_time=now_time, user_id=data['userID'])
             else:
                 TokenRecord.objects.create(access_token=access_token, refresh_token=refresh_token, data=data,
-                                           created_time=now_time, updated_time=now_time)
+                                           created_time=now_time, updated_time=now_time, user_id=data['userID'])
         except Exception as e:
             self.code = 309
             self.update_token(repr(e))