diff --git a/hbmqtt/mqtt/puback.py b/hbmqtt/mqtt/puback.py index f9999d5..90effa0 100644 --- a/hbmqtt/mqtt/puback.py +++ b/hbmqtt/mqtt/puback.py @@ -19,3 +19,9 @@ class PubackPacket(MQTTPacket): super().__init__(header) self.variable_header = variable_header self.payload = None + + @classmethod + def build(cls, packet_id: int): + v_header = PacketIdVariableHeader(packet_id) + packet = PubackPacket(variable_header=v_header, payload=None) + return packet diff --git a/hbmqtt/mqtt/pubcomp.py b/hbmqtt/mqtt/pubcomp.py index c133a40..fb79f2a 100644 --- a/hbmqtt/mqtt/pubcomp.py +++ b/hbmqtt/mqtt/pubcomp.py @@ -19,3 +19,9 @@ class PubcompPacket(MQTTPacket): super().__init__(header) self.variable_header = variable_header self.payload = None + + @classmethod + def build(cls, packet_id: int): + v_header = PacketIdVariableHeader(packet_id) + packet = PubcompPacket(variable_header=v_header, payload=None) + return packet diff --git a/hbmqtt/mqtt/pubrec.py b/hbmqtt/mqtt/pubrec.py index 87e7e2d..9b46c61 100644 --- a/hbmqtt/mqtt/pubrec.py +++ b/hbmqtt/mqtt/pubrec.py @@ -19,3 +19,9 @@ class PubrecPacket(MQTTPacket): super().__init__(header) self.variable_header = variable_header self.payload = None + + @classmethod + def build(cls, packet_id: int): + v_header = PacketIdVariableHeader(packet_id) + packet = PubrecPacket(variable_header=v_header, payload=None) + return packet