summaryrefslogtreecommitdiffstats
path: root/Lib
diff options
context:
space:
mode:
authorMartin Panter <vadmium+py@gmail.com>2015-11-13 22:47:00 (GMT)
committerMartin Panter <vadmium+py@gmail.com>2015-11-13 22:47:00 (GMT)
commit06622ead8072f3602bba8cd1924f0897873ad8b1 (patch)
tree747cf68e713427c3d494083981fe91c4de6f573e /Lib
parent1bb651540e0743c4e58d875d1de240597862fa34 (diff)
downloadcpython-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.py17
-rw-r--r--Lib/test/test_rlcompleter.py13
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')