diff options
author | Martin Panter <vadmium+py@gmail.com> | 2015-11-13 22:47:00 (GMT) |
---|---|---|
committer | Martin Panter <vadmium+py@gmail.com> | 2015-11-13 22:47:00 (GMT) |
commit | 06622ead8072f3602bba8cd1924f0897873ad8b1 (patch) | |
tree | 747cf68e713427c3d494083981fe91c4de6f573e /Lib | |
parent | 1bb651540e0743c4e58d875d1de240597862fa34 (diff) | |
download | cpython-06622ead8072f3602bba8cd1924f0897873ad8b1.zip cpython-06622ead8072f3602bba8cd1924f0897873ad8b1.tar.gz cpython-06622ead8072f3602bba8cd1924f0897873ad8b1.tar.bz2 |
Issue #25590: Make rlcompleter only call getattr() once per attribute
Previously it was called another time via hasattr(), and both calls were
made once for dir(f) and again for dir(f.__class__). This includes a
backport of changing from a list to a set from revision 4dbb315fe667.
Diffstat (limited to 'Lib')
-rw-r--r-- | Lib/rlcompleter.py | 17 | ||||
-rw-r--r-- | Lib/test/test_rlcompleter.py | 13 |
2 files changed, 23 insertions, 7 deletions
diff --git a/Lib/rlcompleter.py b/Lib/rlcompleter.py index d517c0e..be8aee0 100644 --- a/Lib/rlcompleter.py +++ b/Lib/rlcompleter.py @@ -136,20 +136,23 @@ class Completer: return [] # get the content of the object, except __builtins__ - words = dir(thisobject) - if "__builtins__" in words: - words.remove("__builtins__") + words = set(dir(thisobject)) + words.discard("__builtins__") if hasattr(thisobject, '__class__'): - words.append('__class__') - words.extend(get_class_members(thisobject.__class__)) + words.add('__class__') + words.update(get_class_members(thisobject.__class__)) matches = [] n = len(attr) for word in words: - if word[:n] == attr and hasattr(thisobject, word): - val = getattr(thisobject, word) + if word[:n] == attr: + try: + val = getattr(thisobject, word) + except Exception: + continue # Exclude properties that are not set word = self._callable_postfix(val, "%s.%s" % (expr, word)) matches.append(word) + matches.sort() return matches def get_class_members(klass): diff --git a/Lib/test/test_rlcompleter.py b/Lib/test/test_rlcompleter.py index 2da7fce..927df34 100644 --- a/Lib/test/test_rlcompleter.py +++ b/Lib/test/test_rlcompleter.py @@ -64,6 +64,19 @@ class TestRlcompleter(unittest.TestCase): ['egg.{}('.format(x) for x in dir(str) if x.startswith('s')]) + def test_excessive_getattr(self): + # Ensure getattr() is invoked no more than once per attribute + class Foo: + calls = 0 + @property + def bar(self): + self.calls += 1 + return None + f = Foo() + completer = rlcompleter.Completer(dict(f=f)) + self.assertEqual(completer.complete('f.b', 0), 'f.bar') + self.assertEqual(f.calls, 1) + def test_complete(self): completer = rlcompleter.Completer() self.assertEqual(completer.complete('', 0), '\t') |