Kilka opcji:
(1) Pracochłonnie wykonaj mapowanie tożsamości (tj. Nic nie rób) z nazw pól, aby csv.DictWriter mógł przekształcić je z powrotem w listę i przekazać do instancji csv.writer.
(2) Dokumentacja wspomina o „podstawowej writer
instancji”… więc po prostu go użyj (przykład na końcu).
dw.writer.writerow(dw.fieldnames)
(3) Unikaj narzutu csv.Dictwriter i zrób to sam z csv.writer
Zapisywanie danych:
w.writerow([d[k] for k in fieldnames])
lub
w.writerow([d.get(k, restval) for k in fieldnames])
Zamiast extrasaction
„funkcjonalności” wolałbym sam go zakodować; w ten sposób możesz zgłosić WSZYSTKIE „dodatki” za pomocą kluczy i wartości, a nie tylko pierwszego dodatkowego klucza. To, co jest naprawdę uciążliwe w przypadku DictWriter, polega na tym, że jeśli samodzielnie zweryfikowałeś klucze podczas budowania każdego dyktu, musisz pamiętać, aby użyć extrasaction = 'ignore', w przeciwnym razie nastąpi POWOLI (nazwy pól to lista) powtórz sprawdzanie:
wrong_fields = [k for k in rowdict if k not in self.fieldnames]
============
>>> f = open('csvtest.csv', 'wb')
>>> import csv
>>> fns = 'foo bar zot'.split()
>>> dw = csv.DictWriter(f, fns, restval='Huh?')
# dw.writefieldnames(fns) -- no such animal
>>> dw.writerow(fns) # no such luck, it can't imagine what to do with a list
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "C:\python26\lib\csv.py", line 144, in writerow
return self.writer.writerow(self._dict_to_list(rowdict))
File "C:\python26\lib\csv.py", line 141, in _dict_to_list
return [rowdict.get(key, self.restval) for key in self.fieldnames]
AttributeError: 'list' object has no attribute 'get'
>>> dir(dw)
['__doc__', '__init__', '__module__', '_dict_to_list', 'extrasaction', 'fieldnam
es', 'restval', 'writer', 'writerow', 'writerows']
# eureka
>>> dw.writer.writerow(dw.fieldnames)
>>> dw.writerow({'foo':'oof'})
>>> f.close()
>>> open('csvtest.csv', 'rb').read()
'foo,bar,zot\r\noof,Huh?,Huh?\r\n'
>>>
dw.writerow( dict((f,f) for f in dr.fieldnames) )
.