openmedialibrary_platform_d.../lib/python3.7/idlelib/delegator.py

34 lines
1 KiB
Python
Raw Normal View History

2016-02-06 09:36:57 +00:00
class Delegator:
def __init__(self, delegate=None):
self.delegate = delegate
self.__cache = set()
2018-12-31 23:25:26 +00:00
# Cache is used to only remove added attributes
# when changing the delegate.
2016-02-06 09:36:57 +00:00
def __getattr__(self, name):
attr = getattr(self.delegate, name) # May raise AttributeError
setattr(self, name, attr)
self.__cache.add(name)
return attr
def resetcache(self):
2018-12-31 23:25:26 +00:00
"Removes added attributes while leaving original attributes."
# Function is really about resetting delagator dict
# to original state. Cache is just a means
2016-02-06 09:36:57 +00:00
for key in self.__cache:
try:
delattr(self, key)
except AttributeError:
pass
self.__cache.clear()
def setdelegate(self, delegate):
2018-12-31 23:25:26 +00:00
"Reset attributes and change delegate."
2016-02-06 09:36:57 +00:00
self.resetcache()
self.delegate = delegate
2018-12-31 23:25:26 +00:00
if __name__ == '__main__':
from unittest import main
main('idlelib.idle_test.test_delegator', verbosity=2)