source: trunk/packages/invirt-database/scripts/invirt-quota @ 2213

Last change on this file since 2213 was 2199, checked in by price, 16 years ago

invirt-quota: more style fixes

  • Property svn:executable set to *
File size: 2.1 KB
RevLine 
[2132]1#!/usr/bin/env python
2
3"""
[2176]4invirt-quota allows an administrator to set memory, disk, and VM quotas
[2134]5for an owner.  Invoking with only an owner name returns the current quotas for
6that owner.  Setting a parameter to -1 restores the default.
[2132]7"""
8
[2198]9import sys
10import optparse
[2132]11
[2198]12from invirt import database
13
[2132]14def main(argv):
[2198]15    parser = optparse.OptionParser(usage = '%prog <owner> [options]',
[2194]16            description = __doc__.strip())
17    parser.add_option('-m', '--ram-total',
[2135]18            type = 'int',
[2194]19            dest = 'ramtotal',
[2135]20            help = 'set total concurrent RAM quota')
[2194]21    parser.add_option('-n', '--ram-single',
[2135]22            type = 'int',
[2194]23            dest = 'ramsingle',
[2135]24            help = 'set single VM RAM quota')
25    parser.add_option('-d', '--disk-total',
26            type = 'int',
27            dest = 'disktotal',
28            help = 'set total disk quota')
29    parser.add_option('-e', '--disk-single',
30            type = 'int',
31            dest = 'disksingle',
32            help = 'set single VM disk quota')
33    parser.add_option('-v', '--vms-total',
34            type = 'int',
35            dest = 'vmstotal',
36            help = 'set total VM quota')
37    parser.add_option('-w', '--vms-active',
38            type = 'int',
39            dest = 'vmsactive',
40            help = 'set active VM quota')
41    opts, args = parser.parse_args()
[2132]42
[2135]43    if len(args) != 1:
[2198]44        parser.print_help(sys.stderr)
[2135]45        return 1
[2199]46    owner_id = args[0]
[2198]47    database.connect()
48    database.session.begin()
[2135]49   
[2199]50    owner = database.Owner.query().filter_by(owner_id=owner_id).first()
51    if owner is None:
52        owner = database.Owner(owner_id=owner_id)
[2132]53
[2194]54    for resource, scope in [('ram',  'total'), ('ram',  'single'),
55                            ('disk', 'total'), ('disk', 'single'),
56                            ('vms',  'total'), ('vms',  'active')]:
[2198]57        val = getattr(opts, resource+scope)
58        if val is not None:
[2199]59            setattr(owner, resource+'_quota_'+scope, val if val >= 0 else None)
[2132]60
[2198]61    database.session.commit()
[2199]62    print owner
[2135]63    return 0
[2132]64
65if __name__ == '__main__':
[2198]66    sys.exit(main(sys.argv))
Note: See TracBrowser for help on using the repository browser.