feat: new primitives for weakref.proxy

Author: BobTheBuidler

mypyc/primitives/weakref_ops.py

@@ -20,3 +20,20 @@
     c_function_name="PyWeakref_NewRef",
     error_kind=ERR_MAGIC,
 )
+
+new_proxy_op = function_op(
+    name="_weakref.proxy",
+    arg_types=[object_rprimitive],
+    return_type=object_rprimitive,
+    c_function_name="PyWeakref_NewProxy",
+    extra_int_constants=[(0, pointer_rprimitive)],
+    error_kind=ERR_MAGIC,
+)
+
+new_proxy_with_callback_op = function_op(
+    name="_weakref.proxy",
+    arg_types=[object_rprimitive, object_rprimitive],
+    return_type=object_rprimitive,
+    c_function_name="PyWeakref_NewProxy",
+    error_kind=ERR_MAGIC,
+)

mypyc/test-data/run-weakref.test

@@ -2,13 +2,14 @@
 
 [case testWeakrefRef]
 import pytest  # type: ignore [import-not-found]
-from weakref import ref
+from weakref import proxy, ref
 from mypy_extensions import mypyc_attr
 
 @mypyc_attr(native_class=False)
 class Object:
     """some random weakreffable object"""
-    pass
+    def some_meth(self) -> int:
+        return 1
 
 def test_weakref_ref():
     obj = Object()
@@ -24,8 +25,31 @@ def test_weakref_ref_with_callback():
     obj = None
     assert r() is None, r()
 
+def test_weakref_proxy():
+    obj = Object()
+    p = proxy(obj)
+    assert p.some_meth() == 1
+    obj = None
+    with pytest.raises(ReferenceError):
+        p.some_meth()
+
+def test_weakref_proxy_with_callback():
+    obj = Object()
+    p = proxy(obj, lambda x: x)
+    assert p.some_meth() == 1
+    obj = None
+    with pytest.raises(ReferenceError):
+        p.some_meth()
+
 [file driver.py]
-from native import test_weakref_ref, test_weakref_ref_with_callback
+from native import (
+    test_weakref_ref,
+    test_weakref_ref_with_callback,
+    test_weakref_proxy,
+    test_weakref_proxy_with_callback,
+)
 
 test_weakref_ref()
 test_weakref_ref_with_callback()
+test_weakref_proxy()
+test_weakref_proxy_with_callback()