Changeset 2135 for trunk/packages/invirt-base/scripts
- Timestamp:
- Feb 17, 2009, 2:16:12 AM (16 years ago)
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
trunk/packages/invirt-base/scripts/invirt-setquotas
r2134 r2135 8 8 Examples: 9 9 10 invirt-setquota joeuser -mt 512 -ms None10 invirt-setquota joeuser -mt 512 -ms -1 11 11 """ 12 12 … … 15 15 from optparse import OptionParser 16 16 17 class invirt_exception(Exception): pass 17 def main(argv): 18 parser = OptionParser(usage = '%prog owner [options]', 19 description = __doc__.strip().split('\n\n')[0]) 20 parser.add_option('-m', '--mem-total', 21 type = 'int', 22 dest = 'memtotal', 23 help = 'set total concurrent RAM quota') 24 parser.add_option('-n', '--mem-single', 25 type = 'int', 26 dest = 'memsingle', 27 help = 'set single VM RAM quota') 28 parser.add_option('-d', '--disk-total', 29 type = 'int', 30 dest = 'disktotal', 31 help = 'set total disk quota') 32 parser.add_option('-e', '--disk-single', 33 type = 'int', 34 dest = 'disksingle', 35 help = 'set single VM disk quota') 36 parser.add_option('-v', '--vms-total', 37 type = 'int', 38 dest = 'vmstotal', 39 help = 'set total VM quota') 40 parser.add_option('-w', '--vms-active', 41 type = 'int', 42 dest = 'vmsactive', 43 help = 'set active VM quota') 44 opts, args = parser.parse_args() 18 45 19 def main(argv): 20 try: 21 parser = OptionParser(usage = '%prog owner [options]', 22 description = __doc__.strip().split('\n\n')[0]) 23 parser.add_option('-m', '--mem-total', 24 type = 'int', 25 dest = 'memtotal', 26 help = 'set total concurrent RAM quota') 27 parser.add_option('-n', '--mem-single', 28 type = 'int', 29 dest = 'memsingle', 30 help = 'set single VM RAM quota') 31 parser.add_option('-d', '--disk-total', 32 type = 'int', 33 dest = 'disktotal', 34 help = 'set total disk quota') 35 parser.add_option('-e', '--disk-single', 36 type = 'int', 37 dest = 'disksingle', 38 help = 'set single VM disk quota') 39 parser.add_option('-v', '--vms-total', 40 type = 'int', 41 dest = 'vmstotal', 42 help = 'set total VM quota') 43 parser.add_option('-w', '--vms-active', 44 type = 'int', 45 dest = 'vmsactive', 46 help = 'set active VM quota') 47 opts, args = parser.parse_args() 46 if len(args) != 1: 47 print >> stderr, __doc__.strip() 48 return 1 49 owner = args[0] 50 connect() 51 session.begin() 52 53 x = Owner.query().filter_by(owner_id=owner).first() 54 if x == None: 55 x = Owner(owner_id=owner, ram_quota_total=None, ram_quota_single=None, 56 disk_quota_total=None, disk_quota_single=None, 57 vms_quota_total=None, vms_quota_active) 48 58 49 print opts 50 if len(args) != 1: 51 raise invirt_exception(__doc__.strip()) 52 owner = args[0] 53 connect() 54 session.begin() 55 56 x = Owner.query().filter_by(owner_id=owner).first() 59 if opts.memtotal != None: 60 total = int(opts.memtotal) 61 if total == -1: 62 x.ram_quota_total = None 63 else: 64 x.ram_quota_total = total 57 65 58 edited = False 59 if opts.memtotal != None: 60 total = int(opts.memtotal) 61 if total == -1: 62 x.ram_quota_total = None 63 else: 64 x.ram_quota_total = total 65 edited = True 66 if opts.memsingle != None: 67 single = int(opts.memsingle) 68 if single == -1: 69 x.ram_quota_single = None 70 else: 71 x.ram_quota_single = single 66 72 67 if opts.memsingle != None: 68 single = int(opts.memsingle) 69 if single == -1: 70 x.ram_quota_single = None 71 else: 72 x.ram_quota_single = single 73 edited = True 73 if opts.disktotal != None: 74 total = int(opts.disktotal) 75 if total == -1: 76 x.disk_quota_total = None 77 else: 78 x.disk_quota_total = total 74 79 75 if opts.disktotal != None: 76 total = int(opts.disktotal) 77 if total == -1: 78 x.disk_quota_total = None 79 else: 80 x.disk_quota_total = total 81 edited = True 80 if opts.disksingle != None: 81 single = int(opts.disksingle) 82 if single == -1: 83 x.disk_quota_single = None 84 else: 85 x.disk_quota_single = single 82 86 83 if opts.disksingle != None: 84 single = int(opts.disksingle) 85 if single == -1: 86 x.disk_quota_single = None 87 else: 88 x.disk_quota_single = single 89 edited = True 87 if opts.vmstotal != None: 88 total = int(opts.vmstotal) 89 if total == -1: 90 x.vms_quota_total = None 91 else: 92 x.vms_quota_total = total 90 93 91 if opts.vmstotal != None: 92 total = int(opts.vmstotal) 93 if total == -1: 94 x.vms_quota_total = None 95 else: 96 x.vms_quota_total = total 97 edited = True 94 if opts.vmsactive != None: 95 active = int(opts.vmsactive) 96 if active == -1: 97 x.vms_quota_active = None 98 else: 99 x.vms_quota_active = active 98 100 99 if opts.vmsactive != None: 100 active = int(opts.vmsactive) 101 if active == -1: 102 x.vms_quota_active = None 103 else: 104 x.vms_quota_active = active 105 edited = True 106 107 if edited: 108 session.commit() 109 print str(x) 110 111 except invirt_exception, ex: 112 print >> stderr, ex 113 return 1 101 session.commit() 102 print str(x) 103 return 0 114 104 115 105 if __name__ == '__main__':
Note: See TracChangeset
for help on using the changeset viewer.