Ausgabe von Python-Programm in TXT-Datei schreiben

Question

Ich habe ein Programm geschrieben, das eine CSV-Datei liest und den Inhalt als INSERT-Anweisungen ausgibt. Ich schrieb dann ein Ausführungsprogramm, das die Ausgabe des CSV-Parser-Programms nehmen und in eine TXT-Datei schreiben sollte, aber anstatt die gesamte Ausgabe zu schreiben, schreibt es nur die erste Anweisung. Hier

ist der Code für den Vollstrecker:

import sys 

with open('insert.txt', 'wb') as f: 
    subprocess.check_call(["python", "CSVParserMultiple.py"], stdout=f) 

Und der Code für den Parser:

import csv, os 

path = 'C:/Users/user/Desktop/test/' 
for file in os.listdir(path): 
    if file.endswith('.csv'): 
# print ('Parsing file: ' + os.path.basename(path + file)) 
     openFile = open(path + file) 
     csvFile = csv.reader(openFile) 
     getHeader = next(csvFile) 
     columnHeaders = map((lambda x: "'" + x + "'"), getHeader[:-1]) 
     insert = 'INSERT INTO testing (' + "'ID', 'VehicleID', " + ', '.join(columnHeaders) + ') VALUES ' 
     for row in csvFile: 
      values = map((lambda x: "'" + x.strip() + "'"), row[:-1]) 
      print (insert + "(" + ", ".join(values) + ");") 
     openFile.close() 

ich nicht ganz sicher bin, dass es sinnvoll, sie als zwei separate Programme zu haben, macht aber ich konnte sie nicht dazu bringen, als definierte Funktionen in demselben Programm für das Leben von mir zu laufen. Wie kann ich das Ausführungsprogramm dazu bringen, alle Zeilen des Parser-Programms anstatt nur einer einzigen Zeile auszugeben? Und wie kann ich sie zu einem Programm kombinieren?

Answer 1

Sie machen die Dinge komplizierter als sie sein müssen. Verschachteln Sie einfach Ihre offenen Anweisungen mit mit. Ein Programm. Und es druckt auf dem Bildschirm und schreibt in eine Datei.

import csv, os 

path = 'C:/Users/user/Desktop/test/' 
for file in os.listdir(path): 
    if file.endswith('.csv'): 
     # print ('Parsing file: ' + os.path.basename(path + file)) 
     with open(path + file) as infile: 
      with open(path+file+".txt",'w') as outfile: 
       csvFile = csv.reader(infile) 
       getHeader = next(csvFile) 
       columnHeaders = map((lambda x: "'" + x + "'"), getHeader[:-1]) 
       insert = 'INSERT INTO testing (' + "'ID', 'VehicleID', " + ', '.join(columnHeaders) + ') VALUES ' 
       for row in csvFile: 
        values = map((lambda x: "'" + x.strip() + "'"), row[:-1]) 
        print (insert + "(" + ", ".join(values) + ");") 
        outfile.write(insert + "(" + ", ".join(values) + ");" + "\n") 

Answer 2

Nicht sicher, ob dies für Sie funktionieren würde, aber Sie können den Operator>/>> verwenden, um die Standardausgabe in eine Datei umzuleiten.

Edit: Der Unterschied zwischen> und >> ist, dass >> zum Ende der Datei anhängt, während> die Datei kürzt

$python program.py >> output.txt 

die Programme kombinieren Sie die Testamentsvollstrecker Programm haben könnte definiert werden, als Hauptfunktion, indem es als

def main(): 
    <executor code here> 
if __name__ == "__main__": 
    main() 

Dann definieren Sie stdout in eine Datei mithilfe von

umleiten

ähnliche Frage: Redirect stdout to a file in Python?