ClearML/clearml-server
1
0
Fork 0
mirror of https://github.com/clearml/clearml-server synced 2025-02-07 05:27:23 +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
def _get_headers(name: str) -> Tuple[str, ...]:
return tuple(
"-".join(("X", p, name)) for p in ("ClearML", "Trains")
)
class APICall(DataContainer):
HEADER_AUTHORIZATION = "Authorization"
HEADER_REAL_IP = "X-Real-IP"
HEADER_FORWARDED_FOR = "X-Forwarded-For"
""" Standard headers """
_transaction_headers = ("X-Trains-Trx",)
_transaction_headers = _get_headers("Trx")
""" Transaction ID """
@property
def HEADER_TRANSACTION(self):
return self._transaction_headers[0]
_worker_headers = ("X-Trains-Worker",)
_worker_headers = _get_headers("Worker")
""" Worker (machine) ID """
@property
def HEADER_WORKER(self):
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) """
@property
def HEADER_IMPERSONATE_AS(self):
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) """
@property
def HEADER_ACT_AS(self):
return self._act_as_headers[0]
_async_headers = ("X-Trains-Async",)
_async_headers = _get_headers("Async")
""" Specifies that this call should be done asynchronously """
@property