ClearML/clearml-server
1
0
Fork 0
mirror of https://github.com/clearml/clearml-server synced 2025-02-14 16:58:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add support for ClearML headers

Browse Source
This commit is contained in:
allegroai 2021-01-05 18:29:50 +02:00
parent 3c8e27dc94
commit 23736efbc3
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
apiserver/service_repo/apicall.py
View File

@ -299,41 +299,47 @@ class MissingIdentity(Exception):
pass pass
def _get_headers(name: str) -> Tuple[str, ...]:
return tuple(
"-".join(("X", p, name)) for p in ("ClearML", "Trains")
)
class APICall(DataContainer): class APICall(DataContainer):
HEADER_AUTHORIZATION = "Authorization" HEADER_AUTHORIZATION = "Authorization"
HEADER_REAL_IP = "X-Real-IP" HEADER_REAL_IP = "X-Real-IP"
HEADER_FORWARDED_FOR = "X-Forwarded-For" HEADER_FORWARDED_FOR = "X-Forwarded-For"
""" Standard headers """ """ Standard headers """
_transaction_headers = ("X-Trains-Trx",) _transaction_headers = _get_headers("Trx")
""" Transaction ID """ """ Transaction ID """
@property @property
def HEADER_TRANSACTION(self): def HEADER_TRANSACTION(self):
return self._transaction_headers[0] return self._transaction_headers[0]
_worker_headers = ("X-Trains-Worker",) _worker_headers = _get_headers("Worker")
""" Worker (machine) ID """ """ Worker (machine) ID """
@property @property
def HEADER_WORKER(self): def HEADER_WORKER(self):
return self._worker_headers[0] return self._worker_headers[0]
_impersonate_as_headers = ("X-Trains-Impersonate-As",) _impersonate_as_headers = _get_headers("Impersonate-As")
""" Impersonate as someone else (using his identity and permissions) """ """ Impersonate as someone else (using his identity and permissions) """
@property @property
def HEADER_IMPERSONATE_AS(self): def HEADER_IMPERSONATE_AS(self):
return self._impersonate_as_headers[0] return self._impersonate_as_headers[0]
_act_as_headers = ("X-Trains-Act-As",) _act_as_headers = _get_headers("Act-As")
""" Act as someone else (using his identity, but with your own role and permissions) """ """ Act as someone else (using his identity, but with your own role and permissions) """
@property @property
def HEADER_ACT_AS(self): def HEADER_ACT_AS(self):
return self._act_as_headers[0] return self._act_as_headers[0]
_async_headers = ("X-Trains-Async",) _async_headers = _get_headers("Async")
""" Specifies that this call should be done asynchronously """ """ Specifies that this call should be done asynchronously """
@property @property