Expose uv_loop_t pointer for integration with other C-extensions (#310)

pranavtbhat · fantix · web-flow · commit b332eb85e461 · 2022-08-13T15:08:47.000-05:00
Co-authored-by: Fantix King &lt;fantix.king@gmail.com&gt;
diff --git a/setup.py b/setup.py
@@ -34,6 +34,7 @@
     'pycodestyle~=2.7.0',
     'pyOpenSSL~=19.0.0',
     'mypy>=0.800',
+    CYTHON_DEPENDENCY,
 ]
 
 # Dependencies required to build documentation.
diff --git a/tests/cython_helper.pyx b/tests/cython_helper.pyx
@@ -0,0 +1,5 @@
+from cpython.pycapsule cimport PyCapsule_GetPointer
+
+
+def capsule_equals(cap1, cap2):
+    return PyCapsule_GetPointer(cap1, NULL) == PyCapsule_GetPointer(cap2, NULL)
diff --git a/tests/test_pointers.py b/tests/test_pointers.py
@@ -0,0 +1,18 @@
+from uvloop import _testbase as tb
+
+
+class Test_UV_Pointers(tb.UVTestCase):
+    def test_get_uv_loop_t_ptr(self):
+        loop = self.new_loop()
+        cap1 = loop.get_uv_loop_t_ptr()
+        cap2 = loop.get_uv_loop_t_ptr()
+        cap3 = self.new_loop().get_uv_loop_t_ptr()
+
+        import pyximport
+
+        pyximport.install()
+
+        from tests import cython_helper
+
+        self.assertTrue(cython_helper.capsule_equals(cap1, cap2))
+        self.assertFalse(cython_helper.capsule_equals(cap1, cap3))
diff --git a/uvloop/loop.pyx b/uvloop/loop.pyx
@@ -38,6 +38,7 @@ from cpython cimport (
     PyBytes_AsStringAndSize,
     Py_SIZE, PyBytes_AS_STRING, PyBUF_WRITABLE
 )
+from cpython.pycapsule cimport PyCapsule_New
 
 from . import _noop
 
@@ -3180,6 +3181,10 @@ cdef class Loop:
         except Exception as ex:
             self.call_soon_threadsafe(future.set_exception, ex)
 
+    # Expose pointer for integration with other C-extensions
+    def get_uv_loop_t_ptr(self):
+        return PyCapsule_New(<void *>self.uvloop, NULL, NULL)
+
 
 cdef void __loop_alloc_buffer(uv.uv_handle_t* uvhandle,
                               size_t suggested_size,

Original file line number	Diff line number	Diff line change
`@@ -34,6 +34,7 @@`
`34`	`34`	`'pycodestyle~=2.7.0',`
`35`	`35`	`'pyOpenSSL~=19.0.0',`
`36`	`36`	`'mypy>=0.800',`
	`37`	`+ CYTHON_DEPENDENCY,`
`37`	`38`	`]`
`38`	`39`
`39`	`40`	`# Dependencies required to build documentation.`