coding:utf-8
class _SingletonWrapper:
"""
A singleton wrapper class. Its instances would be created
for each decorated class.
"""
def __init__(self, cls):
self.__wrapped__ = cls
self._instance = None
def __call__(self, *args, **kwargs):
"""Returns a single instance of decorated class"""
if self._instance is None:
self._instance = self.__wrapped__(*args, **kwargs)
return self._instance
def singleton(cls):
"""
A singleton decorator. Returns a wrapper objects. A call on that object
returns a single instance object of decorated class. Use the wrapped
attribute to access decorated class directly in unit tests
"""
return _SingletonWrapper(cls)
@singleton
class Singleton:
def init(self):
self.state = 'a'
p1 = Singleton()
p2 = Singleton()
p3 = Singleton()
print p1==p3