mirror of
https://github.com/jupyterhub/jupyterhub.git
synced 2025-10-08 02:24:08 +00:00
7a268c94b0
- `@gen.coroutine def` -> `async def` - `yield future` -> `await future` needs some fine tuning, but this is the big one
64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
"""Proxy handlers"""
|
|
|
|
# Copyright (c) Jupyter Development Team.
|
|
# Distributed under the terms of the Modified BSD License.
|
|
|
|
import json
|
|
from urllib.parse import urlparse
|
|
|
|
from tornado import gen, web
|
|
|
|
from .. import orm
|
|
from ..utils import admin_only
|
|
from .base import APIHandler
|
|
|
|
|
|
class ProxyAPIHandler(APIHandler):
|
|
|
|
@admin_only
|
|
async def get(self):
|
|
"""GET /api/proxy fetches the routing table
|
|
|
|
This is the same as fetching the routing table directly from the proxy,
|
|
but without clients needing to maintain separate
|
|
"""
|
|
routes = await self.proxy.get_all_routes()
|
|
self.write(json.dumps(routes))
|
|
|
|
@admin_only
|
|
async def post(self):
|
|
"""POST checks the proxy to ensure that it's up to date.
|
|
|
|
Can be used to jumpstart a newly launched proxy
|
|
without waiting for the check_routes interval.
|
|
"""
|
|
await self.proxy.check_routes(self.users, self.services)
|
|
|
|
@admin_only
|
|
async def patch(self):
|
|
"""PATCH updates the location of the proxy
|
|
|
|
Can be used to notify the Hub that a new proxy is in charge
|
|
"""
|
|
if not self.request.body:
|
|
raise web.HTTPError(400, "need JSON body")
|
|
|
|
try:
|
|
model = json.loads(self.request.body.decode('utf8', 'replace'))
|
|
except ValueError:
|
|
raise web.HTTPError(400, "Request body must be JSON dict")
|
|
if not isinstance(model, dict):
|
|
raise web.HTTPError(400, "Request body must be JSON dict")
|
|
|
|
if 'api_url' in model:
|
|
self.proxy.api_url = model['api_url']
|
|
if 'auth_token' in model:
|
|
self.proxy.auth_token = model['auth_token']
|
|
self.log.info("Updated proxy at %s", self.proxy)
|
|
await self.proxy.check_routes(self.users, self.services)
|
|
|
|
|
|
default_handlers = [
|
|
(r"/api/proxy", ProxyAPIHandler),
|
|
]
|