我需要接受一些命令行参数,比如:
argparse.py hi.xml --> to produce hi.csv or
argparse.py hi.xml -o hello.csv --> to produce hello.csv
这是我的代码:
import os
import sys
import argparse
import csv
import indent
from xml.etree.ElementTree import ElementTree, Element, SubElement, Comment, tostring
import xml.etree.ElementTree as etree
def get_args(args):
parser = argparse.ArgumentParser(description = "Converts CSV to XML")
parser.add_argument('-v','--verbose',action='store_true',dest='verbose',help='Increases messages being printed to stdout')
parser.add_argument("inputfile", help="Please input the name of the CSV file")
parser.add_argument('-o','--outputfile',help='(optional) Output file name',nargs='?')
args = parser.parse_args()
ext = os.path.splitext(args.inputfile)[1].lower()
if args.outputfile is None:
if ext == ".csv":
args.outputfile = os.path.splitext(args.inputfile)[0] + '.xml'
elif ext == ".xml":
args.outputfile = os.path.splitext(args.inputfile)[0] + '.csv'
elif args.outputfile:
if ext == ".csv":
outputfile = open(args.outputfile, 'w')
elif ext == ".xml":
outputfile = open(args.outputfile,'w')
else:
sys.stderr.write('ERROR: Invalid extension %sn' % ext)
sys.exit(1)
return args
def main(argv):
args = get_args(argv[0:])
if args is None:
return 1
ext = os.path.splitext(args.inputfile)[1].lower()
if ext == ".csv":
reader = read_csv(open(args.inputfile))
generate_xml(reader, args.outputfile)
if ext == ".xml":
root = etree.parse(open(args.inputfile)).getroot()
generate_csv(root, args.outputfile)
if args.verbose:
print ('Verbose Selected')
if args.verbose:
print ('Convert to XML with set name')
return 0
def read_csv(inputfile):
return list(csv.reader(inputfile))
def generate_xml(reader,outfile):
root = Element('Solution')
root.set('version','1.0')
tree = ElementTree(root)
head = SubElement(root, 'DrillHoles')
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
head.set('total_holes', '%s'%i)
indent.indent(root)
tree.write(outfile)
def generate_csv(root, outfile):
with open(outfile, 'w') as file_:
writer = csv.writer(file_, delimiter="t")
for a in zip(root.findall("drillholes/hole/collar"),
root.findall("drillholes/hole/toe"),
root.findall("drillholes/hole/cost")):
writer.writerow([x.text for x in a])
if (__name__ == "__main__"):
sys.exit(main(sys.argv))
我只需要generate.csv部分的帮助。正如您从函数get_args(args)中看到的那样,我获得了参数,这也是我使用ext=…更改文件扩展名的地方。。。。然后我使用返回参数返回它。主函数调用这个函数并获取参数,然后再次检查扩展名,然后根据ext.找到下一个函数。在主函数中,我已经解析了xml文件,所以我只将"root"one_answers"args.outputfile"传递给generate.csv方法。然而,这就是错误发生的地方:xml.etree.ElementTree.parserError:在第1行第0列没有找到元素。然而,我相信这可能不是唯一的错误,也许是我传递参数的方式丢失或修改了。请帮助
这并不是一个最小的例子,所以很难准确地计算出你的问题是什么以及具体的问题是谁
错误表明解析器没有得到任何输入。这表明文件实际上是空的,或者文件指针已经前进到文件的末尾。如果不小心打开该文件进行写入并将其截断为零,则该文件可能为空。如果您不小心打开文件进行读取,读取了其中的所有内容,并且忽略了倒带文件指针,则文件指针可能会前进到文件的末尾。
所以你可以看到发生了什么,为什么不尝试一些诊断方法,在两者之间迈出小步并打印出来呢?也就是说,不要使用etree.parse(open(args.inputfile))
,而是尝试以下操作:
print( args.inputfile )
f = open( args.inputfile )
print( f.tell() )
print( f.read() )
f.seek(0)
xml = etree.parse( f )
print( xml )
root = xml.getroot()
print( root )
这样,你就可以看到发生了什么。