ClearML/clearml-server
1
0
Fork 0
mirror of https://github.com/clearml/clearml-server synced 2025-06-04 03:47:03 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Better support for PyJWT 2.4

Browse Source
This commit is contained in:
allegroai 2022-07-08 17:44:17 +03:00
parent 52c0c4d438
commit ee87778548
2 changed files with 11 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
apiserver/bll/event/events_iterator.py
View File

@ -4,6 +4,7 @@ import attr
import jsonmodels.models
import jwt
from elasticsearch import Elasticsearch
from jwt.algorithms import get_default_algorithms
from apiserver.bll.event.event_common import (
check_empty_data,
@ -77,10 +78,7 @@ class EventsIterator:
with translate_errors_context(), TimingContext("es", "count_task_events"):
es_result = count_company_events(
self.es,
company_id=company_id,
event_type=event_type,
body=es_req,
self.es, company_id=company_id, event_type=event_type, body=es_req,
)
return es_result["count"]
@ -117,10 +115,7 @@ class EventsIterator:
with translate_errors_context(), TimingContext("es", "get_task_events"):
es_result = search_company_events(
self.es,
company_id=company_id,
event_type=event_type,
body=es_req,
self.es, company_id=company_id, event_type=event_type, body=es_req,
)
hits = es_result["hits"]["hits"]
hits_total = es_result["hits"]["total"]["value"]
@ -140,10 +135,7 @@ class EventsIterator:
},
}
es_result = search_company_events(
self.es,
company_id=company_id,
event_type=event_type,
body=es_req,
self.es, company_id=company_id, event_type=event_type, body=es_req,
)
last_second_hits = es_result["hits"]["hits"]
if not last_second_hits or len(last_second_hits) < 2:
@ -199,7 +191,7 @@ class Scroll(jsonmodels.models.Base):
key=config.get(
"services.events.events_retrieval.scroll_id_key", "1234567890"
),
algorithms=["HS256"],
algorithms=get_default_algorithms(),
)
)
except jwt.PyJWTError:

8
apiserver/service_repo/auth/payload/token.py
View File

@ -2,6 +2,8 @@ import jwt
from datetime import datetime, timedelta
from jwt.algorithms import get_default_algorithms
from apiserver.apierrors import errors
from apiserver.config_repo import config
from apiserver.database.model.auth import Role
@ -11,7 +13,6 @@ from .payload import Payload
token_secret = config.get("secure.auth.token_secret")
log = config.logger(__file__)
@ -72,7 +73,10 @@ class Token(Payload):
{"verify_signature": False, "verify_exp": True} if not verify else None
)
return jwt.decode(
encoded_token, token_secret, algorithms=["HS256"], options=options
encoded_token,
token_secret,
algorithms=get_default_algorithms(),
options=options,
)
@classmethod