1

コマンドプロンプトとpythonを使用してcsvファイルの形式で入力を受け取り、それを読み取ってcsvファイルと同じ名前のxmlファイルに変換する必要があります.xmlファイル拡張子を除くか、ユーザーが設定できます-o --output オプションのコマンドライン引数を使用した出力ファイル名とパス。何日もGoogleで検索してきましたが、これまでのところ私のプログラムではコマンドライン引数を入力でき、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() にあると思います。コマンド プロンプトの引数ではなく、コード自体に記述されます。

4

2 に答える 2

1

次のコードは、code.activestate.comのFB36 のレシピから変更されています。

csv ファイルには 1 つのヘッダー (最初の行) しかないはずですが、必要なことは実行され、csv ファイルのヘッダーについて心配する必要はありません。一括変換を行う場合は、このページの下部をご覧ください。

'''Convert csv to xml file

csv2xml.py takes two arguments:  
 1. csvFile: name of the csv file (may need to specify path to file)
 2. xmlFile: name of the desired xml file (path to destination can be specified)

If only the csv file is provided, its name is used for the xml file. 

Command line usage: 
 example1: python csv2xml.py 'fileName.csv' 'desiredName.xml'
 example2: python csv2xml.py '/Documents/fileName.csv' '/NewFolder/desiredName.xml'
 example3: python csv2xml.py 'fileName.csv'

This code has been adapted from: http://code.activestate.com/recipes/577423/
'''

import csv

def converter(csvFile, xmlFile):
    csvData = csv.reader(open(csvFile))

    xmlData = open(xmlFile, 'w')
    xmlData.write('<?xml version="1.0"?>' + "\n")

    # there must be only one top-level tag
    xmlData.write('<csv_data>' + "\n")

    rowNum = 0
    for row in csvData:
        if rowNum == 0:
            tags = row
            # replace spaces w/ underscores in tag names
            for i in range(len(tags)):
                tags[i] = tags[i].replace(' ', '_')
        else: 
            xmlData.write('<row>' + "\n")
            for i in range(len(tags)):
                xmlData.write('    ' + '<' + tags[i] + '>' \
                              + row[i] + '</' + tags[i] + '>' + "\n")
            xmlData.write('</row>' + "\n")

        rowNum +=1

    xmlData.write('</csv_data>' + "\n")
    xmlData.close()

## for using csv2xml.py from the command line
if __name__ == '__main__':
    import sys

    if len(sys.argv)==2:
        import os
        csvFile = sys.argv[1]
        xmlFile = os.path.splitext(csvFile)[0] + '.xml'
        converter(csvFile,xmlFile)
    elif len(sys.argv)==3:
        csvFile = sys.argv[1]
        xmlFile = sys.argv[2]   
        converter(csvFile,xmlFile)
    else:
        print __doc__
于 2013-06-10T18:07:07.380 に答える
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()
于 2013-06-10T15:35:20.223 に答える