A nice class-based decorator for Python