python - 使用 Argparse 在 Python 中创建文件转换器

Question

我必须使用命令提示符和Python来接收csv文件形式的输入，然后读取它并将其转换为与csv文件同名的xml文件，除了.xml文件扩展名或用户可以使用 -o --output 可选命令行参数设置输出文件名和路径。好吧，我已经在谷歌上搜索了好几天，到目前为止，我的程序允许我输入命令行参数，我可以将 csv 转换为 xml 文件，但它不会使用与 csv 文件相同的名称或当用户设置名称。相反，它只是打印出一个空白文件。这是我的代码:

    import sys, argparse
    import csv
    import indent
    from xml.etree.ElementTree import ElementTree, Element, SubElement, Comment, tostring

    parser=argparse.ArgumentParser(description='Convert wordlist text files to various formats.', prog='Text Converter')
    parser.add_argument('-v','--verbose',action='store_true',dest='verbose',help='Increases messages being printed to stdout')
    parser.add_argument('-c','--csv',action='store_true',dest='readcsv',help='Reads CSV file and converts to XML file with same name')
    parser.add_argument('-x','--xml',action='store_true',dest='toxml',help='Convert CSV to XML with different name')
    parser.add_argument('-i','--inputfile',type=argparse.FileType('r'),dest='inputfile',help='Name of file to be imported',required=True)
    parser.add_argument('-o','--outputfile',type=argparse.FileType('w'),dest='outputfile',help='Output file name')
    args = parser.parse_args()

    def main(argv):
        reader = read_csv()
        if args.verbose: 
            print ('Verbose Selected')
        if args.toxml:
            if args.verbose:
               print ('Convert to XML Selected')
            generate_xml(reader)
        if args.readcsv:
            if args.verbose:
                print ('Reading CSV file')
            read_csv()
        if not (args.toxml or args.readcsv):
            parser.error('No action requested')
        return 1

    def read_csv():
        with open ('1250_12.csv', 'r') as data:
            return list(csv.reader(data))

    def generate_xml(reader):
        root = Element('Solution')
        root.set('version','1.0')
        tree = ElementTree(root)

        head = SubElement(root, 'DrillHoles')
        head.set('total_holes', '238')

        description = SubElement(head,'description')
        current_group = None
        i = 0
        for row in reader:
            if i > 0:
                x1,y1,z1,x2,y2,z2,cost = row
                if current_group is None or i != current_group.text:
                    current_group = SubElement(description, 'hole',{'hole_id':"%s"%i})

                    collar = SubElement (current_group, 'collar',{'':', '.join((x1,y1,z1))}),
                    toe = SubElement (current_group, 'toe',{'':', '.join((x2,y2,z2))})                                       
                    cost = SubElement(current_group, 'cost',{'':cost})
            i+=1    
        indent.indent(root)
        tree.write(open('hole.xml','w'))
    if (__name__ == "__main__"):

sys.exit(main(sys.argv))

对于generate_xml()函数，您可以忽略它，因为它接受以某种方式格式化的csv文件，所以您可能不理解它，但是，我认为问题在于tree.write()，因为该部分生成了xml文件写在代码本身中的名称，而不是命令提示符处的参数。

Answer 1

您需要将文件参数传递给generate_xml()。您似乎在 args.outputfile 中有输出文件。

generate_xml(reader, args.outputfile)

...
def generate_xml(reader, outfile):
    ...
    tree.write(outfile)

您可能还应该使用args.inputfile:

reader = read_csv(args.inputfile)
...
def read_csv(inputfile):
    return list(csv.reader(inputfile))

这一行没有做任何有用的事情，它处理 .csv 文件，但不对结果做任何事情:

read_csv()