Fluent interface decorators

fluent.py

#!/usr/bin/env python
"""
Decorator for fluent interface classes.
"""
import functools
import inspect


def chained(method):
    """Method decorator to allow chaining."""
    @functools.wraps(method)
    def wrapper(self, *args, **kwargs):
        result = method(self, *args, **kwargs)
        return self if result is None else result
    return wrapper


def fluent(cls):
    """Class decorator to allow method chaining."""
    for name, member in cls.__dict__.iteritems():
        if inspect.isfunction(member):
            setattr(cls, name, chained(member))
    return cls


if __name__ == '__main__':
    class Foo(object):
        @chained
        def set_value(self, value):
            self.value = value

        def get_value(self):
            return self.value

    @fluent
    class Bar(object):
        def set_value(self, value):
            self.value = value
        
        def get_value(self):
            return self.value

    print Foo().set_value("Foo").get_value()
    print Bar().set_value("Bar").get_value()

Hello. Thank you for posting these decorator definitions, they are exactly what I was looking for!

Unfortunately, PyCharm raises a bunch of errors in many use cases of these decorators, such as "unfilled parameter" when I use @chained on a method with no arguments or "Cannot find reference 'get_value' in 'None'" for your example of using the @fluent interface (This is probably because the inspector checks for errors in the wrong place).

If you know how to fix this, or a workaround I could use, that would be perfect! Otherwise, I am wondering whether I should report these issues between decorators and methods of a class.