Question

我是python的新手，目前正在尝试使用argparse添加命令行选项。但是我的代码不起作用，尽管查看了各种在线教程并阅读了argparse，我仍然不完全理解它。我的问题是每当我尝试调用我的-option它给我一个find.py错误：参数正则表达式：

这是我的电话：

./find.py ../Python -name '[0-9]*\.txt'

../ Python是我当前目录背后的一个目录，有一个文件/目录列表。如果没有-name选项，我打印出带有路径的文件（这样可以正常工作）但是使用-name选项我想打印出与正则表达式匹配的文件，但它不起作用。这是我现在拥有的：

#!/usr/bin/python2.7

import os, sys, argparse,re 
from stat import *

def regex_type(s, pattern=re.compile(r"[a-f0-9A-F]")):
   if not pattern.match(s):
      raise argparse.ArgumentTypeError
   return s

def main():
   direc = sys.argv[1]

   for f in os.listdir(direc):
      pathname = os.path.join(direc, f)
      mode = os.stat(pathname).st_mode
      if S_ISREG(mode):
          print pathname

   parser = argparse.ArgumentParser()
   parser.add_argument(
    '-name', default=[sys.stdin], nargs="*")
   parser.add_argument('regex', type=regex_type) 
   args = parser.parse_args()



if __name__ == '__main__': 

      main()

Answer 1

我调整了你的类型函数以提供更多信息：

def regex_type(s, pattern=re.compile(r"[a-f0-9A-F]")):
   print('regex string', s)
   if not pattern.match(s):
      raise argparse.ArgumentTypeError('pattern not match')
   return s

用

调用

2104:~/mypy$ python2 stack50072557.py .

我明白了：

<director list>
('regex string', '.')
usage: stack50072557.py [-h] [-name [NAME [NAME ...]]] regex
stack50072557.py: error: argument regex: pattern not match

因此它尝试将sys.argv[1]（脚本名称后面的第一个字符串）传递给regex_type函数。如果失败，则会发出错误消息和用法。

好的，问题是..;我将创建一个目录：

2108:~/mypy$ mkdir foo
2136:~/mypy$ python2 stack50072557.py foo
('regex string', 'foo')
Namespace(name=[<open file '<stdin>', mode 'r' at 0x7f3bea2370c0>], regex='foo')

2138:~/mypy$ python2 stack50072557.py foo -name a b c
('regex string', 'foo')
Namespace(name=['a', 'b', 'c'], regex='foo')

'-name'后面的字符串被分配给该属性。您的代码中没有任何内容可以测试它们或通过regex_type函数传递它们。只有第一个非标志字符串才能这样做。

最初阅读sys.argv[1]并未将其从列表中删除。它仍然存在于解析器中。

我会设置一个使用store_true --name参数和2个位置的解析器 - 一个用于dir，另一个用于regex。

解析后检查args.name。如果为false，则打印args.dir的内容。如果为true，请对这些内容执行args.regex过滤器。 glob可能有用。

解析器会找出您的用户想要的内容。你自己的代码就会对它起作用。特别是作为初学者，分离这两个步骤更容易，更清洁。

使用：

def parse(argv=None):
    parser = argparse.ArgumentParser()
    parser.add_argument('-n', '--name', action='store_true')
    parser.add_argument('--dir', default='.')
    parser.add_argument('--regex', default=r"[a-f0-9A-F]")
    args = parser.parse_args(argv)
    print(args)
    return args

def main(argv=None):
    args = parse(argv)
    dirls = os.listdir(args.dir)
    if args.name:
        dirls = [f for f in dirls if re.match(args.regex, f)]
        print(dirls)
    else:
        print(dirls)

我的运行方式如下：

1005:~/mypy$ python stack50072557.py 
Namespace(dir='.', name=False, regex='[a-f0-9A-F]')
['test.npz', 'stack49909128.txt', 'stack49969840.txt', 'stack49824248.py', 'test.h5', 'stack50072557.py', 'stack49963862.npy', 'Mcoo.npz', 'test_attribute.h5', 'stack49969861.py', 'stack49969605.py', 'stack49454474.py', 'Mcsr.npz', 'Mdense.npy', 'stack49859957.txt', 'stack49408644.py', 'Mdok', 'test.mat5', 'stack50012754.py', 'foo', 'test']
1007:~/mypy$ python stack50072557.py -n
Namespace(dir='.', name=True, regex='[a-f0-9A-F]')
['foo']
1007:~/mypy$ python stack50072557.py -n --regex='.*\.txt'
Namespace(dir='.', name=True, regex='.*\\.txt')
['stack49909128.txt', 'stack49969840.txt', 'stack49859957.txt']

和帮助：

1007:~/mypy$ python stack50072557.py -h
usage: stack50072557.py [-h] [-n] [--dir DIR] [--regex REGEX]

optional arguments:
  -h, --help     show this help message and exit
  -n, --name
  --dir DIR
  --regex REGEX

如果我将dir行更改为：

parser.add_argument('dir', default='.')

现在

帮助

1553:~/mypy$ python stack50072557.py -h
usage: stack50072557.py [-h] [-n] [--regex REGEX] dir

positional arguments:
  dir

optional arguments:
  -h, --help     show this help message and exit
  -n, --name
  --regex REGEX

并且运行是：

1704:~/mypy$ python stack50072557.py -n
usage: stack50072557.py [-h] [-n] [--regex REGEX] dir
stack50072557.py: error: too few arguments

1705:~/mypy$ python stack50072557.py . -n
Namespace(dir='.', name=True, regex='[a-f0-9A-F]')
['foo']

1705:~/mypy$ python stack50072557.py ../mypy -n --regex='.*\.txt'
Namespace(dir='../mypy', name=True, regex='.*\\.txt')
['stack49909128.txt', 'stack49969840.txt', 'stack49859957.txt']

我收到错误，因为它现在需要一个目录，即使它是'。'。

请注意，该脚本仍然使用：

if __name__ == '__main__': 
    main()

我的main加载dir，并将regex过滤器应用于该名称列表。我的args.dir取代了您的direc。

给定路径后的argparse命令行-option

1 个答案: