X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-web.git/blobdiff_plain/dbd56d540e7ec9cbb22025f37c0a69278674d3bb..45cc4357dad4c486b8f67a937fb3afa91fcfa09d:/code/webcommon.py diff --git a/code/webcommon.py b/code/webcommon.py index e82f790..0e3fd1d 100644 --- a/code/webcommon.py +++ b/code/webcommon.py @@ -1,6 +1,7 @@ """Exceptions for the web interface.""" import time +from invirt import database from invirt.database import Machine, MachineAccess class MyException(Exception): @@ -44,9 +45,11 @@ class State(object): def getMachines(self): if self.isadmin: - return Machine.select() + return Machine.query().join('acl').filter( + database.or_(MachineAccess.user==self.username, + Machine.adminable==True)) else: - return Machine.query().join('acl').select_by(user=self.username) + return Machine.query().join('acl').filter_by(user=self.username) machines = cachedproperty(getMachines) xmlist_raw = cachedproperty(lambda self: controls.getList())