Issue #23977: Tweak IDLE Delegator and its test.

This commit is contained in:
Terry Jan Reedy 2016-05-15 22:06:49 -04:00
parent 5f4ac9fcf3
commit 33a8fb9920
2 changed files with 23 additions and 12 deletions

View file

@ -1,10 +1,10 @@
class Delegator:
# The cache is only used to be able to change delegates!
def __init__(self, delegate=None):
self.delegate = delegate
self.__cache = set()
# Cache is used to only remove added attributes
# when changing the delegate.
def __getattr__(self, name):
attr = getattr(self.delegate, name) # May raise AttributeError
@ -13,6 +13,9 @@ class Delegator:
return attr
def resetcache(self):
"Removes added attributes while leaving original attributes."
# Function is really about resetting delagator dict
# to original state. Cache is just a means
for key in self.__cache:
try:
delattr(self, key)
@ -21,5 +24,10 @@ class Delegator:
self.__cache.clear()
def setdelegate(self, delegate):
"Reset attributes and change delegate."
self.resetcache()
self.delegate = delegate
if __name__ == '__main__':
from unittest import main
main('idlelib.idle_test.test_delegator', verbosity=2)