Запись вывода программы Python в файл .txt

Question

Я написал программу, которая читает файл CSV и выводит содержимое в качестве вставных операторов. Затем я написал программу выполнения, в которой должен взять вывод программы анализатора CSV и записать его в .txt-файл, но вместо того, чтобы писать весь вывод, он записывает только первый оператор.

Вот код для исполнителя:

import sys 

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

И код парсера:

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() 

Я не совсем уверен, что это имеет смысл, чтобы иметь их как две отдельные программы но я не мог заставить их работать как определенные функции в одной программе для жизни меня. Как я могу заставить программу выполнения выводить все строки программы парсера вместо одной строки? И как я могу объединить их в одну программу?

Answer 1

Вы делаете вещи более сложными, чем они должны быть. Просто вставьте свои открытые заявления, используя с. Одна программа. И он печатает на экране и записывает в файл.

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

Не уверен, что это сработает для вас, но вы можете использовать оператор>/>> для перенаправления stdout в файл.

Edit: Разница между> и >> >> что присоединяет к концу файла, а> обрезает файл

$python program.py >> output.txt 

Чтобы объединить программы, которые вы могли бы иметь программу ИСПОЛНИТЕЛЬ быть определен в качестве основной функции, определив его как

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

Затем вы можете перенаправить стандартный вывод в файл с помощью

sys.stdout = open("file",'w') 

Похожий вопрос: Redirect stdout to a file in Python?