WIP adding renderers

omniduct/renderers/base.py

@@ -0,0 +1,37 @@
+from abc import abstractmethod
+from omniduct.utils.magics import MagicsProvider, process_line_arguments, process_line_cell_arguments
+
+
+class BaseRenderer(MagicsProvider):
+
+    @abstractmethod
+    def render_template(self, statement):
+        pass
+
+    def _register_magics(self, base_name):
+        """
+        The following magic functions will be registered (assuming that
+        the base name is chosen to be 'hive'):
+        - Cell Magics:
+            - `%%html`: For querying the database.
+        - Line Magics:
+            - `%hive`: For querying the database using a named template.
+
+        Documentation for these magics is provided online.
+        """
+        from IPython.core.magic import register_cell_magic
+
+        def template_render_magic(template, context=None, **kwargs):
+
+            ip = get_ipython()
+
+            if context is None:
+                context = ip.user_ns
+
+            return self.render_template(template, context)
+
+
+        @register_cell_magic(base_name)
+        @process_line_cell_arguments
+        def render_magic(*args, **kwargs):
+            return template_render_magic(*args, **kwargs)

omniduct/renderers/jinja.py

@@ -0,0 +1,10 @@
+from .base import BaseRenderer
+from jinja2 import Template
+
+
+class JinjaRenderer(BaseRenderer):
+
+    name = 'jinja'
+
+    def render_template(self, template, context=None, **kwargs):
+        return Template(template).render(**context)

omniduct/renderers/md.py

@@ -0,0 +1,20 @@
+from .base import BaseRenderer
+from jinja2 import Template
+import markdown
+
+
+class MdRepr(object):
+
+    def __init__(self, rendered):
+        self.html = markdown.markdown(rendered)
+
+    def _repr_html_(self):
+        return self.html
+
+
+class MdRenderer(BaseRenderer):
+
+    name = 'md'
+
+    def render_template(self, template, context=None, **kwargs):
+        return MdRepr(Template(template).render(**context))