diff options
author | Skip Montanaro <skip@pobox.com> | 2008-08-08 22:52:51 (GMT) |
---|---|---|
committer | Skip Montanaro <skip@pobox.com> | 2008-08-08 22:52:51 (GMT) |
commit | a032bf41f6a23856f92690c0cd14520d31214e7e (patch) | |
tree | ebe930c281477abc06d382170c66488fbeb91d51 /Lib/csv.py | |
parent | b04d4853a7d76ace2e78de20a7155a9b5a11ce55 (diff) | |
download | cpython-a032bf41f6a23856f92690c0cd14520d31214e7e.zip cpython-a032bf41f6a23856f92690c0cd14520d31214e7e.tar.gz cpython-a032bf41f6a23856f92690c0cd14520d31214e7e.tar.bz2 |
accept issue 3436
Diffstat (limited to 'Lib/csv.py')
-rw-r--r-- | Lib/csv.py | 22 |
1 files changed, 18 insertions, 4 deletions
@@ -71,7 +71,7 @@ register_dialect("excel-tab", excel_tab) class DictReader: def __init__(self, f, fieldnames=None, restkey=None, restval=None, dialect="excel", *args, **kwds): - self.fieldnames = fieldnames # list of keys for the dict + self._fieldnames = fieldnames # list of keys for the dict self.restkey = restkey # key to catch long rows self.restval = restval # default value for short rows self.reader = reader(f, dialect, *args, **kwds) @@ -81,11 +81,25 @@ class DictReader: def __iter__(self): return self + @property + def fieldnames(self): + if self._fieldnames is None: + try: + self._fieldnames = self.reader.next() + except StopIteration: + pass + self.line_num = self.reader.line_num + return self._fieldnames + + @fieldnames.setter + def fieldnames(self, value): + self._fieldnames = value + def next(self): + if self.line_num == 0: + # Used only for its side effect. + self.fieldnames row = self.reader.next() - if self.fieldnames is None: - self.fieldnames = row - row = self.reader.next() self.line_num = self.reader.line_num # unlike the basic reader, we prefer not to return blanks, |