例如,如果我输入“python prog.py create filename”,结果应该是创建一个名为“filename”的文件。但我不知道如何将“args”放入创造的定义中。有人可以给我一些建议吗?这是代码:
import argparse
import sys
class ExecuteShell(object):
def create(self, args):
"""create a file"""
print 'xxxxxxxxxxx', args
#return args
def list(self, args):
"""ccccccc"""
print args
#return args
def delete(self, args):
"""ddddddd"""
print 'ddddddd'
#return args
class TestShell(object):
def _find_actions(self, subparsers, actions_module, args):
for attr in (action for action in dir(actions_module) if not action.startswith('__')):
callback = getattr(actions_module, attr)
desc = callback.__doc__ or ''
subparser = subparsers.add_parser(attr, description=desc, add_help=False)
subparser.add_argument('-h', '--help', action='help',
help=argparse.SUPPRESS)
.......
def main(self, args):
parser = argparse.ArgumentParser()
subparsers = parser.add_subparsers()
a = ExecuteShell()
subcommand_parser = self._find_actions(subparsers, a, args)
(options, args) = parser.parse_known_args(args)
if __name__ == "__main__":
a = TestShell()
a.main(sys.argv[1:])
非常感谢!
答案 0 :(得分:1)
在您概述的框架内,我可能会像以下程序一样实现它。关键项目是:
parser.add_subparsers
采用dest
参数。这可以让您知道调用了哪个subparser。subparser.add_argument('files',...)
下面)getattr(a,options.command) ...
行)
import argparse
import sys
class ExecuteShell(object):
# This function demonstrates that you can put
# methods in ExecuteShell(), if you need to,
# without having them appear as command-line commands
# simply by prepending an underscore.
def _useful(self):
"""Useful method that ISNT a command."""
return 3.14159
def create(self, args):
"""create a file"""
print 'xxxxxxxxxxx', args
def list(self, args):
"""ccccccc"""
print args
def delete(self, args):
"""ddddddd"""
print 'ddddddd'
class TestShell(object):
def find_actions(self, subparsers, actions_module, args):
for attr in (action for action in dir(actions_module) if not action.startswith('_')):
callback = getattr(actions_module, attr)
desc = callback.__doc__ or ''
subparser = subparsers.add_parser(attr, description=desc, help=desc)
# This add_argument describes the positional argument, like so:
# 'files' -- the result of matching this argument will be
# stored in `options.files`
# metavar='FILE' -- The help messages will use the word "FILE"
# when describing this argument
# nargs='+' -- `narg` describes how many arguments this should
# match. The value `+` means "all of them, but at
# least one."
subparser.add_argument('files', metavar='FILE', nargs='+')
def main(self, args):
# setup parser
parser = argparse.ArgumentParser()
subparsers = parser.add_subparsers(dest='command')
a = ExecuteShell()
subcommand_parser = self.find_actions(subparsers, a, args)
# Run parser
args = parser.parse_args(args)
# Run selected command
# If we reach this statement, then:
# args.command is the command the user chose,
# for example, 'create' or 'list'
# args.files is the list of arguments that followed 'create' or 'list'
# a is an instance of ExecuteShell
# getattr(a, args.command) is the attribute or method of `a`
# the name in `args.command`, for example, a.create or a.list
# getattr(a, args.command)(args.files) invokes that method,
# passing in the list of user arguments
# So, this line in effect might call `a.create(['file1', 'file2'])`
getattr(a, args.command)(args.files)
if __name__ == "__main__":
a = TestShell()
a.main(sys.argv[1:])