def test_filters():
template = get_template("bytes.html.j2")
assert format_bytes in FILTERS.values()
assert format_bytes(2e9) in template.render(foo=2e9)
template = get_template("custom_filter.html.j2")
assert "baz" in template.render(foo=None)
def test_unknown_template():
with pytest.raises(jinja2.TemplateNotFound) as e:
get_template("does_not_exist.html.j2")
# The error should contain all the registered template directories to help the user
# understand where jinja2 is looking. Including the one we registered in the fixture.
assert os.path.dirname(os.path.abspath(__file__)) in str(e)
def _repr_html_(self, layer_index="", highlevelgraph_key=""):
if highlevelgraph_key != "":
shortname = key_split(highlevelgraph_key)
elif hasattr(self, "name"):
shortname = key_split(self.name)
else:
shortname = self.__class__.__name__
svg_repr = ""
if (self.collection_annotations
and self.collection_annotations.get("type")
== "dask.array.core.Array"):
chunks = self.collection_annotations.get("chunks")
if chunks:
from dask.array.svg import svg
svg_repr = svg(chunks)
return get_template("highlevelgraph_layer.html.j2").render(
materialized=self.is_materialized(),
shortname=shortname,
layer_index=layer_index,
highlevelgraph_key=highlevelgraph_key,
info=self.layer_info_dict(),
svg_repr=svg_repr,
)
def _repr_html_(self):
return get_template("highlevelgraph.html.j2").render(
type=type(self).__name__,
layers=self.layers,
toposort=self._toposort_layers(),
n_outputs=len(self.get_all_external_keys()),
)
def _repr_html_(self, cluster_status=None):
cluster_status = get_template("local_cluster.html.j2").render(
status=self.status.name,
processes=self.processes,
cluster_status=cluster_status,
)
return super()._repr_html_(cluster_status=cluster_status)
def get_snippet(name: str) -> str:
"""Get a code snippet for connecting to a cluster.
Parameters
----------
name
Name of cluster to get a snippet for.
Returns
-------
str
Code snippet.
Examples
--------
>>> from dask.distributed import LocalCluster # doctest: +SKIP
>>> cluster = LocalCluster(scheduler_port=8786) # doctest: +SKIP
>>> get_snippet("proxycluster-8786") # doctest: +SKIP
from dask.distributed import Client
from dask_ctl.proxy import ProxyCluster
cluster = ProxyCluster.from_name("proxycluster-8786")
client = Client(cluster)
"""
cluster = get_cluster(name)
try:
return cluster.get_snippet()
except AttributeError:
*module, cm = typename(type(cluster)).split(".")
module = ".".join(module)
return get_template("snippet.py.j2").render(
module=module, cm=cm, name=name, cluster=cluster
)
def _repr_html_(self, cluster_status=None):
try:
scheduler_info_repr = self.scheduler_info._repr_html_()
except AttributeError:
scheduler_info_repr = "Scheduler not started yet."
return get_template("cluster.html.j2").render(
type=type(self).__name__,
name=self.name,
workers=self.scheduler_info["workers"],
dashboard_link=self.dashboard_link,
scheduler_info_repr=scheduler_info_repr,
cluster_status=cluster_status,
)
def _repr_html_(self):
return get_template("process_interface.html.j2").render(
process_interface=self)
def _repr_html_(self):
return get_template("logs.html.j2").render(logs=self)
def _repr_html_(self):
return get_template("security.html.j2").render(security=self._attr_to_dict())
def _repr_html_(self):
return get_template("who_has.html.j2").render(who_has=self)
def _repr_html_(self):
return get_template("has_what.html.j2").render(has_what=self)
def test_widgets():
template = get_template("example.html.j2")
assert isinstance(template, jinja2.Template)
rendered = template.render(foo="bar")
assert "Hello bar" in rendered
