python-patterns/decorator.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""http://stackoverflow.com/questions/3118929/implementing-the-decorator-pattern-in-python"""


class foo_decorator(object):

    def __init__(self, decoratee):
        self._decoratee = decoratee

    def f1(self):
        print("decorated f1")
        self._decoratee.f1()

    def __getattr__(self, name):
        return getattr(self._decoratee, name)


class undecorated_foo(object):

    def f1(self):
        print("original f1")

    def f2(self):
        print("original f2")


@foo_decorator
class decorated_foo(object):

    def f1(self):
        print("original f1")

    def f2(self):
        print("original f2")


def main():
    u = undecorated_foo()
    v = foo_decorator(u)
    # The @foo_decorator syntax is just shorthand for calling
    # foo_decorator on the decorated object right after its
    # declaration.

    v.f1()
    v.f2()

if __name__ == '__main__':
    main()

### OUTPUT ###
# decorated f1
# original f1
# original f2