X-Git-Url: https://git.llucax.com/software/sercom.git/blobdiff_plain/94b32d115d412a3baa90a96c9165d7f57a189ba0..aa8c5a4c31de79cf2d8e3ab3bb0583af2ffa89ce:/sercom/menu.py?ds=sidebyside

diff --git a/sercom/menu.py b/sercom/menu.py
index a72d6a2..beb4f57 100644
--- a/sercom/menu.py
+++ b/sercom/menu.py
@@ -1,16 +1,20 @@
+# vim: set et sw=4 sts=4 encoding=utf-8 foldmethod=marker :
 
 from turbogears import url
-from turbogears.controllers import Controller
+from turbogears.identity import SecureResource
 from turbogears import identity
 
 class Menu:
     def __init__(self, base):
         self.base = base
-        self.items = filter(lambda i: isinstance(getattr(base, i), Controller), base.__dict__)
+        self.items = [i for i in dir(base)
+            if isinstance(getattr(base, i), SecureResource)]
         self.items.sort()
         self.items = ['dashboard'] + self.items
 
     def _check(self, c):
+        if hasattr(c, 'hide_to_admin') and 'admin' in identity.current.permissions: return False
+        if hasattr(c, 'hide_to_entregar') and 'admin' not in identity.current.permissions: return False
         return c.require.eval_with_object(identity.current)
 
     def __repr__(self):