import inspect

from pyramid.config import Configurator


class InspectProxy(object):
    """
    Proxy to the `inspect` module that allows us to use the pyramid include
    mechanism for cythonized modules without source file.
    """

    def _get_cyfunction_func_code(self, cyfunction):
        """
        Unpack the `func_code` attribute of a cython function.
        """
        if inspect.ismethod(cyfunction):
            cyfunction = cyfunction.im_func
        return getattr(cyfunction, 'func_code')

    def getmodule(self, *args, **kwds):
        """
        Simple proxy to `inspect.getmodule`.
        """
        return inspect.getmodule(*args, **kwds)

    def getsourcefile(self, obj):
        """
        Proxy to `inspect.getsourcefile` or `inspect.getfile` depending on if
        it's called to look up the source file that contains the magic pyramid
        `includeme` callable.

        For cythonized modules the source file may be deleted. Therefore we
        return the result of `inspect.getfile` instead. In the case of the
        `configurator.include` method this is OK, because the result is passed
        to `os.path.dirname` which strips the file name. So it doesn't matter
        if we return the path to the source file or another file in the same
        directory.
        """
        # Check if it's called to look up the source file that contains the
        # magic pyramid `includeme` callable.
        if getattr(obj, '__name__') == 'includeme':
            try:
                return inspect.getfile(obj)
            except TypeError as e:
                # Cython functions are not recognized as functions by the
                # inspect module. We have to unpack the func_code attribute
                # ourself.
                if 'cyfunction' in e.message:
                    obj = self._get_cyfunction_func_code(obj)
                    return inspect.getfile(obj)
                raise
        else:
            return inspect.getsourcefile(obj)


class CythonCompatConfigurator(Configurator):
    """
    Customized configurator to replace the inspect class attribute with
    a custom one that is cython compatible.
    """
    inspect = InspectProxy()