def main(argv=None):
port = random_port()
hub_auth = HubAuth()
hub_auth.client_ca = os.environ.get('JUPYTERHUB_SSL_CLIENT_CA', '')
hub_auth.certfile = os.environ.get('JUPYTERHUB_SSL_CERTFILE', '')
hub_auth.keyfile = os.environ.get('JUPYTERHUB_SSL_KEYFILE', '')
hub_auth._api_request(method='POST',
url=url_path_join(hub_auth.api_url, 'batchspawner'),
json={'port': port})
cmd_path = which(sys.argv[1])
sys.argv = sys.argv[1:] + ['--port={}'.format(port)]
run_path(cmd_path, run_name="__main__")
def get_access_token():
# Helps getting the correct ssl configs
hub = HubAuth()
response = requests.get(
os.environ['JUPYTERHUB_HANDLER_CUSTOM_AUTH_URL'],
headers={'Authorization': 'token %s' % hub.api_token},
cert=(hub.certfile, hub.keyfile),
verify=hub.client_ca,
allow_redirects=False)
if response.status_code == 200:
return response.json()['access_token']
else:
raise AuthError
def update_tokens():
# Helps getting the correct ssl configs
hub = HubAuth()
response = requests.get(
os.environ['JUPYTERHUB_HANDLER_CUSTOM_AUTH_URL'],
headers={'Authorization': 'token %s' % hub.api_token},
cert=(hub.certfile, hub.keyfile),
verify=hub.client_ca,
allow_redirects=False)
if response.status_code == 200:
SparkContext._active_spark_context._conf.set(
"spark.ssb.access",
response.json()['access_token'])
else:
raise AuthError
def main(argv=None):
port = random_port()
hub_auth = HubAuth()
hub_auth.client_ca = os.environ.get("JUPYTERHUB_SSL_CLIENT_CA", "")
hub_auth.certfile = os.environ.get("JUPYTERHUB_SSL_CERTFILE", "")
hub_auth.keyfile = os.environ.get("JUPYTERHUB_SSL_KEYFILE", "")
hub_auth._api_request(
method="POST",
url=url_path_join(hub_auth.api_url, "batchspawner"),
json={"port": port},
)
cmd_path = which(sys.argv[1])
sys.argv = sys.argv[1:] + ["--port={}".format(port)]
run_path(cmd_path, run_name="__main__")
def main():
# FIXME: remove when we can declare routes in Hub config
add_route()
hub_auth = HubAuth(
cookie_name='jupyter-hub-token',
api_token=os.environ['WHOAMI_HUB_API_TOKEN'],
login_url='http://127.0.0.1:8000/hub/login',
)
app = Application([
(r"/hub/whoami", WhoAmIHandler, dict(hub_auth=hub_auth)),
], login_url=hub_auth.login_url)
http_server = HTTPServer(app)
http_server.listen(9999)
IOLoop.current().start()
whoami service authentication with the Hub
"""
from functools import wraps
import json
import os
from urllib.parse import quote
from flask import Flask, redirect, request, Response
from jupyterhub.services.auth import HubAuth
prefix = os.environ.get('JUPYTERHUB_SERVICE_PREFIX', '/')
auth = HubAuth(
api_token=os.environ['JUPYTERHUB_API_TOKEN'],
cache_max_age=60,
)
app = Flask(__name__)
def authenticated(f):
"""Decorator for authenticating with the Hub"""
@wraps(f)
def decorated(*args, **kwargs):
cookie = request.cookies.get(auth.cookie_name)
if cookie:
user = auth.user_for_cookie(cookie)
else:
user = None
if user:
def init_hub_auth(self):
"""Initialize hub authentication"""
self.hub_auth = HubAuth()