Source code for dataprep.eda.container

    This module implements the Container class.

import random
import sys
import webbrowser
from tempfile import NamedTemporaryFile
from typing import Any, Dict

from bokeh.embed import components
from import output_notebook
from bokeh.resources import INLINE
from jinja2 import Environment, PackageLoader

from ..utils import is_notebook
from .configs import Config

output_notebook(INLINE, hide_banner=True)  # for offline usage

ENV_LOADER = Environment(
    loader=PackageLoader("dataprep", "eda/templates"),


GRID_VISUAL_TYPES = {"distribution_grid", "missing_impact_1vn", "comparison_grid"}

[docs]class Container: """ This class creates a customized Container object for the plot* function. """ def __init__(self, to_render: Dict[str, Any], visual_type: str, cfg: Config) -> None: self.context = Context(**to_render) setattr(self.context, "rnd", random.randint(0, 9999)) if visual_type in GRID_VISUAL_TYPES: self.template_base = ENV_LOADER.get_template("grid_base.html") elif visual_type in TAB_VISUAL_TYPES: if visual_type == "missing_impact": setattr(self.context, "highlight", False) else: setattr(self.context, "highlight", cfg.insight.enable) if to_render.get("tabledata"): self.context.meta.insert(0, "Stats") # type: ignore if to_render.get("value_table"): self.context.meta.append("Value Table") # type: ignore if visual_type == "correlation_impact": self.template_base = ENV_LOADER.get_template("tab_base_corr.html") else: self.template_base = ENV_LOADER.get_template("tab_base.html") else: raise TypeError(f"Unsupported Visual Type: {visual_type}.")
[docs] def save(self, filename: str) -> None: """ save function """ with open(filename, "w", encoding="utf-8") as f: f.write(self.template_base.render(context=self.context))
def _repr_html_(self) -> str: """ Display itself inside a notebook """ output_html = self.template_base.render(context=self.context) return output_html
[docs] def show(self) -> None: """ Render the report. This is useful when calling plot in a for loop. """ # if not called from notebook environment, ref to show_browser function. if not is_notebook(): print( "The plot will not show in a notebook environment, " "please try 'show_browser' if you want to open it in browser", file=sys.stderr, ) try: from IPython.display import ( # pylint: disable=import-outside-toplevel HTML, display, ) display(HTML(self._repr_html_())) except ImportError: pass
[docs] def show_browser(self) -> None: """ Open the plot in the browser. This is useful when plotting from terminmal or when the fig is very large in notebook. """ # set delete = False to avoid early delete when user open multiple plots. with NamedTemporaryFile(suffix=".html", delete=False) as tmpf: pass with open(, "w", encoding="utf-8") as file: file.write(self.template_base.render(context=self.context)) webbrowser.open_new_tab(f"file://{}")
[docs]class Context: """ Define the context class that stores all the parameters needed by template engine. The instance is read-only. Since we use same template to render different components without strict evaluation, when the engine tries to read an attribute from Context object, it will get None if the attribute doesn't exist, making the rendering keep going instead of being interrupted. Here we override __getitem__() and __getattr__() to do the trick, it also makes the object have a key-value pair which works as same as accessing its attribute. """ _title = "DataPrep.EDA Report" _resources = INLINE.render() _container_width = 650 # default width just in case nothing got passed in def __init__(self, **param: Any) -> None: self.title = self._title self.resources = self._resources self.container_width = self._container_width for attr, value in param.items(): if attr == "layout": if len(value) == 0: setattr(self, "components", ("", [])) else: setattr(self, "components", components(value)) elif attr == "meta": setattr(self, attr, value) figs = [val for val in value if val != "Stats"] setattr(self, "figs", figs) else: setattr(self, attr, value) def __getitem__(self, key: str) -> Any: if hasattr(self, key): return getattr(self, key) else: return None def __getattr__(self, attr: str) -> None: return None