django-esi 5.3.0b1__tar.gz → 5.3.0b2__tar.gz

{django_esi-5.3.0b1 → django_esi-5.3.0b2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: django-esi
-Version: 5.3.0b1
+Version: 5.3.0b2
 Summary: Django app for accessing the EVE Swagger Interface (ESI).
 Author-email: Alliance Auth <adarnof@gmail.com>
 Requires-Python: >=3.8

{django_esi-5.3.0b1 → django_esi-5.3.0b2}/esi/__init__.py

@@ -2,5 +2,5 @@
 
 default_app_config = 'esi.apps.EsiConfig'
 
-__version__ = '5.3.0b1'
+__version__ = '5.3.0b2'
 __title__ = 'django-esi'

{django_esi-5.3.0b1 → django_esi-5.3.0b2}/esi/managers.py

@@ -1,18 +1,16 @@
-from datetime import timedelta
 import logging
-from typing import Union, Any
+from datetime import timedelta
+from typing import Any, Union
 
 import requests
-from requests_oauthlib import OAuth2Session
-
 from django.db import models
 from django.utils import timezone
-
-from .errors import TokenError, IncompleteResponseError
-from . import app_settings
-
 from jose import jwt
 from jose.exceptions import ExpiredSignatureError, JWTError
+from requests_oauthlib import OAuth2Session
+
+from . import app_settings
+from .errors import IncompleteResponseError, TokenError
 
 logger = logging.getLogger(__name__)
 
@@ -75,26 +73,12 @@ class TokenQueryset(models.QuerySet):
         Returns:
             All tokens which are still valid.
         """
-        expired = self.get_expired()
-        valid = list(
-            self.exclude(
-                pk__in=expired
-            ).values_list(
-                "pk",
-                flat=True
-            )
-        )
-        valid_expired = list(
-            expired.bulk_refresh(
-            ).values_list(
-                "pk",
-                flat=True
-            )
-        )
-        _qs = self.filter(
-            pk__in=(valid + valid_expired)
-        )
-        return _qs
+        expired_pks = set(self.get_expired().values_list("pk", flat=True))
+        fresh_pks = set(self.exclude(pk__in=expired_pks).values_list("pk", flat=True))
+        refreshed = self.filter(pk__in=expired_pks).bulk_refresh()
+        refreshed_pks = set(refreshed.values_list("pk", flat=True))
+        qs = self.filter(pk__in=fresh_pks | refreshed_pks)
+        return qs
 
     def require_scopes(self, scope_string: Union[str, list]) -> models.QuerySet:
         """Filter tokens which have at least a subset of given scopes.