Merge 2bf6abd into 3ce7b1a

great_expectations/jupyter_ux/__init__.py

@@ -1,13 +1,21 @@
+"""Utility functions for working with great_expectations within jupyter notebooks or jupyter lab.
+"""
+
 import json
 import os
 import logging
 import great_expectations as ge
+import great_expectations.render as render
 from datetime import datetime
 
 import tzlocal
 from IPython.core.display import display, HTML
 
 def set_data_source(context, data_source_type=None):
+    """
+    TODO: Needs a docstring and tests.
+    """
+
     data_source_name = None
 
     if not data_source_type:
@@ -67,6 +75,10 @@ def set_data_source(context, data_source_type=None):
     return data_source_name
 
 def setup_notebook_logging(logger=None):
+    """
+    TODO: Needs a docstring and tests.
+    """
+
     def posix2local(timestamp, tz=tzlocal.get_localzone()):
         """Seconds since the epoch -> local time as an aware datetime object."""
         return datetime.fromtimestamp(timestamp, tz)
@@ -99,6 +111,10 @@ def formatTime(self, record, datefmt=None):
     warnings.filterwarnings('ignore')
 
 def list_available_data_asset_names(context, data_source_name=None):
+    """
+    TODO: Needs a docstring and tests.
+    """
+
     datasources = context.list_datasources()
     for datasource in datasources:
         if data_source_name and datasource['name'] != data_source_name:
@@ -125,4 +141,134 @@ def list_available_data_asset_names(context, data_source_name=None):
                 </p>
                             """))
 
-    #TODO: add expectation suite names (existing)
+    #TODO: add expectation suite names (existing)
+
+bootstrap_link_element = """<link rel="stylesheet" href="https://stackpath.bootstrapcdn.com/bootstrap/4.3.1/css/bootstrap.min.css" integrity="sha384-ggOyR0iXCbMQv3Xipma34MD+dH/1fQ784/j6cY/iJTQUOhcWr7x9JvoRxT2MZw1T" crossorigin="anonymous">"""
+cooltip_style_element = """<style type="text/css">
+.cooltip {
+    display:inline-block;
+    position:relative;
+    text-align:left;
+}
+
+.cooltip .top {
+    min-width:200px; 
+    top:-6px;
+    left:50%;
+    transform:translate(-50%, -100%);
+    padding:10px 20px;
+    color:#FFFFFF;
+    background-color:#222222;
+    font-weight:normal;
+    font-size:13px;
+    border-radius:8px;
+    position:absolute;
+    z-index:99999999;
+    box-sizing:border-box;
+    box-shadow:0 1px 8px rgba(0,0,0,0.5);
+    display:none;
+}
+
+.cooltip:hover .top {
+    display:block;
+}
+
+.cooltip .top i {
+    position:absolute;
+    top:100%;
+    left:50%;
+    margin-left:-12px;
+    width:24px;
+    height:12px;
+    overflow:hidden;
+}
+
+.cooltip .top i::after {
+    content:'';
+    position:absolute;
+    width:12px;
+    height:12px;
+    left:50%;
+    transform:translate(-50%,-50%) rotate(45deg);
+    background-color:#222222;
+    box-shadow:0 1px 8px rgba(0,0,0,0.5);
+}
+</style>
+"""
+
+def display_column_expectations_as_section(
+    expectation_suite,
+    column,
+    section_renderer=render.renderer.column_section_renderer.PrescriptiveColumnSectionRenderer,
+    view_renderer=render.view.view.DefaultJinjaSectionView,
+    include_styling=True,
+    return_without_displaying=False,
+):
+    """This is a utility function to render all of the Expectations in an ExpectationSuite with the same column name as an HTML block.
+
+    By default, the HTML block is rendered using PrescriptiveColumnSectionRenderer and the view is rendered using DefaultJinjaSectionView.
+    Therefore, it should look exactly the same as the default renderer for build_docs. 
+
+    Example usage:
+    exp = context.get_expectation_suite("notable_works_by_charles_dickens", "BasicDatasetProfiler")
+    display_column_expectations_as_section(exp, "Type")
+    """
+
+    #TODO: replace this with a generic utility function, preferably a method on an ExpectationSuite class
+    column_expectation_list = [ e for e in expectation_suite["expectations"] if "column" in e["kwargs"] and e["kwargs"]["column"] == column ]
+
+    #TODO: Handle the case where zero evrs match the column name
+
+    document = render.renderer.PrescriptiveColumnSectionRenderer.render(column_expectation_list)
+    view = render.view.DefaultJinjaSectionView.render({
+        "section": document,
+        "section_loop": {"index": 1},
+    })
+
+    if include_styling:
+        html_to_display = bootstrap_link_element+cooltip_style_element+view
+    else:
+        html_to_display = view
+
+    if return_without_displaying:
+        return html_to_display
+    else:
+        display(HTML(html_to_display))
+
+def display_column_evrs_as_section(
+    evrs,
+    column,
+    section_renderer=render.renderer.column_section_renderer.DescriptiveColumnSectionRenderer,
+    view_renderer=render.view.view.DefaultJinjaSectionView,
+    include_styling=True,
+    return_without_displaying=False,
+):
+    """This is a utility function to render all of the EVRs in an ExpectationSuite with the same column name as an HTML block.
+
+    By default, the HTML block is rendered using PrescriptiveColumnSectionRenderer and the view is rendered using DefaultJinjaSectionView.
+    Therefore, it should look exactly the same as the default renderer for build_docs. 
+
+    Example usage:
+    display_column_evrs_as_section(exp, "my_column")
+    """
+
+    #TODO: replace this with a generic utility function, preferably a method on an ExpectationSuite class
+    column_evr_list = [ e for e in evrs["results"] if "column" in e["expectation_config"]["kwargs"] and e["expectation_config"]["kwargs"]["column"] == column ]
+
+    #TODO: Handle the case where zero evrs match the column name
+
+    document = render.renderer.DescriptiveColumnSectionRenderer.render(column_evr_list)
+    view = render.view.DefaultJinjaSectionView.render({
+        "section": document,
+        "section_loop": {"index": 1},
+    })
+
+    if include_styling:
+        html_to_display = bootstrap_link_element+cooltip_style_element+view
+    else:
+        html_to_display = view
+
+    if return_without_displaying:
+        return html_to_display
+    else:
+        display(HTML(html_to_display))

great_expectations/render/__init__.py

@@ -1 +1,3 @@
-from .view import DefaultJinjaPageView
+from .view import (
+    DefaultJinjaPageView,
+)

great_expectations/render/renderer/__init__.py

@@ -1,2 +1,8 @@
-from .column_section_renderer import DescriptiveColumnSectionRenderer, PrescriptiveColumnSectionRenderer
-from .page_renderer import DescriptivePageRenderer, PrescriptivePageRenderer
+from .column_section_renderer import (
+    DescriptiveColumnSectionRenderer,
+    PrescriptiveColumnSectionRenderer,
+)
+from .page_renderer import (
+    DescriptivePageRenderer,
+    PrescriptivePageRenderer,
+)

great_expectations/render/view/__init__.py

@@ -1,4 +1,5 @@
 from .view import (
     DefaultJinjaPageView,
     DefaultJinjaIndexPageView,
+    DefaultJinjaSectionView,
 )

requirements-dev.txt

@@ -26,3 +26,4 @@ mock>=3.0.5
 pytest-cov>=2.6.1
 coveralls>=1.3
 altair>=3.1.0
+tzlocal>=1.2