Add admins table.
[invirt/packages/invirt-database.git] / python / database / owner.py
1 from invirt.database import record
2 from invirt.database.models import session
3
4 class Owner(record.Record):
5     _f = {
6         'ram_quota_total': (512, 'MiB'),
7         'ram_quota_single': (512, 'MiB'),
8         'disk_quota_total': (50, 'GiB'),
9         'disk_quota_single': (50, 'GiB'),
10         'vms_quota_total': (10, ''),
11         'vms_quota_active': (4, '')
12     }
13     _default = dict([(_k,_v[0]) for _k,_v in _f.items()])
14     def _unitFormatter(unit):
15         return lambda v:'%s%s'%(v,unit)
16     _format = dict([(_k,_unitFormatter(_v[1])) for _k,_v in _f.items()])
17     _identity_field = 'owner_id'
18
19     @classmethod
20     def getMemoryQuotas(cls, owner):
21         owner_info = cls.query().get(owner)
22         if owner_info == None:
23             owner_info = cls(owner_id=owner)
24             session.expunge(owner_info)
25         return (owner_info.get('ram_quota_total'), owner_info.get('ram_quota_single'))
26
27     @classmethod
28     def getDiskQuotas(cls, owner):
29         owner_info = cls.query().get(owner)
30         if owner_info == None:
31             owner_info = cls(owner_id=owner)
32             session.expunge(owner_info)
33         return (owner_info.get('disk_quota_total'), owner_info.get('disk_quota_single'))
34
35     @classmethod
36     def getVMQuotas(cls, owner):
37         owner_info = cls.query().get(owner)
38         if owner_info == None:
39             owner_info = cls(owner_id=owner)
40             session.expunge(owner_info)
41         return (owner_info.get('vms_quota_total'), owner_info.get('vms_quota_active'))