🔨 code refactoring

Author: chfw

.moban.d/helper_and_partial.rst.jj2

@@ -4,13 +4,12 @@ Let's save the following file a `script.py` under `helper_and_partial` folder:
 
 .. code-block:: python
 
-   from moban_handlebars.helpers import HandlebarHelper
-   from moban_handlebars.partials import register_partial
+   from moban_handlebars.api import Helper, register_partial
 
    register_partial('header', '<h1>People</h1>')
 
 
-   @HandlebarHelper('list')
+   @Helper('list')
    def _list(this, options, items):
        result = [u'<ul>']
        for thing in items:
@@ -24,7 +23,7 @@ Let's save the following file a `script.py` under `helper_and_partial` folder:
 Let invoke handlebar template:
 {% raw %}
 
-.. code-block: bash
+.. code-block:: bash
 
    $ moban --template-type hbs -pd helper_and_partial -c data.json "{{>header}}{{#list people}}{{name}} {{age}}{{/list}}"
    Handlebars-ing {{>header}... to moban.output

moban_handlebars/api.py

@@ -0,0 +1,24 @@
+from pybars import Compiler
+from lml.plugin import PluginInfo
+
+from moban_handlebars.constants import HELPER_EXTENSION, PARTIALS_EXTENSION
+
+
+class Helper(PluginInfo):
+    """
+    @Helper('tag_used_inside_handlebar')
+    def the_actual_implementation(this, options, items):
+        ...
+    """
+    def __init__(self, helper):
+        super(Helper, self).__init__(HELPER_EXTENSION)
+        self.helper = helper
+
+    def tags(self):
+        yield self.helper
+
+
+def register_partial(identifier, partial):
+    plugin = PluginInfo(PARTIALS_EXTENSION, tags=[identifier])
+    partial = Compiler().compile(partial)
+    plugin({identifier: partial})

moban_handlebars/helpers.py

@@ -1,10 +1,9 @@
-from moban_handlebars.helpers import HandlebarHelper
-from moban_handlebars.partials import register_partial
+from moban_handlebars.api import Helper, register_partial
 
 register_partial("header", "<h1>People</h1>")
 
 
-@HandlebarHelper("list")
+@Helper("list")
 def _list(this, options, items):
     result = [u"<ul>"]
     for thing in items: