From: Greg Price Date: Fri, 27 Feb 2009 07:56:46 +0000 (-0500) Subject: invirt-quota: more style fixes X-Git-Tag: 0.1.8~2 X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-database.git/commitdiff_plain/a754e7beadaaa4d401bfeaac5b8b67df29fb7893?hp=6dc450951772a078d728538a24006896112bfefb invirt-quota: more style fixes svn path=/trunk/packages/invirt-database/; revision=2199 --- diff --git a/scripts/invirt-quota b/scripts/invirt-quota index 22c663a..f0fb16f 100755 --- a/scripts/invirt-quota +++ b/scripts/invirt-quota @@ -43,23 +43,23 @@ def main(argv): if len(args) != 1: parser.print_help(sys.stderr) return 1 - owner = args[0] + owner_id = args[0] database.connect() database.session.begin() - x = database.Owner.query().filter_by(owner_id=owner).first() - if x is None: - x = database.Owner(owner_id=owner) + owner = database.Owner.query().filter_by(owner_id=owner_id).first() + if owner is None: + owner = database.Owner(owner_id=owner_id) for resource, scope in [('ram', 'total'), ('ram', 'single'), ('disk', 'total'), ('disk', 'single'), ('vms', 'total'), ('vms', 'active')]: val = getattr(opts, resource+scope) if val is not None: - setattr(x, resource+'_quota_'+scope, val if val >= 0 else None) + setattr(owner, resource+'_quota_'+scope, val if val >= 0 else None) database.session.commit() - print str(x) + print owner return 0 if __name__ == '__main__':