Speed up noise frame helper (#693)

2023-11-24 12:04:12 -06:00 · 2023-11-24 12:04:12 -06:00 · 33d1d3d8c4
parent aa3b8af246
commit 33d1d3d8c4
1 changed files with 6 additions and 5 deletions
--- a/aioesphomeapi/_frame_helper/noise.pxd
+++ b/aioesphomeapi/_frame_helper/noise.pxd
@ -24,15 +24,16 @@ cdef class APINoiseFrameHelper(APIFrameHelper):

    @cython.locals(
        header=bytes,
-        preamble=cython.uint,
-        msg_size_high=cython.uint,
-        msg_size_low=cython.uint,
+        preamble="unsigned char",
+        msg_size_high="unsigned char",
+        msg_size_low="unsigned char",
    )
    cpdef data_received(self, object data)

    @cython.locals(
-        type_high=cython.uint,
-        type_low=cython.uint
+        msg=bytes,
+        type_high="unsigned char",
+        type_low="unsigned char"
    )
    cdef _handle_frame(self, bytes frame)