93 lines
2.4 KiB
Python
93 lines
2.4 KiB
Python
from quota import Quota
|
|
from share import Share
|
|
from synergy.common.serializer import SynergyObject
|
|
|
|
|
|
__author__ = "Lisa Zangrando"
|
|
__email__ = "lisa.zangrando[AT]pd.infn.it"
|
|
__copyright__ = """Copyright (c) 2015 INFN - INDIGO-DataCloud
|
|
All Rights Reserved
|
|
|
|
Licensed under the Apache License, Version 2.0;
|
|
you may not use this file except in compliance with the
|
|
License. You may obtain a copy of the License at:
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing,
|
|
software distributed under the License is distributed on an
|
|
"AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
|
|
either express or implied.
|
|
See the License for the specific language governing
|
|
permissions and limitations under the License."""
|
|
|
|
|
|
class Project(SynergyObject):
|
|
|
|
def __init__(self):
|
|
super(Project, self).__init__()
|
|
|
|
self.set("data", {})
|
|
self.set("users", {})
|
|
self.set("share", Share())
|
|
self.set("quota", Quota())
|
|
self.set("TTL", float(0))
|
|
self.set("enabled", False)
|
|
|
|
def getData(self):
|
|
return self.get("data")
|
|
|
|
def getQuota(self):
|
|
return self.get("quota")
|
|
|
|
def setQueue(self, queue):
|
|
self.set("queue", queue)
|
|
|
|
def getQueue(self):
|
|
return self.get("queue")
|
|
|
|
def getShare(self):
|
|
return self.get("share")
|
|
|
|
def getTTL(self):
|
|
return self.get("TTL")
|
|
|
|
def setTTL(self, TTL):
|
|
self.set("TTL", TTL)
|
|
|
|
def addUser(self, user):
|
|
if self.get("users").get(user.getId(), None):
|
|
raise Exception("user %r already exists!" % (user.getId()))
|
|
|
|
self.get("users")[user.getId()] = user
|
|
|
|
def getUser(self, id=None, name=None):
|
|
if id:
|
|
return self.get("users").get(id, None)
|
|
elif name:
|
|
for user in self.get("users").values():
|
|
if name == user.getName():
|
|
return user
|
|
return None
|
|
|
|
def getUsers(self):
|
|
return self.get("users").values()
|
|
|
|
def isEnabled(self):
|
|
return self.get("enabled")
|
|
|
|
def removeUser(self, user_id):
|
|
return self.get("users").pop(user_id, None)
|
|
|
|
def setEnabled(self, enabled=True):
|
|
self.set("enabled", enabled)
|
|
|
|
def serialize(self):
|
|
queue = self.getQueue()
|
|
usage = queue.getUsage(self.getId())
|
|
|
|
self.getData()["queue_usage"] = usage
|
|
self.getData()["queue_size"] = queue.getSize()
|
|
|
|
return super(Project, self).serialize()
|