mirror of
https://github.com/espressif/openthread.git
synced 2026-09-05 16:50:05 +00:00
[style] apply google python style guide (#4501)
This commit applies and enforces Google's python style for tests.
This commit is contained in:
@@ -101,6 +101,7 @@ def create_mesh_cop_message_type_set():
|
||||
|
||||
# Channel TLV (0)
|
||||
class Channel(object):
|
||||
|
||||
def __init__(self, channel_page, channel):
|
||||
self._channel_page = channel_page
|
||||
self._channel = channel
|
||||
@@ -116,23 +117,20 @@ class Channel(object):
|
||||
def __eq__(self, other):
|
||||
common.expect_the_same_class(self, other)
|
||||
|
||||
return (
|
||||
self._channel_page == other._channel_page
|
||||
and self._channel == other.__channel
|
||||
)
|
||||
return (self._channel_page == other._channel_page and
|
||||
self._channel == other.__channel)
|
||||
|
||||
def __repr__(self):
|
||||
return 'Channel(channel_page={},channel={})'.format(
|
||||
self._channel_page, self._channel
|
||||
)
|
||||
self._channel_page, self._channel)
|
||||
|
||||
def to_hex(self):
|
||||
return struct.pack(
|
||||
'>BBBH', TlvType.CHANNEL, 3, self.channel_page, self.channel
|
||||
)
|
||||
return struct.pack('>BBBH', TlvType.CHANNEL, 3, self.channel_page,
|
||||
self.channel)
|
||||
|
||||
|
||||
class ChannelFactory(object):
|
||||
|
||||
def parse(self, data, message_info):
|
||||
data_tp = struct.unpack('>BH', data.read(3))
|
||||
channel_page = data_tp[0]
|
||||
@@ -155,6 +153,7 @@ class PanidFactory(object):
|
||||
|
||||
# ExtendedPanid TLV (2)
|
||||
class ExtendedPanid(object):
|
||||
|
||||
def __init__(self, extended_panid):
|
||||
self._extended_panid = extended_panid
|
||||
|
||||
@@ -163,16 +162,15 @@ class ExtendedPanid(object):
|
||||
return self._extended_panid
|
||||
|
||||
def __eq__(self, other):
|
||||
return (
|
||||
isinstance(self, type(other))
|
||||
and self.extended_panid == other.extended_panid
|
||||
)
|
||||
return (isinstance(self, type(other)) and
|
||||
self.extended_panid == other.extended_panid)
|
||||
|
||||
def __repr__(self):
|
||||
return "ExtendedPanid(extended_panid={})".format(self.extended_panid)
|
||||
|
||||
|
||||
class ExtendedPanidFactory(object):
|
||||
|
||||
def parse(self, data, message_info):
|
||||
extended_panid = struct.unpack(">Q", data.read(8))[0]
|
||||
return ExtendedPanid(extended_panid)
|
||||
@@ -180,6 +178,7 @@ class ExtendedPanidFactory(object):
|
||||
|
||||
# NetworkName TLV (3)
|
||||
class NetworkName(object):
|
||||
|
||||
def __init__(self, network_name):
|
||||
self._network_name = network_name
|
||||
|
||||
@@ -188,16 +187,15 @@ class NetworkName(object):
|
||||
return self._network_name
|
||||
|
||||
def __eq__(self, other):
|
||||
return (
|
||||
isinstance(self, type(other))
|
||||
and self.network_name == other.network_name
|
||||
)
|
||||
return (isinstance(self, type(other)) and
|
||||
self.network_name == other.network_name)
|
||||
|
||||
def __repr__(self):
|
||||
return "NetworkName(network_name={})".format(self.network_name)
|
||||
|
||||
|
||||
class NetworkNameFactory(object):
|
||||
|
||||
def parse(self, data, message_info):
|
||||
len = message_info.length
|
||||
network_name = struct.unpack("{}s".format(10), data.read(len))[0]
|
||||
@@ -258,6 +256,7 @@ class NetworkMeshLocalPrefixFactory(object):
|
||||
|
||||
# Steering Data TLV (8)
|
||||
class SteeringData(object):
|
||||
|
||||
def __init__(self, bloom_filter):
|
||||
self._bloom_filter = bloom_filter
|
||||
|
||||
@@ -272,18 +271,16 @@ class SteeringData(object):
|
||||
|
||||
def __repr__(self):
|
||||
return "SteeringData(bloom_filter={})".format(
|
||||
hexlify(self._bloom_filter)
|
||||
)
|
||||
hexlify(self._bloom_filter))
|
||||
|
||||
def to_hex(self):
|
||||
bloom_filter_len = len(self.bloom_filter)
|
||||
return (
|
||||
struct.pack('>BB', TlvType.STEERING_DATA, bloom_filter_len)
|
||||
+ self.bloom_filter
|
||||
)
|
||||
return (struct.pack('>BB', TlvType.STEERING_DATA, bloom_filter_len) +
|
||||
self.bloom_filter)
|
||||
|
||||
|
||||
class SteeringDataFactory:
|
||||
|
||||
def parse(self, data, message_info):
|
||||
bloom_filter = data.read(message_info.length)
|
||||
return SteeringData(bloom_filter)
|
||||
@@ -291,6 +288,7 @@ class SteeringDataFactory:
|
||||
|
||||
# Border Agent Locator TLV (9)
|
||||
class BorderAgentLocator(object):
|
||||
|
||||
def __init__(self, address):
|
||||
self._border_agent_locator = address
|
||||
|
||||
@@ -305,16 +303,15 @@ class BorderAgentLocator(object):
|
||||
|
||||
def __repr__(self):
|
||||
return "BorderAgentLocator(rloc16={})".format(
|
||||
hex(self._border_agent_locator)
|
||||
)
|
||||
hex(self._border_agent_locator))
|
||||
|
||||
def to_hex(self):
|
||||
return struct.pack(
|
||||
'>BBH', TlvType.BORDER_AGENT_LOCATOR, 2, self.border_agent_locator
|
||||
)
|
||||
return struct.pack('>BBH', TlvType.BORDER_AGENT_LOCATOR, 2,
|
||||
self.border_agent_locator)
|
||||
|
||||
|
||||
class BorderAgentLocatorFactory:
|
||||
|
||||
def parse(self, data, message_info):
|
||||
border_agent_locator = struct.unpack(">H", data.read(2))[0]
|
||||
return BorderAgentLocator(border_agent_locator)
|
||||
@@ -322,6 +319,7 @@ class BorderAgentLocatorFactory:
|
||||
|
||||
# CommissionerId TLV (10)
|
||||
class CommissionerId(object):
|
||||
|
||||
def __init__(self, commissioner_id):
|
||||
self._commissioner_id = commissioner_id
|
||||
|
||||
@@ -333,12 +331,11 @@ class CommissionerId(object):
|
||||
return self.commissioner_id == other.commissioner_id
|
||||
|
||||
def __repr__(self):
|
||||
return "CommissionerId(commissioner_id={})".format(
|
||||
self.commissioner_id
|
||||
)
|
||||
return "CommissionerId(commissioner_id={})".format(self.commissioner_id)
|
||||
|
||||
|
||||
class CommissionerIdFactory(object):
|
||||
|
||||
def parse(self, data, message_info):
|
||||
commissioner_id = data.getvalue().decode('utf-8')
|
||||
return CommissionerId(commissioner_id)
|
||||
@@ -346,6 +343,7 @@ class CommissionerIdFactory(object):
|
||||
|
||||
# Commissioner Session ID TLV (11)
|
||||
class CommissionerSessionId(object):
|
||||
|
||||
def __init__(self, commissioner_session_id):
|
||||
self._commissioner_session_id = commissioner_session_id
|
||||
|
||||
@@ -360,8 +358,7 @@ class CommissionerSessionId(object):
|
||||
|
||||
def __repr__(self):
|
||||
return "CommissionerSessionId(commissioner_session_id={})".format(
|
||||
self._commissioner_session_id
|
||||
)
|
||||
self._commissioner_session_id)
|
||||
|
||||
def to_hex(self):
|
||||
return struct.pack(
|
||||
@@ -373,6 +370,7 @@ class CommissionerSessionId(object):
|
||||
|
||||
|
||||
class CommissionerSessionIdFactory:
|
||||
|
||||
def parse(self, data, message_info):
|
||||
session_id = struct.unpack(">H", data.read(2))[0]
|
||||
return CommissionerSessionId(session_id)
|
||||
@@ -419,6 +417,7 @@ class ActiveTimestampFactory(object):
|
||||
|
||||
# Commissioner UDP Port TLV (15)
|
||||
class CommissionerUdpPort(object):
|
||||
|
||||
def __init__(self, udp_port):
|
||||
self._udp_port = udp_port
|
||||
|
||||
@@ -436,6 +435,7 @@ class CommissionerUdpPort(object):
|
||||
|
||||
|
||||
class CommissionerUdpPortFactory:
|
||||
|
||||
def parse(self, data, message_info):
|
||||
udp_port = struct.unpack(">H", data.read(2))[0]
|
||||
return CommissionerUdpPort(udp_port)
|
||||
@@ -443,6 +443,7 @@ class CommissionerUdpPortFactory:
|
||||
|
||||
# State TLV (16)
|
||||
class State(object):
|
||||
|
||||
def __init__(self, state):
|
||||
self._state = state
|
||||
|
||||
@@ -458,6 +459,7 @@ class State(object):
|
||||
|
||||
|
||||
class StateFactory:
|
||||
|
||||
def parse(self, data, message_info):
|
||||
state = ord(data.read(1))
|
||||
return State(state)
|
||||
@@ -478,6 +480,7 @@ class JoinerDtlsEncapsulationFactory(object):
|
||||
|
||||
# JoinerUdpPort TLV (18)
|
||||
class JoinerUdpPort(object):
|
||||
|
||||
def __init__(self, udp_port):
|
||||
self._udp_port = udp_port
|
||||
|
||||
@@ -486,15 +489,15 @@ class JoinerUdpPort(object):
|
||||
return self._udp_port
|
||||
|
||||
def __eq__(self, other):
|
||||
return (
|
||||
isinstance(self, type(other)) and self.udp_port == other.udp_port
|
||||
)
|
||||
return (isinstance(self, type(other)) and
|
||||
self.udp_port == other.udp_port)
|
||||
|
||||
def __repr__(self):
|
||||
return "JoinerUdpPort(udp_port={})".format(self.udp_port)
|
||||
|
||||
|
||||
class JoinerUdpPortFactory(object):
|
||||
|
||||
def parse(self, data, message_info):
|
||||
udp_port = struct.unpack(">H", data.read(2))[0]
|
||||
return JoinerUdpPort(udp_port)
|
||||
@@ -541,6 +544,7 @@ class JoinerRouterKEKFactory(object):
|
||||
|
||||
# ProvisioningURL TLV (32)
|
||||
class ProvisioningUrl(object):
|
||||
|
||||
def __init__(self, url):
|
||||
self._url = url
|
||||
|
||||
@@ -553,6 +557,7 @@ class ProvisioningUrl(object):
|
||||
|
||||
|
||||
class ProvisioningUrlFactory:
|
||||
|
||||
def parse(self, data, message_info):
|
||||
url = data.decode('utf-8')
|
||||
return ProvisioningUrl(url)
|
||||
@@ -560,6 +565,7 @@ class ProvisioningUrlFactory:
|
||||
|
||||
# VendorName TLV (33)
|
||||
class VendorName(object):
|
||||
|
||||
def __init__(self, vendor_name):
|
||||
self._vendor_name = vendor_name
|
||||
|
||||
@@ -575,6 +581,7 @@ class VendorName(object):
|
||||
|
||||
|
||||
class VendorNameFactory:
|
||||
|
||||
def parse(self, data, message_info):
|
||||
vendor_name = data.getvalue().decode('utf-8')
|
||||
return VendorName(vendor_name)
|
||||
@@ -582,6 +589,7 @@ class VendorNameFactory:
|
||||
|
||||
# VendorModel TLV (34)
|
||||
class VendorModel(object):
|
||||
|
||||
def __init__(self, vendor_model):
|
||||
self._vendor_model = vendor_model
|
||||
|
||||
@@ -597,6 +605,7 @@ class VendorModel(object):
|
||||
|
||||
|
||||
class VendorModelFactory:
|
||||
|
||||
def parse(self, data, message_info):
|
||||
vendor_model = data.getvalue().decode('utf-8')
|
||||
return VendorModel(vendor_model)
|
||||
@@ -604,6 +613,7 @@ class VendorModelFactory:
|
||||
|
||||
# VendorSWVersion TLV (35)
|
||||
class VendorSWVersion(object):
|
||||
|
||||
def __init__(self, vendor_sw_version):
|
||||
self._vendor_sw_version = vendor_sw_version
|
||||
|
||||
@@ -615,12 +625,11 @@ class VendorSWVersion(object):
|
||||
return self.vendor_sw_version == other.vendor_sw_version
|
||||
|
||||
def __repr__(self):
|
||||
return "VendorName(vendor_sw_version={})".format(
|
||||
self.vendor_sw_version
|
||||
)
|
||||
return "VendorName(vendor_sw_version={})".format(self.vendor_sw_version)
|
||||
|
||||
|
||||
class VendorSWVersionFactory:
|
||||
|
||||
def parse(self, data, message_info):
|
||||
vendor_sw_version = data.getvalue()
|
||||
return VendorSWVersion(vendor_sw_version)
|
||||
@@ -628,6 +637,7 @@ class VendorSWVersionFactory:
|
||||
|
||||
# VendorData TLV (36)
|
||||
class VendorData(object):
|
||||
|
||||
def __init__(self, data):
|
||||
self._vendor_data = data
|
||||
|
||||
@@ -640,12 +650,14 @@ class VendorData(object):
|
||||
|
||||
|
||||
class VendorDataFactory(object):
|
||||
|
||||
def parse(self, data, message_info):
|
||||
return VendorData(data)
|
||||
|
||||
|
||||
# VendorStackVersion TLV (37)
|
||||
class VendorStackVersion(object):
|
||||
|
||||
def __init__(self, stack_vendor_oui, build, rev, minor, major):
|
||||
self._stack_vendor_oui = stack_vendor_oui
|
||||
self._build = build
|
||||
@@ -680,6 +692,7 @@ class VendorStackVersion(object):
|
||||
|
||||
|
||||
class VendorStackVersionFactory:
|
||||
|
||||
def parse(self, data, message_info):
|
||||
stack_vendor_oui = struct.unpack(">H", data.read(2))[0]
|
||||
rest = struct.unpack(">BBBB", data.read(4))
|
||||
@@ -809,6 +822,7 @@ class EnergyListFactory(object):
|
||||
|
||||
# Discovery Request TLV (128)
|
||||
class DiscoveryRequest(object):
|
||||
|
||||
def __init__(self, version, joiner_flag):
|
||||
self._version = version
|
||||
self._joiner_flag = joiner_flag
|
||||
@@ -822,19 +836,17 @@ class DiscoveryRequest(object):
|
||||
return self._joiner_flag
|
||||
|
||||
def __eq__(self, other):
|
||||
return (
|
||||
isinstance(self, type(other))
|
||||
and self.version == other.version
|
||||
and self.joiner_flag == other.joiner_flag
|
||||
)
|
||||
return (isinstance(self, type(other)) and
|
||||
self.version == other.version and
|
||||
self.joiner_flag == other.joiner_flag)
|
||||
|
||||
def __repr__(self):
|
||||
return "DiscoveryRequest(version={}, joiner_flag={})".format(
|
||||
self.version, self.joiner_flag
|
||||
)
|
||||
self.version, self.joiner_flag)
|
||||
|
||||
|
||||
class DiscoveryRequestFactory(object):
|
||||
|
||||
def parse(self, data, message_info):
|
||||
data_byte = struct.unpack(">B", data.read(1))[0]
|
||||
version = (data_byte & 0xf0) >> 4
|
||||
@@ -845,6 +857,7 @@ class DiscoveryRequestFactory(object):
|
||||
|
||||
# Discovery Response TLV (128)
|
||||
class DiscoveryResponse(object):
|
||||
|
||||
def __init__(self, version, native_flag):
|
||||
self._version = version
|
||||
self._native_flag = native_flag
|
||||
@@ -858,19 +871,17 @@ class DiscoveryResponse(object):
|
||||
return self._native_flag
|
||||
|
||||
def __eq__(self, other):
|
||||
return (
|
||||
isinstance(self, type(other))
|
||||
and self.version == other.version
|
||||
and self.native_flag == other.native_flag
|
||||
)
|
||||
return (isinstance(self, type(other)) and
|
||||
self.version == other.version and
|
||||
self.native_flag == other.native_flag)
|
||||
|
||||
def __repr__(self):
|
||||
return "DiscoveryResponse(version={}, native_flag={})".format(
|
||||
self.version, self.native_flag
|
||||
)
|
||||
self.version, self.native_flag)
|
||||
|
||||
|
||||
class DiscoveryResponseFactory(object):
|
||||
|
||||
def parse(self, data, message_info):
|
||||
data_byte = struct.unpack(">B", data.read(1))[0]
|
||||
version = (data_byte & 0xf0) >> 4
|
||||
@@ -880,6 +891,7 @@ class DiscoveryResponseFactory(object):
|
||||
|
||||
|
||||
class MeshCopCommand(object):
|
||||
|
||||
def __init__(self, _type, tlvs):
|
||||
self._type = _type
|
||||
self._tlvs = tlvs
|
||||
@@ -907,6 +919,7 @@ def create_deault_mesh_cop_msg_type_map():
|
||||
|
||||
|
||||
class MeshCopCommandFactory:
|
||||
|
||||
def __init__(self, tlvs_factories):
|
||||
self._tlvs_factories = tlvs_factories
|
||||
self._mesh_cop_msg_type_map = create_deault_mesh_cop_msg_type_map()
|
||||
@@ -920,26 +933,22 @@ class MeshCopCommandFactory:
|
||||
except KeyError:
|
||||
raise KeyError(
|
||||
"Could not find TLV factory. Unsupported TLV type: {}".format(
|
||||
_type
|
||||
)
|
||||
)
|
||||
_type))
|
||||
|
||||
def _parse_tlv(self, data):
|
||||
_type = TlvType(ord(data.read(1)))
|
||||
length = self._get_length(data)
|
||||
value = data.read(length)
|
||||
factory = self._get_tlv_factory(_type)
|
||||
return factory.parse(
|
||||
io.BytesIO(value), None
|
||||
) # message_info not needed here
|
||||
return factory.parse(io.BytesIO(value),
|
||||
None) # message_info not needed here
|
||||
|
||||
def _get_mesh_cop_msg_type(self, msg_type_str):
|
||||
try:
|
||||
return self._mesh_cop_msg_type_map[msg_type_str]
|
||||
except KeyError:
|
||||
raise KeyError(
|
||||
'Mesh cop message type not found: {}'.format(msg_type_str)
|
||||
)
|
||||
'Mesh cop message type not found: {}'.format(msg_type_str))
|
||||
|
||||
def parse(self, cmd_type_str, data):
|
||||
cmd_type = self._get_mesh_cop_msg_type(cmd_type_str)
|
||||
@@ -965,5 +974,6 @@ def create_default_mesh_cop_tlv_factories():
|
||||
|
||||
|
||||
class ThreadDiscoveryTlvsFactory(SubTlvsFactory):
|
||||
|
||||
def __init__(self, sub_tlvs_factories):
|
||||
super(ThreadDiscoveryTlvsFactory, self).__init__(sub_tlvs_factories)
|
||||
|
||||
Reference in New Issue
Block a user