add pxd headers for correct work

fixes definitions
can run until API call
This commit is contained in:
Alex Bezdieniezhnykh
2025-01-16 17:56:58 +02:00
parent 7439005ed7
commit e21dd7e70f
17 changed files with 207 additions and 165 deletions
+2 -12
View File
@@ -1,23 +1,13 @@
import msgpack
cdef enum CommandType:
INFERENCE = 1
LOAD = 2
cdef enum ProcessorType:
SOCKET = 1,
RABBIT = 2
cdef class FileCommand:
cdef str filename
def __init__(self, command_type: CommandType, processor_type: ProcessorType, str filename):
def __init__(self, command_type: CommandType, ProcessorType processor_type, str filename):
self.command_type = command_type
self.processor_type = processor_type
self.filename = filename
@staticmethod
cdef from_msgpack(bytes data, processor_type: ProcessorType):
cdef from_msgpack(bytes data, ProcessorType processor_type):
unpacked = msgpack.unpackb(data, strict_map_key=False)
return FileCommand(unpacked.get("CommandType"), processor_type, unpacked.get("Filename")
)