如何在 Python 中传递任意参数

Question

在我必须做的任务中，我应该创建一个替代“rm”命令的方法。我必须传递可变数量的参数，例如，

$ rm.py /path/to/some/file ./somefile someotherfile
$ rm.py /path/to/some/file ./somefile someotherfile -r
$ rm.py -r /path/to/some/file ./somefile someotherfile
$ rm.py *.java

-r 参数可以在任何地方作为任何参数传递。 -r 的含义与使用常规“rm”命令时的含义相同。 它递归地删除一个目录及其内容。

它运行脚本，并且给出的任何路径都像

rm.py /path/to/some/file

然后将其移动到输出目录 '~/rm_trash' 如果有重复，我也担心。

但首先，我在理解如何处理它时遇到了一些麻烦。 我是否应该在 for 循环中进行处理，如果其中一个参数等于“-r”，则从那里开始？

我应该导入和使用 argparse 吗？

我还有一些问题，但我想先处理一下我上面问的问题。

不得不再次发布，仍然不明白我如何获得除“r”之外的任意数量参数的具体参数

Answer 1

我会使用 argparse，因为您可以指定乱序操作：

https://docs.python.org/3/library/argparse.html

Answer 2

如果您不想使用argparse，或者不能用于分配目的，您可以查看sys.argv：

is_recursive = any(arg == '-r' for arg in sys.argv)  

# `paths` will be a list of all other arguments
paths = [arg for arg in sys.argv if arg != '-r']   

Answer 3

定义一个解析函数，它查找'-r'字符串：

def parse(alist, astr='-r'):
    try:
        idx = alist.index(astr)
        r = alist.pop(idx)
        return True, alist
    except ValueError:
        return False, alist

如果我们这样做

import sys
r, rest = parse(sys.argv[1:])

应该为r 提供一个真/假值，以及一个文件名列表。在其余代码中使用它们。

测试：

In [326]: list1 = 'rm.py /path/to/some/file ./somefile someotherfile'.split()   
In [329]: r, rest = parse(list1[1:])                                            
In [330]: r, rest                                                               
Out[330]: (False, ['/path/to/some/file', './somefile', 'someotherfile'])

In [331]: list2 = 'rm.py /path/to/some/file ./somefile someotherfile -r'.split()                                                                     
In [332]: r, rest = parse(list2[1:])                                            
In [333]: r, rest                                                               
Out[333]: (True, ['/path/to/some/file', './somefile', 'someotherfile'])

为了说明出现“-r”时会发生什么：

In [335]: list2.index('-r')                                                     
Out[335]: 4
In [336]: list2.pop(4)                                                          
Out[336]: '-r'
In [337]: list2                                                                 
Out[337]: ['rm.py', '/path/to/some/file', './somefile', 'someotherfile']

如果它不存在，我们会得到一个ValueError：

In [338]: list1.index('-r')                                                     
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-338-807fbe60be44> in <module>
----> 1 list1.index('-r')

ValueError: '-r' is not in list

======

使用argparse的等效功能：

In [339]: import argparse                                                       
In [340]: parser = argparse.ArgumentParser()                                    
In [341]: parser.add_argument('-r', action='store_true', help='recursive flag');                                                                       
In [342]: parser.add_argument('rest', nargs='*', help='file names');            
In [343]: args = parser.parse_args(list1[1:])                                   

使用args = parser.parse_args() 读取sys.argv。

结果是一个具有所需属性的namespace 对象：

In [344]: args                                                                  
Out[344]: Namespace(r=False, rest=['/path/to/some/file', './somefile', 'someotherfile'])
In [345]: args.r                                                                
Out[345]: False
In [346]: args.rest                                                             
Out[346]: ['/path/to/some/file', './somefile', 'someotherfile']

args.r 和 args.rest 的使用方式与上述 r 和 rest 相同。

“-r”可以在文件名列表之前或之后。 argparse 在您想要定义更多选项（如“-r”）时变得更加有用。它还负责显示帮助消息和有意义的错误。

In [347]: parser.parse_args(['--help'])                                         
usage: ipython3 [-h] [-r] [rest [rest ...]]

positional arguments:
  rest        file names

optional arguments:
  -h, --help  show this help message and exit
  -r          recursive flag

Answer 4

为什么不使用：

os.system( 'rm ' + ' '.join( sys.argv[1:] )

并完成它。