的Python：argparse閱讀CSV文件的功能

Question

我使用argparse，我想是這樣的：test.py --file hello.csv

def parser(): 
    parser.add_argument("--file", type=FileType('r')) 
    options = parser.parse_args() 

    return options 

def csvParser(filename): 
    with open(filename, 'rb') as f: 
     csv.reader(f) 
     .... 
    .... 
    return par_file 

csvParser(options.filename) 

我得到一個錯誤：類型錯誤脅迫爲Unicode：需要字符串或緩衝區，找到文件。

我該如何解決這個問題？

Answer 1

FileType()​​類型返回已打開 fileobject。

你不需要再次打開它：

def csvParser(f): 
    with f: 
     csv.reader(f) 

從argparse documentation：

To ease the use of various types of files, the argparse module provides the factory FileType which takes the mode= , bufsize= , encoding= and errors= arguments of the open() function. For example, FileType('w') can be used to create a writable file:

>>> 
>>> parser = argparse.ArgumentParser() 
>>> parser.add_argument('bar', type=argparse.FileType('w')) 
>>> parser.parse_args(['out.txt']) 
Namespace(bar=<_io.TextIOWrapper name='out.txt' encoding='UTF-8'>) 

，並從FileType() objects文檔：

Arguments that have FileType objects as their type will open command-line arguments as files with the requested modes, buffer sizes, encodings and error handling (see the open() function for more details)