Add missing parameters to __init__

use class name in __repr__
pull/8/head
Nicolas Jouanin 2015-06-18 11:39:20 +02:00
rodzic 68003c6bd5
commit 2307d3b13b
4 zmienionych plików z 6 dodań i 6 usunięć

Wyświetl plik

@ -43,14 +43,14 @@ class ConnackVariableHeader(MQTTVariableHeader):
return out
def __repr__(self):
return 'ConnackVariableHeader(session_parent={0}, return_code={1})'.format(hex(self.session_parent), hex(self.return_code.value))
return type(self).__name__ + '(session_parent={0}, return_code={1})'.format(hex(self.session_parent), hex(self.return_code.value))
class ConnackPacket(MQTTPacket):
VARIABLE_HEADER = ConnackVariableHeader
PAYLOAD = None
def __init__(self, fixed: MQTTFixedHeader, variable_header: ConnackVariableHeader=None):
def __init__(self, fixed: MQTTFixedHeader=None, variable_header: ConnackVariableHeader=None, payload=None):
if fixed is None:
header = MQTTFixedHeader(PacketType.CONNACK, 0x00)
else:

Wyświetl plik

@ -120,7 +120,7 @@ class MQTTFixedHeader:
return cls(msg_type, flags, remain_length)
def __repr__(self):
return 'MQTTFixedHeader(type={0}, length={1}, flags={2})'.format(self.packet_type, self.remaining_length, hex(self.flags))
return type(self).__name__ + '(type={0}, length={1}, flags={2})'.format(self.packet_type, self.remaining_length, hex(self.flags))
class MQTTVariableHeader(metaclass=abc.ABCMeta):
def __init__(self):
@ -209,7 +209,7 @@ class MQTTPacket:
return cls(fixed_header, variable_header, payload)
def __repr__(self):
return 'MQPacket(fixed={0!r}, variable={1!r}, payload={2!r})'.format(self.fixed_header, self.variable_header, self.payload)
return type(self).__name__ + '(fixed={0!r}, variable={1!r}, payload={2!r})'.format(self.fixed_header, self.variable_header, self.payload)
class PacketIdVariableHeader(MQTTVariableHeader):

Wyświetl plik

@ -8,7 +8,7 @@ class PingReqPacket(MQTTPacket):
VARIABLE_HEADER = None
PAYLOAD = None
def __init__(self, fixed: MQTTFixedHeader=None):
def __init__(self, fixed: MQTTFixedHeader=None, variable_header=None, payload=None):
if fixed is None:
header = MQTTFixedHeader(PacketType.PINGREQ, 0x00)
else:

Wyświetl plik

@ -8,7 +8,7 @@ class PingRespPacket(MQTTPacket):
VARIABLE_HEADER = None
PAYLOAD = None
def __init__(self, fixed: MQTTFixedHeader=None):
def __init__(self, fixed: MQTTFixedHeader=None, variable_header=None, payload=None):
if fixed is None:
header = MQTTFixedHeader(PacketType.PINGREQ, 0x00)
else: