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
52
53
54
55
56
|
from __future__ import annotations
import functools
import warnings
from bokeh.application import Application
from bokeh.application.handlers.function import FunctionHandler
from bokeh.server.server import BokehTornado
from bokeh.server.util import create_hosts_allowlist
from packaging.version import parse as parse_version
import dask
from distributed.dashboard.utils import BOKEH_VERSION
from distributed.versions import MAX_BOKEH_VERSION, MIN_BOKEH_VERSION
if BOKEH_VERSION < parse_version(MIN_BOKEH_VERSION) or BOKEH_VERSION > parse_version(
MAX_BOKEH_VERSION
):
warnings.warn(
f"\nDask needs bokeh >= {MIN_BOKEH_VERSION}, < 3 for the dashboard."
f"\nYou have bokeh=={BOKEH_VERSION}."
"\nContinuing without the dashboard."
)
raise ImportError(
f"Dask needs bokeh >= {MIN_BOKEH_VERSION}, < 3, not bokeh=={BOKEH_VERSION}"
)
if BOKEH_VERSION.major < 3:
from bokeh.models import Panel as TabPanel # noqa: F401
else:
from bokeh.models import TabPanel # noqa: F401
def BokehApplication(applications, server, prefix="/", template_variables=None):
template_variables = template_variables or {}
prefix = "/" + prefix.strip("/") + "/" if prefix else "/"
extra = {"prefix": prefix, **template_variables}
funcs = {k: functools.partial(v, server, extra) for k, v in applications.items()}
apps = {k: Application(FunctionHandler(v)) for k, v in funcs.items()}
kwargs = dask.config.get("distributed.scheduler.dashboard.bokeh-application").copy()
extra_websocket_origins = create_hosts_allowlist(
kwargs.pop("allow_websocket_origin"), server.http_server.port
)
return BokehTornado(
apps,
prefix=prefix,
use_index=False,
extra_websocket_origins=extra_websocket_origins,
**kwargs,
)
|
