class WhatsGoingOn:
def __init__(self, filename, fieldNames, maxLines):
self.file_to_write = filename
self.fieldNames = fieldNames'
self.maxLines = maxLines
# Open the file for reading and writing. Create it if it doesn't exist,
# and truncate it if it does.
self.file = open(self.file_to_write, 'w+b')
self.csvReader = csv.DictReader(self.file, fieldnames=self.fieldNames)
self.csvWriter = csv.DictWriter(self.file, fieldnames=self.fieldNames, extrasaction='ignore')
def looper(self):
# Infinitly (don't worry about that - this is a daemon),
# write to the file. When a certain number of lines have been written,
# read the file and then truncate it.
try:
numRowsWritten = 0
while True:
# Nevermind what's being written
self.csvWriter.writerow({'name_of_field_0': str(numRowsWritten ), 'name_of_field_1': str(numRowsWritten )})
numRowsWritten += 1
if numRowsWritten >= self.maxLines:
# Iterate through each row of the file
self.file.seek(0)
# This only works the first time...
for row in self.csvReader:
print row['name_of_field']
# Truncate the file, and restart the
self.file.truncate(0)
numRowsWritten = 0
except Exception as e:
print 'Exception!: {0}'.format(e)
self.file.close()
输出: Exception!: line contains NULL byte
for row in self.csvReader:
行第二次被命中,异常被抛出,当我查看该文件时,该文件确实在开始时有一大堆NULL字节。显然,在文件被截断后,DictWriter写了一大堆NULL字节(或者至少是我的假设)。 如何防止将NULL字节写入我的文件?
答案 0 :(得分:1)
显然,通过截断文件,你会搞乱编写器的某些内部状态。而不是截断,关闭和重新打开文件(模式w+b
截断),然后重新初始化csvReader
和csvWriter
。