From b719a8283c61a7b9bec0ad6e5164d5fe27c8fbb5 Mon Sep 17 00:00:00 2001 From: nico Date: Thu, 2 Jul 2015 14:02:10 +0200 Subject: [PATCH] HBMQTT-6 Add build() class method for packet PUBACK, PUBREC and PUBREL. --- hbmqtt/mqtt/puback.py | 6 ++++++ hbmqtt/mqtt/pubcomp.py | 6 ++++++ hbmqtt/mqtt/pubrec.py | 6 ++++++ 3 files changed, 18 insertions(+) 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