Question

我是python的新手，目前正在使用argpase。我正在尝试使用作为命令行参数给出的目录路径调用函数，后跟一个argparse选项（-name）和一个遍历目录中所有文件的正则表达式，并将所有匹配项吐出到正则表达式中：

./find.py ../seek -name '[a-z]*\.txt'

但是，我收到的错误似乎是

usage: find.py [-h] [--path PATH] [-name] [--regex REGEX] find.py: error: unrecognized arguments: . . / s e e k / p r o g r a m . c

如果没有-name，它只会打印路径中的所有文件。

这是我到目前为止所做的：

#!/usr/bin/python2.7

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

def parse(argv=None):
   parser = argparse.ArgumentParser()
   parser.add_argument('--path', help='path of directory', action='store')
   parser.add_argument('-name', '--name', action='store_true')
   parser.add_argument('--regex', default=r"[a-z0-9A-Z]")
   args = parser.parse_args(argv)
   print(args)
   return args

def main(argv=None):

   direc = sys.argv[1]
   files = []

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

      if S_ISREG(mode):  
         args = parse(pathname)      
         if args.name:  
            dirls = [re.match(args.regex, pathname)]
            print(dirls)
         else:
            print pathname

if __name__ == '__main__':main()

有什么想法吗？

Answer 1

为了让您的程序运行，您需要一条路径。因此，--path选项必须采用参数。修改您的parse()功能以更改行

parser.add_argument('--path', help='path of directory', action='store')

到

parser.add_argument('--path', help='path of directory', action='store', required=True)

您只需拨打parse_args()一次。删除parse()调用到循环顶部。

你不需要

direc = sys.argv[1]

如果您使用argparse。

re.match()会返回match object，这可能不是您要打印的内容。

您可能需要查看match() versus search()。

match()函数仅检查RE在字符串开头是否匹配，而search()将向前扫描字符串以进行匹配。

如果要打印与正则表达式匹配的文件名，可以执行

  if S_ISREG(mode):  
     #args = parse(pathname)      
     if args.name:  
        #dirls = re.match(args.regex, pathname)
        dirls = re.search(args.regex, pathname)
        if( dirls ):
           print(pathname)
     else:
        print pathname

所以main()应该是

def main(argv=None):
    args = parse(sys.argv[1:])
    print(args)
    #direc = sys.argv[1]
    direc = args.path
    files = []

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

    if S_ISREG(mode):  
       #args = parse(pathname)      
       if args.name:  
          #dirls = re.match(args.regex, pathname)
          dirls = re.search(args.regex, pathname)
          if( dirls ):
             print(pathname)
       else:
          print pathname

要指定与文件名匹配的正则表达式，必须使用--regex选项指定正则表达式。默认情况下，您已使其匹配仅包含数字和（英文）字母的名称。

./find.py --path ../seek -name --regex [a-z]\*.txt

或

./find.py --path ../seek -name --regex '[a-z]*.txt'

Answer 2

参数解析器PATH示例：添加了自定义处理程序的不同类型的参数。对于路径，您可以传递“ -path”，然后传递路径值作为参数

import os
import argparse
from datetime import datetime


def parse_arguments():
    parser = argparse.ArgumentParser(description='Process command line arguments.')
    parser.add_argument('-path', type=dir_path)
    parser.add_argument('-e', '--yearly', nargs = '*', help='yearly date', type=date_year)
    parser.add_argument('-a', '--monthly', nargs = '*',help='monthly date', type=date_month)

    return parser.parse_args()


def dir_path(path):
    if os.path.isdir(path):
    return path
    else:
    raise argparse.ArgumentTypeError(f"readable_dir:{path} is not a valid path")


def date_year(date):
    if not date:
    return

    try:
    return datetime.strptime(date, '%Y')
    except ValueError:
    raise argparse.ArgumentTypeError(f"Given Date({date}) not valid")


def date_month(date):
    if not date:
    return

    try:
    return datetime.strptime(date, '%Y/%m')
    except ValueError:
    raise argparse.ArgumentTypeError(f"Given Date({date}) not valid")


def main():
    parsed_args = parse_arguments()

if __name__ == "__main__":
main()

命令行参数后跟argparse选项

2 个答案: