Commit c1690bf3 authored by Stefan Behnel's avatar Stefan Behnel

support @cython.final decorator in override .pxd files

parent 2f609c41
...@@ -909,16 +909,17 @@ class InterpretCompilerDirectives(CythonTransform, SkipDeclarations): ...@@ -909,16 +909,17 @@ class InterpretCompilerDirectives(CythonTransform, SkipDeclarations):
return self.visit_with_directives(body, directives) return self.visit_with_directives(body, directives)
def visit_CVarDefNode(self, node): def visit_CVarDefNode(self, node):
if not node.decorators: directives = self._extract_directives(node, 'function')
if not directives:
return node return node
for dec in node.decorators: for name, value in directives.iteritems():
for directive in self.try_to_parse_directives(dec.decorator) or (): if name == 'locals':
if directive is not None and directive[0] == u'locals': node.directive_locals = value
node.directive_locals = directive[1] elif name != 'final':
else:
self.context.nonfatal_error(PostParseError(dec.pos, self.context.nonfatal_error(PostParseError(dec.pos,
"Cdef functions can only take cython.locals() decorator.")) "Cdef functions can only take cython.locals() or final decorators, got %s." % name))
return node body = Nodes.StatListNode(node.pos, stats=[node])
return self.visit_with_directives(body, directives)
def visit_CClassDefNode(self, node): def visit_CClassDefNode(self, node):
directives = self._extract_directives(node, 'cclass') directives = self._extract_directives(node, 'cclass')
...@@ -948,7 +949,7 @@ class InterpretCompilerDirectives(CythonTransform, SkipDeclarations): ...@@ -948,7 +949,7 @@ class InterpretCompilerDirectives(CythonTransform, SkipDeclarations):
directives.append(directive) directives.append(directive)
else: else:
realdecs.append(dec) realdecs.append(dec)
if realdecs and isinstance(node, (Nodes.CFuncDefNode, Nodes.CClassDefNode)): if realdecs and isinstance(node, (Nodes.CFuncDefNode, Nodes.CClassDefNode, Nodes.CVarDefNode)):
raise PostParseError(realdecs[0].pos, "Cdef functions/classes cannot take arbitrary decorators.") raise PostParseError(realdecs[0].pos, "Cdef functions/classes cannot take arbitrary decorators.")
else: else:
node.decorators = realdecs node.decorators = realdecs
......
PYTHON -c "import a; assert a.__file__.rstrip('co').endswith('.py'), a.__file__; a.test()"
PYTHON setup.py build_ext --inplace
PYTHON -c "import a; assert not a.__file__.rstrip('co').endswith('.py'), a.__file__; a.test()"
######## setup.py ########
from Cython.Build.Dependencies import cythonize
from distutils.core import setup
setup(
ext_modules = cythonize("a.py"),
)
######## a.pxd ########
cimport cython
cdef class ExtType:
@cython.final
cdef int final_func(self)
@cython.final
cdef class FinalExtType:
cdef int func(self)
@cython.final
cdef class FinalExtSubType(ExtType):
cdef int func(self)
cdef class NonFinalSubType(ExtType):
cdef int func(self)
######## a.py ########
import cython
class ExtType(object):
@cython.test_assert_path_exists("//CFuncDefNode[@entry.is_final_cmethod=True]")
def final_func(self):
return 1
class FinalExtType(object):
@cython.test_assert_path_exists("//CFuncDefNode[@entry.is_final_cmethod=True]")
def func(self):
return 2
class FinalExtSubType(ExtType):
@cython.test_assert_path_exists("//CFuncDefNode[@entry.is_final_cmethod=True]")
def func(self):
return 3
class NonFinalSubType(ExtType):
@cython.test_assert_path_exists("//CFuncDefNode[@entry.is_final_cmethod=True]")
@cython.final
def func(self):
return 4
def test():
assert ExtType().final_func() == 1
assert FinalExtSubType().final_func() == 1
assert NonFinalSubType().final_func() == 1
assert FinalExtType().func() == 2
assert FinalExtSubType().func() == 3
assert NonFinalSubType().func() == 4
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment