amqtt/hbmqtt/mqtt/pubrec.py

28 wiersze
1.0 KiB
Python

# Copyright (c) 2015 Nicolas JOUANIN
#
# See the file license.txt for copying permission.
from hbmqtt.mqtt.packet import MQTTPacket, MQTTFixedHeader, PacketType, PacketIdVariableHeader
from hbmqtt.errors import HBMQTTException
class PubrecPacket(MQTTPacket):
VARIABLE_HEADER = PacketIdVariableHeader
PAYLOAD = None
def __init__(self, fixed: MQTTFixedHeader=None, variable_header: PacketIdVariableHeader=None, payload=None):
if fixed is None:
header = MQTTFixedHeader(PacketType.PUBREC, 0x00)
else:
if fixed.packet_type is not PacketType.PUBREC:
raise HBMQTTException("Invalid fixed packet type %s for PubrecPacket init" % fixed.packet_type)
header = fixed
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