2014-11-01 2 views
0

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

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

+0

Да, конечно, но это не служба написания кода; что вы пробовали, и в чем именно проблема? – jonrsharpe

+0

Возможный дубликат [python, как проверить файл пустым или нет] (http://stackoverflow.com/questions/2507808/python-how-to-check-file-empty-or-not) –

+0

Я попробую посмотреть, файл пуст. Я этого не сделал, и я посмотрел и попытался написать код, если там, где нет строк, но которые, похоже, не работают –

ответ

0

Просто используйте флаг:

headers_written = False 

затем при написании строки:

if not headers_written: 
    writer.writerow(headers) 
    headers_written = True 

writer.writerow(somerow) 

Вы бы на самом деле отложить создание писателя, пока вы не уверены, что у вас есть вещи, чтобы написать:

writer = None 

# ... 

if not writer: 
    writer = csv.writer(open(filename, 'wb')) 
    writer.writerow(headers) 

writer.writerow(somerow) 
Смежные вопросы