Merge pull request #404 from codegangsta/runtests-argparse
Use argparse in runtests
This commit is contained in:
commit
7f0ca9a349
16
runtests
16
runtests
@ -1,6 +1,7 @@
|
|||||||
#!/usr/bin/env python
|
#!/usr/bin/env python
|
||||||
from __future__ import print_function
|
from __future__ import print_function
|
||||||
|
|
||||||
|
import argparse
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
import tempfile
|
import tempfile
|
||||||
@ -14,16 +15,19 @@ PACKAGE_NAME = os.environ.get(
|
|||||||
|
|
||||||
|
|
||||||
def main(sysargs=sys.argv[:]):
|
def main(sysargs=sys.argv[:]):
|
||||||
target = 'test'
|
targets = {
|
||||||
if len(sysargs) > 1:
|
|
||||||
target = sysargs[1]
|
|
||||||
|
|
||||||
{
|
|
||||||
'vet': _vet,
|
'vet': _vet,
|
||||||
'test': _test,
|
'test': _test,
|
||||||
'gfmxr': _gfmxr
|
'gfmxr': _gfmxr
|
||||||
}[target]()
|
}
|
||||||
|
|
||||||
|
parser = argparse.ArgumentParser()
|
||||||
|
parser.add_argument(
|
||||||
|
'target', nargs='?', choices=tuple(targets.keys()), default='test'
|
||||||
|
)
|
||||||
|
args = parser.parse_args(sysargs[1:])
|
||||||
|
|
||||||
|
targets[args.target]()
|
||||||
return 0
|
return 0
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user