Switch back to _impl and fix names

Author: oremanj

netfilterqueue/__init__.py

@@ -0,0 +1,12 @@
+from ._impl import (
+    COPY_NONE as COPY_NONE,
+    COPY_META as COPY_META,
+    COPY_PACKET as COPY_PACKET,
+    Packet as Packet,
+    NetfilterQueue as NetfilterQueue,
+    PROTOCOLS as PROTOCOLS,
+)
+from ._version import (
+    VERSION as VERSION,
+    __version__ as __version__,
+)

netfilterqueue/__init__.pxd netfilterqueue/_impl.pxdnetfilterqueue/__init__.pxd renamed to netfilterqueue/_impl.pxd

@@ -23,12 +23,12 @@ DEF SockCopySize = MaxCopySize + SockOverhead
 # Socket queue should hold max number of packets of copysize bytes
 DEF SockRcvSize = DEFAULT_MAX_QUEUELEN * SockCopySize // 2
 
-__package__ = "netfilterqueue"
-
-from ._version import __version__, VERSION
-
 from cpython.exc cimport PyErr_CheckSignals
 
+cdef extern from "Python.h":
+    ctypedef struct PyTypeObject:
+        const char* tp_name
+
 # A negative return value from this callback will stop processing and
 # make nfq_handle_packet return -1, so we use that as the error flag.
 cdef int global_callback(nfq_q_handle *qh, nfgenmsg *nfmsg,
@@ -346,6 +346,17 @@ cdef class NetfilterQueue:
             else:
                 nfq_handle_packet(self.h, buf, len(buf))
 
+cdef void _fix_names():
+    # Avoid ._impl showing up in reprs. This doesn't work on PyPy; there we would
+    # need to modify the name before PyType_Ready(), but I can't find any way to
+    # write Cython code that would execute at that time.
+    cdef PyTypeObject* tp = <PyTypeObject*>Packet
+    tp.tp_name = "netfilterqueue.Packet"
+    tp = <PyTypeObject*>NetfilterQueue
+    tp.tp_name = "netfilterqueue.NetfilterQueue"
+
+_fix_names()
+
 PROTOCOLS = {
     0: "HOPOPT",
     1: "ICMP",

netfilterqueue/__init__.pyi netfilterqueue/_impl.pyinetfilterqueue/__init__.pyi renamed to netfilterqueue/_impl.pyi

@@ -10,8 +10,8 @@
 
     ext_modules = cythonize(
         Extension(
-            "netfilterqueue.__init__",
-            ["netfilterqueue/__init__.pyx"],
+            "netfilterqueue._impl",
+            ["netfilterqueue/_impl.pyx"],
             libraries=["netfilter_queue"],
         ),
         compiler_directives={"language_level": "3str"},
@@ -23,7 +23,7 @@
         # setup_requires below.
         setup_requires = ["cython"]
     elif not os.path.exists(
-        os.path.join(os.path.dirname(__file__), "netfilterqueue/__init__.c")
+        os.path.join(os.path.dirname(__file__), "netfilterqueue/_impl.c")
     ):
         sys.stderr.write(
             "You must have Cython installed (`pip install cython`) to build this "
@@ -34,8 +34,8 @@
         sys.exit(1)
     ext_modules = [
         Extension(
-            "netfilterqueue.__init__",
-            ["netfilterqueue/__init__.c"],
+            "netfilterqueue._impl",
+            ["netfilterqueue/_impl.c"],
             libraries=["netfilter_queue"],
         )
     ]