Diese Frage wurde schon lange beantwortet, aber ich dachte, ich würde meine Decorator-Klasse mit Ihnen teilen, die das Schreiben neuer Dekoratoren einfach und kompakt macht.
from abc import ABCMeta, abstractclassmethod
class Decorator(metaclass=ABCMeta):
""" Acts as a base class for all decorators """
def __init__(self):
self.method = None
def __call__(self, method):
self.method = method
return self.call
@abstractclassmethod
def call(self, *args, **kwargs):
return self.method(*args, **kwargs)
Zum einen denke ich, dass dies das Verhalten der Dekoratoren sehr klar macht, aber es macht es auch einfach, neue Dekoratoren sehr prägnant zu definieren. Für das oben aufgeführte Beispiel könnte man es dann wie folgt lösen:
class MakeBold(Decorator):
def call():
return "<b>" + self.method() + "</b>"
class MakeItalic(Decorator):
def call():
return "<i>" + self.method() + "</i>"
@MakeBold()
@MakeItalic()
def say():
return "Hello"
Sie können es auch für komplexere Aufgaben verwenden, wie zum Beispiel einen Dekorator, der die Funktion automatisch rekursiv auf alle Argumente in einem Iterator anwendet:
class ApplyRecursive(Decorator):
def __init__(self, *types):
super().__init__()
if not len(types):
types = (dict, list, tuple, set)
self._types = types
def call(self, arg):
if dict in self._types and isinstance(arg, dict):
return {key: self.call(value) for key, value in arg.items()}
if set in self._types and isinstance(arg, set):
return set(self.call(value) for value in arg)
if tuple in self._types and isinstance(arg, tuple):
return tuple(self.call(value) for value in arg)
if list in self._types and isinstance(arg, list):
return list(self.call(value) for value in arg)
return self.method(arg)
@ApplyRecursive(tuple, set, dict)
def double(arg):
return 2*arg
print(double(1))
print(double({'a': 1, 'b': 2}))
print(double({1, 2, 3}))
print(double((1, 2, 3, 4)))
print(double([1, 2, 3, 4, 5]))
Welche Drucke:
2
{'a': 2, 'b': 4}
{2, 4, 6}
(2, 4, 6, 8)
[1, 2, 3, 4, 5, 1, 2, 3, 4, 5]
Beachten Sie, dass dieses Beispiel nicht die list Typ in der Instanziierung des Dekorators, so dass in der abschließenden Druckanweisung die Methode auf die Liste selbst und nicht auf die Elemente der Liste angewendet wird.
