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

4

1 回答 1

1

显然，通过截断文件，您会弄乱编写器的某些内部状态。而不是截断，关闭并重新打开文件（模式w+b截断），然后重新初始化csvReaderand csvWriter。

于 2013-02-26T16:32:32.980 回答