try:
    # python2
    string = (str, unicode)
except NameError:
    # python3
    string = (str,)


class fileobj(object):
    """
    fileobj decorator:
    inner function may take a file-like object or a string as path to file
    The decorated function must take this as the first argument
    """


    def __init__(self, function):
        self.wrapped = function

    def __call__(self, fp, *args, **kwargs):

        if isinstance(fp, string):
            with open(fp) as f:
                return self.wrapped(f, *args, **kwargs)

        return self.wrapped(fp, *args, **kwargs)