ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 18:52:12 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Get notebook auth token from the JUPYTERHUB_API_TOKEN environment

Browse Source 
variable if it is not present in the server info
This commit is contained in:
Michael Allman 2021-05-24 10:05:11 -07:00 committed by Jake Henning
parent 0325c2f32a
commit fa6b2e8c7c
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
clearml/backend_interface/task/repo/scriptinfo.py
View File

@ -543,10 +543,11 @@ class ScriptInfo(object):
data={'_xsrf': cookies['_xsrf'], 'password': password}) data={'_xsrf': cookies['_xsrf'], 'password': password})
cookies.update(r.cookies) cookies.update(r.cookies)
auth_token = server_info.get('token') or os.getenv('JUPYTERHUB_API_TOKEN') or ''
try: try:
r = requests.get( r = requests.get(
url=server_info['url'] + 'api/sessions', cookies=cookies, url=server_info['url'] + 'api/sessions', cookies=cookies,
headers={'Authorization': 'token {}'.format(server_info.get('token', '')), }) headers={'Authorization': 'token {}'.format(auth_token), })
except requests.exceptions.SSLError: except requests.exceptions.SSLError:
# disable SSL check warning # disable SSL check warning
from urllib3.exceptions import InsecureRequestWarning from urllib3.exceptions import InsecureRequestWarning
@ -555,7 +556,7 @@ class ScriptInfo(object):
# fire request # fire request
r = requests.get( r = requests.get(
url=server_info['url'] + 'api/sessions', cookies=cookies, url=server_info['url'] + 'api/sessions', cookies=cookies,
headers={'Authorization': 'token {}'.format(server_info.get('token', '')), }, verify=False) headers={'Authorization': 'token {}'.format(auth_token), }, verify=False)
# enable SSL check warning # enable SSL check warning
import warnings import warnings
warnings.simplefilter('default', InsecureRequestWarning) warnings.simplefilter('default', InsecureRequestWarning)