configurator.py
61 lines
| 2.2 KiB
| text/x-python
|
PythonLexer
r7 | 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() | ||||