Merge pull request #195 from wlorenzetti/master

Add custom fields to tree utils function for custom Tree model.
idlesign · Sep 16, 2018 · bdc41b9 · bdc41b9
2 parents ee43d35 + ef5170a
commit bdc41b9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sitetree/utils.py b/sitetree/utils.py
@@ -16,15 +16,15 @@ def generate_id_for(obj):
     return id(obj)
 
 
-def tree(alias, title='', items=None):
+def tree(alias, title='', items=None, **kwargs):
     """Dynamically creates and returns a sitetree.
 
     :param str|unicode alias:
     :param str|unicode title:
     :param iterable items: dynamic sitetree items objects created by `item` function.
     :rtype: TreeBase
     """
-    tree_obj = get_tree_model()(alias=alias, title=title)
+    tree_obj = get_tree_model()(alias=alias, title=title, **kwargs)
     tree_obj.id = generate_id_for(tree_obj)
     tree_obj.is_dynamic = True