source: trunk/packages/invirt-database/python/database/owner.py @ 2142

Last change on this file since 2142 was 2142, checked in by iannucci, 16 years ago

Fixed another stupid bug.

  • Property svn:executable set to *
File size: 2.6 KB
Line 
1MAX_MEMORY_TOTAL = 512
2MAX_MEMORY_SINGLE = 512
3MAX_DISK_TOTAL = 50
4MAX_DISK_SINGLE = 50
5MAX_VMS_TOTAL = 10
6MAX_VMS_ACTIVE = 4
7
8class Owner(object):
9    def __repr__(self):
10        return """<Owner %s: ram_quota_total=%s MB ram_quota_single=%s MB
11disk_quota_total=%s MB disk_quota_single=%s MB
12vms_quota_total=%s vms_quota_active=%s >""" % (self.owner_id, 
13               self.ram_quota_total if self.ram_quota_total else MAX_MEMORY_TOTAL,
14               self.ram_quota_single if self.ram_quota_single else MAX_MEMORY_SINGLE, 
15               self.disk_quota_total if self.disk_quota_total else MAX_DISK_TOTAL,
16               self.disk_quota_single if self.disk_quota_single else MAX_DISK_SINGLE, 
17               self.vms_quota_total if self.vms_quota_total else MAX_VMS_TOTAL,
18               self.vms_quota_active if self.vms_quota_active else MAX_VMS_ACTIVE)
19    def getMemoryQuotas(owner):
20        owner_info = Owner.query().filter_by(owner_id=owner).first()
21        if owner_info != None:
22            quota_total = owner_info.ram_quota_total
23            if quota_total == None:
24                quota_total = MAX_MEMORY_TOTAL
25            quota_single = owner_info.ram_quota_single
26            if quota_single == None:
27                quota_single = MAX_MEMORY_SINGLE
28        else:
29            quota_total = MAX_MEMORY_TOTAL
30            quota_single = MAX_MEMORY_SINGLE
31        return (quota_total, quota_single)
32    getMemoryQuotas = staticmethod(getMemoryQuotas)
33    def getDiskQuotas(owner):
34        owner_info = Owner.query().filter_by(owner_id=owner).first()
35        if owner_info != None:
36            quota_total = owner_info.disk_quota_total
37            if quota_total == None:
38                quota_total = MAX_DISK_TOTAL
39            quota_single = owner_info.disk_quota_single
40            if quota_single == None:
41                quota_single = MAX_DISK_SINGLE
42        else:
43            quota_total = MAX_DISK_TOTAL
44            quota_single = MAX_DISK_SINGLE
45        return (quota_total, quota_single)
46    getDiskQuotas = staticmethod(getDiskQuotas)
47    def getVMQuotas(owner):
48        owner_info = Owner.query().filter_by(owner_id=owner).first()
49        if owner_info != None:
50            quota_total = owner_info.vms_quota_total
51            if quota_total == None:
52                quota_total = MAX_VMS_TOTAL
53            quota_active = owner_info.vms_quota_active
54            if quota_active == None:
55                quota_active = MAX_VMS_ACTIVE
56        else:
57            quota_total = MAX_VMS_TOTAL
58            quota_active = MAX_VMS_ACTIVE
59        return (quota_total, quota_active)
60    getVMQuotas = staticmethod(getVMQuotas)
Note: See TracBrowser for help on using the repository browser.