1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
from __future__ import annotations
import importlib
import os
import toolz
from tornado import web
from distributed.utils import has_keyword
dirname = os.path.dirname(__file__)
class RequestHandler(web.RequestHandler):
def initialize(self, dask_server=None, extra=None):
self.server = dask_server
self.extra = extra or {}
def get_template_path(self):
return os.path.join(dirname, "templates")
def redirect(path):
class Redirect(RequestHandler):
def get(self):
self.redirect(path)
return Redirect
def get_handlers(server, modules: list[str], prefix="/"): # type: ignore[no-untyped-def]
prefix = prefix or ""
prefix = "/" + prefix.strip("/")
if not prefix.endswith("/"):
prefix = prefix + "/"
_routes = []
for module_name in modules:
module = importlib.import_module(module_name)
_routes.extend(module.routes)
routes = []
for url, cls, kwargs in _routes:
if has_keyword(cls.initialize, "dask_server"):
kwargs = toolz.assoc(kwargs, "dask_server", server)
routes.append((prefix + url.lstrip("/"), cls, kwargs))
return routes
|
