mirror of
https://github.com/clearml/clearml-server
synced 2025-01-31 19:06:55 +00:00
97 lines
2.7 KiB
Python
97 lines
2.7 KiB
Python
import jwt
|
|
|
|
from datetime import datetime, timedelta
|
|
|
|
from apiserver.apierrors import errors
|
|
from apiserver.config_repo import config
|
|
from apiserver.database.model.auth import Role
|
|
|
|
from .auth_type import AuthType
|
|
from .payload import Payload
|
|
|
|
token_secret = config.get('secure.auth.token_secret')
|
|
|
|
|
|
class Token(Payload):
|
|
default_expiration_sec = config.get('apiserver.auth.default_expiration_sec')
|
|
|
|
def __init__(self, exp=None, iat=None, nbf=None, env=None, identity=None, entities=None, **_):
|
|
super(Token, self).__init__(
|
|
AuthType.bearer_token, identity=identity, entities=entities)
|
|
self.exp = exp
|
|
self.iat = iat
|
|
self.nbf = nbf
|
|
self._env = env or config.get('env', '<unknown>')
|
|
|
|
@property
|
|
def env(self):
|
|
return self._env
|
|
|
|
@property
|
|
def exp(self):
|
|
return self._exp
|
|
|
|
@exp.setter
|
|
def exp(self, value):
|
|
self._exp = value
|
|
|
|
@property
|
|
def iat(self):
|
|
return self._iat
|
|
|
|
@iat.setter
|
|
def iat(self, value):
|
|
self._iat = value
|
|
|
|
@property
|
|
def nbf(self):
|
|
return self._nbf
|
|
|
|
@nbf.setter
|
|
def nbf(self, value):
|
|
self._nbf = value
|
|
|
|
def get_log_entry(self):
|
|
d = super(Token, self).get_log_entry()
|
|
d.update(iat=self.iat, exp=self.exp, env=self.env)
|
|
return d
|
|
|
|
def encode(self, **extra_payload):
|
|
payload = self.to_dict(**extra_payload)
|
|
return jwt.encode(payload, token_secret)
|
|
|
|
@classmethod
|
|
def decode(cls, encoded_token, verify=True):
|
|
return jwt.decode(encoded_token, token_secret, verify=verify)
|
|
|
|
@classmethod
|
|
def from_encoded_token(cls, encoded_token, verify=True):
|
|
decoded = cls.decode(encoded_token, verify=verify)
|
|
try:
|
|
token = Token.from_dict(decoded)
|
|
assert isinstance(token, Token)
|
|
if not token.identity:
|
|
raise errors.unauthorized.InvalidToken('token missing identity')
|
|
return token
|
|
except Exception as e:
|
|
raise errors.unauthorized.InvalidToken('failed parsing token, %s' % e.args[0])
|
|
|
|
@classmethod
|
|
def create_encoded_token(cls, identity, expiration_sec=None, entities=None, **extra_payload):
|
|
if identity.role not in (Role.system,):
|
|
# limit expiration time for all roles but an internal service
|
|
expiration_sec = expiration_sec or cls.default_expiration_sec
|
|
|
|
now = datetime.utcnow()
|
|
|
|
token = cls(
|
|
identity=identity,
|
|
entities=entities,
|
|
iat=now)
|
|
|
|
if expiration_sec:
|
|
# add 'expiration' claim
|
|
token.exp = now + timedelta(seconds=expiration_sec)
|
|
|
|
return token.encode(**extra_payload)
|