demo: Add tabs to form examples

[people/ms/westferry.git] / src / westferry / ui / tabs.py

diff --git a/src/westferry/ui/tabs.py b/src/westferry/ui/tabs.py
index 82f7e484b6e7c8695c1ee422c30d8edbc36e9eea..5db68003012dbd1e65afd891c7631f6e77f0b267 100644 (file)
--- a/src/westferry/ui/tabs.py
+++ b/src/westferry/ui/tabs.py
@@ -22,6 +22,7 @@
 import uuid
 
 from . import base
+from . import forms
 from . import graphs
 
 class TabsModule(base.BaseUIModule):
@@ -72,8 +73,14 @@ class Tab(object):
                # List to store all items that have been added to this tab
                self.items = []
 
-       def add_graph(self, *args, **kwargs):
-               graph = graphs.Graph(self.handler, *args, **kwargs)
-               self.items.append(graph)
+       def _add_item(self, cls, *args, **kwargs):
+               item = cls(self.handler, *args, **kwargs)
+               self.items.append(item)
+
+               return item
 
-               return graph
+       def add_form(self, *args, **kwargs):
+               return self._add_item(forms.Form, *args, **kwargs)
+
+       def add_graph(self, *args, **kwargs):
+               return self._add_item(graphs.Graph, *args, **kwargs)