python · JukkaL · Aug 3, 2020 · Aug 2, 2020
diff --git a/mypyc/codegen/emitclass.py b/mypyc/codegen/emitclass.py
@@ -129,8 +129,10 @@ def generate_class(cl: ClassIR, module: str, emitter: Emitter) -> None:
     generate_full = not cl.is_trait and not cl.builtin_base
     needs_getseters = not cl.is_generated
 
-    if generate_full:
+    if not cl.builtin_base:
         fields['tp_new'] = new_name
+
+    if generate_full:
         fields['tp_dealloc'] = '(destructor){}_dealloc'.format(name_prefix)
         fields['tp_traverse'] = '(traverseproc){}_traverse'.format(name_prefix)
         fields['tp_clear'] = '(inquiry){}_clear'.format(name_prefix)
@@ -229,6 +231,10 @@ def emit_line() -> None:
         emit_line()
         generate_getseters_table(cl, getseters_name, emitter)
         emit_line()
+
+    if cl.is_trait:
+        generate_new_for_trait(cl, new_name, emitter)
+
     generate_methods_table(cl, methods_name, emitter)
     emit_line()
 
@@ -545,6 +551,27 @@ def generate_new_for_class(cl: ClassIR,
     emitter.emit_line('}')
 
 
+def generate_new_for_trait(cl: ClassIR,
+                           func_name: str,
+                           emitter: Emitter) -> None:
+    emitter.emit_line('static PyObject *')
+    emitter.emit_line(
+        '{}(PyTypeObject *type, PyObject *args, PyObject *kwds)'.format(func_name))
+    emitter.emit_line('{')
+    emitter.emit_line('if (type != {}) {{'.format(emitter.type_struct_name(cl)))
+    emitter.emit_line(
+        'PyErr_SetString(PyExc_TypeError, '
+        '"interpreted classes cannot inherit from compiled traits");'
+    )
+    emitter.emit_line('} else {')
+    emitter.emit_line(
+        'PyErr_SetString(PyExc_TypeError, "traits may not be directly created");'
+    )
+    emitter.emit_line('}')
+    emitter.emit_line('return NULL;')
+    emitter.emit_line('}')
+
+
 def generate_traverse_for_class(cl: ClassIR,
                                 func_name: str,
                                 emitter: Emitter) -> None:

diff --git a/mypyc/test-data/run-traits.test b/mypyc/test-data/run-traits.test
@@ -383,3 +383,29 @@ g(c1, c2)
 assert c1.x == 1
 assert c2.x == 2
 [out]
+
+[case testTraitErrorMessages]
+from mypy_extensions import trait
+
+@trait
+class Trait:
+    pass
+
+def create() -> Trait:
+    return Trait()
+
+[file driver.py]
+from native import Trait, create
+from testutil import assertRaises
+
+with assertRaises(TypeError, "traits may not be directly created"):
+    Trait()
+
+with assertRaises(TypeError, "traits may not be directly created"):
+    create()
+
+class Sub(Trait):
+    pass
+
+with assertRaises(TypeError, "interpreted classes cannot inherit from compiled traits"):
+    Sub()