def test_defaults():
command, options = args.parse(
['test', '--num', '1', '--help', '--stuff', 'The remainder.'])
args.ensure_defaults(options, {'num': 2, 'help': True, 'stuff': None})
assert options['help'] == True
assert options['num'] == 1
command, options = args.parse(
['test', '--num', '1', '--help', '--stuff', 'The remainder.'])
args.ensure_defaults(options, {
'num': 2,
'extras': 3,
'help': None,
'stuff': None
})
assert options['extras'] == 3
assert options['num'] == 1
assert_raises(args.ArgumentError, args.ensure_defaults, options, {
'num': 2,
'extras': 3,
'help': None,
'TRAILING': None
})
assert_raises(args.ArgumentError, args.ensure_defaults, options, {
'num': 2,
'extras': 3,
'help': None,
'bad': None
})
def test_parse():
command, options = args.parse(['test', '--num', '1', '--help', '--stuff', 'The remainder.', '--tail'])
assert command, "There should be a command."
assert options, "There should be options."
assert command == "test", "command should be test"
assert options["num"] == 1, "num option wrong"
assert options["help"] == True, "help should be true"
assert options["stuff"] == 'The remainder.', "stuff should a string"
assert options['tail'] == True, "There should be a True tail."
command2, options = args.parse(['--num', '1', '--help', '--stuff', 'The remainder.'])
assert not command2, "There should NOT be a command."
assert options, "There should be options."
assert options["num"] == 1, "num option wrong"
assert options["help"] == True, "help should be true"
assert options["stuff"] == 'The remainder.', "stuff should a string"
_, options = args.parse(['--foo', 'True', '--bar', 'False'])
assert options['foo']
assert not options['bar']
_, options = args.parse(['--foo', 'true', '--bar', 'false'])
assert options['foo']
assert not options['bar']
_, options = args.parse(['--foo', 'yes', '--bar', 'no'])
assert options['foo']
assert not options['bar']
def test_parse():
command, options = args.parse([
'test', '--num', '1', '--help', '--stuff', 'The remainder.', '--tail'
])
assert command, "There should be a command."
assert options, "There should be options."
assert command == "test", "command should be test"
assert options["num"] == 1, "num option wrong"
assert options["help"] == True, "help should be true"
assert options["stuff"] == 'The remainder.', "stuff should a string"
assert options['tail'] == True, "There should be a True tail."
command2, options = args.parse(
['--num', '1', '--help', '--stuff', 'The remainder.'])
assert not command2, "There should NOT be a command."
assert options, "There should be options."
assert options["num"] == 1, "num option wrong"
assert options["help"] == True, "help should be true"
assert options["stuff"] == 'The remainder.', "stuff should a string"
_, options = args.parse(['--foo', 'True', '--bar', 'False'])
assert options['foo']
assert not options['bar']
_, options = args.parse(['--foo', 'true', '--bar', 'false'])
assert options['foo']
assert not options['bar']
_, options = args.parse(['--foo', 'yes', '--bar', 'no'])
assert options['foo']
assert not options['bar']
def test_trailing():
command, options = args.parse(['test', '--num', '1', '--', 'Trailing 1', 'Trailing 2'])
expected = ['Trailing 1', 'Trailing 2']
assert command == 'test'
assert options['TRAILING']
for e in expected: assert e in options['TRAILING']
# test with a corner case of a switch option before trailing
command, options = args.parse(['test', '--num', '1', '--switch', '--', 'Trailing 1', 'Trailing 2'])
for e in expected: assert e in options['TRAILING']
def test_trailing():
command, options = args.parse(
['test', '--num', '1', '--', 'Trailing 1', 'Trailing 2'])
expected = ['Trailing 1', 'Trailing 2']
assert command == 'test'
assert options['TRAILING']
for e in expected:
assert e in options['TRAILING']
# test with a corner case of a switch option before trailing
command, options = args.parse(
['test', '--num', '1', '--switch', '--', 'Trailing 1', 'Trailing 2'])
for e in expected:
assert e in options['TRAILING']
def main():
# O modargs sempre espera que a chamada da linha de comando tenha sido
# $ prog command args
# Por isso ignoramos o primeiro item da tupla retornada por ele.
_, params = args.parse(sys.argv[1:])
_validate_params(params)
user = os.environ['user']
key = os.environ['key']
container = params.get('container', None)
dryrun = params.get('dryrun', False)
rule = params.get('rule', None)
rule_param = params.get('ruleparam', None)
rule_instance = AVAILABLE_RULES[rule](rule_param)
try:
deleted = collect(container=container, dryrun=dryrun, rule=rule_instance, user=user, key=key)
print deleted
print "Removing {0} objects from container {1}".format(len(deleted), container)
except AuthenticationFailed as auth:
print "User or API KEY wrong ", auth.message
except NoSuchContainer as nocontainer:
print "No such container: ", nocontainer.message
except Exception as e:
print "Ops, an error ocurred : ", e.message
import traceback
traceback.print_exc()
sys.exit(0)
def test_defaults():
command, options = args.parse(['test', '--num', '1', '--help', '--stuff', 'The remainder.'])
args.ensure_defaults(options, {'num': 2, 'help': True, 'stuff': None})
assert options['help'] == True
assert options['num'] == 1
command, options = args.parse(['test', '--num', '1', '--help', '--stuff', 'The remainder.'])
args.ensure_defaults(options, {'num': 2, 'extras': 3, 'help': None,
'stuff': None})
assert options['extras'] == 3
assert options['num'] == 1
assert_raises(args.ArgumentError,
args.ensure_defaults,
options, {'num': 2, 'extras': 3, 'help': None, 'TRAILING': None})
assert_raises(args.ArgumentError,
args.ensure_defaults,
options, {'num': 2, 'extras': 3, 'help': None, 'bad': None})
def test_multiple_kwargs():
options = args.parse(['testmanykwargs', '-foo', 'bar', '-foo', 'baz'])[1]
assert options['foo'] == ['bar', 'baz']
options = args.parse(['testmanykwargs', '-foo', 'bar', '-foo', 'baz', '-foo', 'bag'])[1]
assert options['foo'] == ['bar', 'baz', 'bag']
def test_multiple_kwargs():
options = args.parse(['testmanykwargs', '-foo', 'bar', '-foo', 'baz'])[1]
assert options['foo'] == ['bar', 'baz']
options = args.parse(
['testmanykwargs', '-foo', 'bar', '-foo', 'baz', '-foo', 'bag'])[1]
assert options['foo'] == ['bar', 'baz', 'bag']