Python:为什么DictWriter写'NULL'字节?

时间:2013-02-25 23:40:01

标签: python dictionary null python-2.6

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字节写入我的文件?

1 个答案:

答案 0 :(得分:1)

显然,通过截断文件,你会搞乱编写器的某些内部状态。而不是截断,关闭和重新打开文件(模式w+b截断),然后重新初始化csvReadercsvWriter