feat: new primitive for weakref.__call__

Author: BobTheBuidler

mypyc/primitives/weakref_ops.py

@@ -1,6 +1,6 @@
 from mypyc.ir.ops import ERR_MAGIC
 from mypyc.ir.rtypes import object_rprimitive
-from mypyc.primitives.registry import function_op
+from mypyc.primitives.registry import ERR_NEG_INT, function_op
 
 # Weakref operations
 
@@ -11,3 +11,11 @@
     c_function_name="PyWeakref_NewRef",
     error_kind=ERR_MAGIC,
 )
+
+deref_op = function_op(
+    name="weakref.ReferenceType.__call__",
+    arg_types=[object_rprimitive],
+    return_type=object_rprimitive,
+    c_function_name="PyWeakref_GetRef",
+    error_kind=ERR_NEG_INT,
+)

mypyc/test-data/irbuild-weakref.test

@@ -2,24 +2,26 @@
 import weakref
 from typing import Any, Callable
 def f(x: object, cb: Callable[[object], Any]) -> object:
-    return weakref.ref(x, cb)
+    return weakref.ref(x, cb)()
 
 [out]
 def f(x, cb):
-    x, cb, r0 :: object
+    x, cb, r0, r1 :: object
 L0:
     r0 = PyWeakref_NewRef(x, cb)
-    return r0
+    r1 = PyWeakref_GetRef(r0)
+    return r1
 
 [case testFromWeakrefRefCallback]
 from typing import Any, Callable
 from weakref import ref
 def f(x: object, cb: Callable[[object], Any]) -> object:
-    return ref(x, cb)
+    return ref(x, cb)()
 
 [out]
 def f(x, cb):
-    x, cb, r0 :: object
+    x, cb, r0, r1 :: object
 L0:
     r0 = PyWeakref_NewRef(x, cb)
-    return r0
+    r1 = PyWeakref_GetRef(r0)
+    return r1