파이썬은 argparse : 옵션 및 위치 모두 값을

Question

다음 사용을 고려 인수에 대해 :

usage: do.py [-h] [-s | -m] filename 

이 전체 사용하지 않습니다.

--filename FILENAME 

뿐만 아니라, filename 그래서 난 표준 입력에서 읽을 수있는 선택해야한다 :하지만 효과적으로 filename가 아닌 파일의 실제 값이 될 것입니다 싶다. UNIX에서 cat 프로그램에 대해 생각해보십시오.

당신은 단순히 말 :

cat filename 

또는

cat 

편집 : 나는 명령 줄 옵션없이 프로그램 do.py을 실행하면 지금, 내가 error: too few arguments을 얻을 것이다. 대신 유효한 filename을 제공하지 않더라도 여전히 실행하고 싶습니다. 어떻게해야합니까?

Answer 1

업데이트 2 : ArgParse 문서에서,

One of the more common uses of nargs='?' is to allow optional input and output files:

>>> parser = argparse.ArgumentParser() 
>>> parser.add_argument('infile', nargs='?', type=argparse.FileType('r'), 
...      default=sys.stdin) 
>>> parser.add_argument('outfile', nargs='?', type=argparse.FileType('w'), 
...      default=sys.stdout) 
>>> parser.parse_args(['input.txt', 'output.txt']) 
Namespace(infile=<open file 'input.txt', mode 'r' at 0x...>, 
      outfile=<open file 'output.txt', mode 'w' at 0x...>) 
>>> parser.parse_args([]) 
Namespace(infile=<open file '<stdin>', mode 'r' at 0x...>, 
      outfile=<open file '<stdout>', mode 'w' at 0x...>) 

원래 답 :

이 간단하다 : 단지 기본 값과 nargs='*'와 위치 인수를 추가합니다. 제로 인수가있는 경우 기본값은 그렇지 않으면 명령 행 인수가 사용됩니다 사용됩니다

>>> p = argparse.ArgumentParser() 
>>> p.add_argument('filename', nargs='*', default=['-']) 
>>> p.parse_args([]) 
Namespace(filename=['-']) 
>>> p.parse_args(['abc']) 
Namespace(filename=['abc']) 

일반적으로 - 표준 입력/표준 출력을 참조하는 데 사용됩니다.

def get_inputs(ns): 
    """Iterate over input files.""" 
    for path in ns.filename: 
     if path == '-': 
      yield sys.stdin 
     else: 
      yield open(path, 'r') 

업데이트 :

그런 다음 당신은 이런 식으로 뭔가를 할 내가 여러 파일 이름을 원하는 가정, cat 여러 파일 이름을 소요하기 때문이다. 0 또는 하나의 파일 이름을 원할 경우 nargs='?'을 사용할 수 있습니다.