X-Git-Url: https://git.llucax.com/software/sercom.git/blobdiff_plain/22af73dbf4c0af8a7c8aa0a976c4e4748fcd2f35..f7393e0e6363d1c56d579a9fe86c99f9a45bb4bf:/sercom/menu.py diff --git a/sercom/menu.py b/sercom/menu.py index ef9f590..a72d6a2 100644 --- a/sercom/menu.py +++ b/sercom/menu.py @@ -1,18 +1,21 @@ from turbogears import url -from turbogears import controllers +from turbogears.controllers import Controller +from turbogears import identity class Menu: - def __init__(self, controller): - # Armo la lista de subcontrollers - self.items = [] - for i in controller.__dict__: - if isinstance(getattr(controller, i),controllers.Controller): - self.items.append(i) + def __init__(self, base): + self.base = base + self.items = filter(lambda i: isinstance(getattr(base, i), Controller), base.__dict__) self.items.sort() + self.items = ['dashboard'] + self.items + + def _check(self, c): + return c.require.eval_with_object(identity.current) def __repr__(self): - t = """ + option = u"""" """ + template = """
""" - s = '' + s = option % ('', '-----') for i in self.items: - s = s + u"""" """ % (url('/' + i), i.capitalize().replace('_', ' ')) - return t % s + if i == 'dashboard' or self._check(getattr(self.base, i)): + s += option % (url('/' + i), i.capitalize().replace('_', ' ')) + return template % s