From ecdad8575ea0825f6cf6057906afd0fc94a375fc Mon Sep 17 00:00:00 2001 From: "Andrew M. Kuchling" Date: Tue, 7 Jun 2005 19:36:10 +0000 Subject: [PATCH] [Bug #1172763] dumbdbm uses eval() on lines, so it chokes if there's an extra \r on the end of a line; fixed by stripping off trailing whitespace. --- Lib/dumbdbm.py | 1 + Lib/test/test_dumbdbm.py | 18 ++++++++++++++++++ 2 files changed, 19 insertions(+) diff --git a/Lib/dumbdbm.py b/Lib/dumbdbm.py index b85844dc611..84a76658908 100644 --- a/Lib/dumbdbm.py +++ b/Lib/dumbdbm.py @@ -81,6 +81,7 @@ class _Database(UserDict.DictMixin): pass else: for line in f: + line = line.rstrip() key, pos_and_siz_pair = eval(line) self._index[key] = pos_and_siz_pair f.close() diff --git a/Lib/test/test_dumbdbm.py b/Lib/test/test_dumbdbm.py index 12df673eaa4..d320110b907 100644 --- a/Lib/test/test_dumbdbm.py +++ b/Lib/test/test_dumbdbm.py @@ -74,6 +74,24 @@ class DumbDBMTestCase(unittest.TestCase): self.assertEqual(f['1'], 'hello2') f.close() + def test_line_endings(self): + # test for bug #1172763: dumbdbm would die if the line endings + # weren't what was expected. + f = dumbdbm.open(_fname) + f['1'] = 'hello' + f['2'] = 'hello2' + f.close() + + # Mangle the file by adding \r before each newline + data = open(_fname + '.dir').read() + data = data.replace('\n', '\r\n') + open(_fname + '.dir', 'wb').write(data) + + f = dumbdbm.open(_fname) + self.assertEqual(f['1'], 'hello') + self.assertEqual(f['2'], 'hello2') + + def read_helper(self, f): keys = self.keys_helper(f) for key in self._dict: