From: Stefan Behnel <scoder@users.berlios.de>
Date: Sat, 30 Oct 2010 17:04:39 +0000 (+0200)
Subject: support for 'final' cdef types using a directive decorator
X-Git-Tag: 0.14.alpha0~280
X-Git-Url: http://git.tremily.us/?a=commitdiff_plain;h=9b57b1c56271b06108d92cc78f623c1c9b16a12a;p=cython.git

support for 'final' cdef types using a directive decorator
---

diff --git a/Cython/Compiler/Options.py b/Cython/Compiler/Options.py
index ccc5c0f9..fd7666ac 100644
--- a/Cython/Compiler/Options.py
+++ b/Cython/Compiler/Options.py
@@ -81,6 +81,7 @@ directive_defaults = {
 
 # Override types possibilities above, if needed
 directive_types = {
+    'final' : bool,  # final cdef classes and methods
     'infer_types' : bool, # values can be True/None/False
     }
 
@@ -90,6 +91,7 @@ for key, val in directive_defaults.items():
 
 directive_scopes = { # defaults to available everywhere
     # 'module', 'function', 'class', 'with statement'
+    'final' : ('cclass',),   # add 'method' in the future
     'autotestdict' : ('module',),
     'test_assert_path_exists' : ('function',),
     'test_fail_if_path_exists' : ('function',),
diff --git a/Cython/Compiler/TypeSlots.py b/Cython/Compiler/TypeSlots.py
index b4dbdda9..6e1b1131 100644
--- a/Cython/Compiler/TypeSlots.py
+++ b/Cython/Compiler/TypeSlots.py
@@ -323,7 +323,9 @@ class TypeFlagsSlot(SlotDescriptor):
     #  Descriptor for the type flags slot.
     
     def slot_code(self, scope):
-        value = "Py_TPFLAGS_DEFAULT|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_BASETYPE|Py_TPFLAGS_HAVE_NEWBUFFER"
+        value = "Py_TPFLAGS_DEFAULT|Py_TPFLAGS_CHECKTYPES|Py_TPFLAGS_HAVE_NEWBUFFER"
+        if not scope.directives.get('final', False):
+            value += "|Py_TPFLAGS_BASETYPE"
         if scope.needs_gc():
             value += "|Py_TPFLAGS_HAVE_GC"
         return value
diff --git a/tests/run/final_cdef_class.pyx b/tests/run/final_cdef_class.pyx
new file mode 100644
index 00000000..6845c71e
--- /dev/null
+++ b/tests/run/final_cdef_class.pyx
@@ -0,0 +1,43 @@
+
+cimport cython
+
+@cython.final
+cdef class FinalClass:
+    """
+    >>> f = FinalClass()
+    >>> test_final_class(f)
+    Type tested
+
+    >>> try:
+    ...     class SubType(FinalClass): pass
+    ... except TypeError:
+    ...     print 'PASSED!'
+    PASSED!
+    """
+
+cdef class NonFinalClass:
+    """
+    >>> class SubType(NonFinalClass): pass
+    >>> s = SubType()
+    """
+
+@cython.final
+cdef class FinalSubClass(NonFinalClass):
+    """
+    >>> f = FinalSubClass()
+    >>> test_non_final_class(f)
+    Type tested
+
+    >>> try:
+    ...     class SubType(FinalSubClass): pass
+    ... except TypeError:
+    ...     print 'PASSED!'
+    PASSED!
+    """
+
+
+def test_final_class(FinalClass c):
+    print u"Type tested"
+
+def test_non_final_class(NonFinalClass c):
+    print u"Type tested"