espressif/openthread
1
0
Fork 0
mirror of https://github.com/espressif/openthread.git synced 2026-09-03 07:40:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

[style] update python style to conform to PEP 8 (#3951)

Browse Source 
With the exception of line length set to 119 vs. 79.

Add tests/ and tools/ to py-pretty-check.
This commit is contained in:
Jonathan Hui
2019-07-11 11:45:55 -04:00
committed by GitHub
parent a938ee2845
commit f924adcb60
341 changed files with 8702 additions and 4460 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/scripts/thread-cert/simulator.py
+102 -34
View File
@@ -29,31 +29,31 @@
import binascii
import bisect
import cmd
import os
import socket
import struct
import sys
import traceback
import time
import io
import config
import dtls
import mesh_cop
import message
import pcap
def dbg_print(*args):
if False:
print(args)
class BaseSimulator(object):
class BaseSimulator(object):
def __init__(self):
self._nodes = {}
self.commissioning_messages = {}
self._payload_parse_factory = mesh_cop.MeshCopCommandFactory(mesh_cop.create_default_mesh_cop_tlv_factories())
self._payload_parse_factory = mesh_cop.MeshCopCommandFactory(
mesh_cop.create_default_mesh_cop_tlv_factories()
)
self._mesh_cop_msg_set = mesh_cop.create_mesh_cop_message_type_set()
def __del__(self):
@@ -77,16 +77,22 @@ class BaseSimulator(object):
def read_cert_messages_in_commissioning_log(self, nodeids):
for nodeid in nodeids:
node = self._nodes[nodeid]
node = self._nodes[nodeid]
if not node:
continue
for direction, type, payload in node.read_cert_messages_in_commissioning_log():
for (
direction,
type,
payload,
) in node.read_cert_messages_in_commissioning_log():
if direction == b'send':
msg = self._payload_parse_factory.parse(type.decode("utf-8"), io.BytesIO(payload))
msg = self._payload_parse_factory.parse(
type.decode("utf-8"), io.BytesIO(payload)
)
self.commissioning_messages[nodeid].append(msg)
class RealTime(BaseSimulator):
class RealTime(BaseSimulator):
def __init__(self):
super(RealTime, self).__init__()
self._sniffer = config.create_default_thread_sniffer()
@@ -97,7 +103,9 @@ class RealTime(BaseSimulator):
def get_messages_sent_by(self, nodeid):
messages = self._sniffer.get_messages_sent_by(nodeid).messages
ret = message.MessagesSet(messages, self.commissioning_messages[nodeid])
ret = message.MessagesSet(
messages, self.commissioning_messages[nodeid]
)
self.commissioning_messages[nodeid] = []
return ret
@@ -107,11 +115,12 @@ class RealTime(BaseSimulator):
def stop(self):
pass
class VirtualTime(BaseSimulator):
OT_SIM_EVENT_ALARM_FIRED = 0
OT_SIM_EVENT_ALARM_FIRED = 0
OT_SIM_EVENT_RADIO_RECEIVED = 1
OT_SIM_EVENT_UART_WRITE = 2
OT_SIM_EVENT_UART_WRITE = 2
OT_SIM_EVENT_RADIO_SPINEL_WRITE = 3
OT_SIM_EVENT_POSTCMD = 4
@@ -125,12 +134,12 @@ class VirtualTime(BaseSimulator):
BASE_PORT = 9000
MAX_NODES = 34
MAX_MESSAGE = 1024
END_OF_TIME = 0x7fffffff
END_OF_TIME = 0x7FFFFFFF
PORT_OFFSET = int(os.getenv('PORT_OFFSET', '0'))
BLOCK_TIMEOUT = 4
RADIO_ONLY = os.getenv('RADIO_DEVICE') != None
RADIO_ONLY = os.getenv('RADIO_DEVICE') is not None
NCP_SIM = os.getenv('NODE_TYPE', 'sim') == 'ncp-sim'
def __init__(self):
@@ -184,7 +193,8 @@ class VirtualTime(BaseSimulator):
def get_messages_sent_by(self, nodeid):
""" Get sniffed messages.
Note! This method flushes the message queue so calling this method again will return only the newly logged messages.
Note! This method flushes the message queue so calling this
method again will return only the newly logged messages.
Args:
nodeid (int): node id
@@ -197,7 +207,9 @@ class VirtualTime(BaseSimulator):
messages = self.devices[addr]['msgs']
self.devices[addr]['msgs'] = []
ret = message.MessagesSet(messages, self.commissioning_messages[nodeid])
ret = message.MessagesSet(
messages, self.commissioning_messages[nodeid]
)
self.commissioning_messages[nodeid] = []
return ret
@@ -227,7 +239,14 @@ class VirtualTime(BaseSimulator):
def receive_events(self):
""" Receive events until all devices are asleep. """
while True:
if self.current_event or len(self.awake_devices) or (self._next_event_time() > self._pause_time and self.current_nodeid):
if (
self.current_event
or len(self.awake_devices)
or (
self._next_event_time() > self._pause_time
and self.current_nodeid
)
):
self.sock.settimeout(self.BLOCK_TIMEOUT)
try:
msg, addr = self.sock.recvfrom(self.MAX_MESSAGE)
@@ -258,7 +277,7 @@ class VirtualTime(BaseSimulator):
self.devices[addr]['msgs'] = []
self.devices[addr]['time'] = self.current_time
self.awake_devices.discard(addr)
#print "New device:", addr, self.devices
# print "New device:", addr, self.devices
delay, type, datalen = struct.unpack('=QBH', msg[:11])
data = msg[11:]
@@ -266,7 +285,14 @@ class VirtualTime(BaseSimulator):
event_time = self.current_time + delay
if data:
dbg_print("New event: ", event_time, addr, type, datalen, binascii.hexlify(data))
dbg_print(
"New event: ",
event_time,
addr,
type,
datalen,
binascii.hexlify(data),
)
else:
dbg_print("New event: ", event_time, addr, type, datalen)
@@ -274,19 +300,22 @@ class VirtualTime(BaseSimulator):
# remove any existing alarm event for device
if self.devices[addr]['alarm']:
self.event_queue.remove(self.devices[addr]['alarm'])
#print "-- Remove\t", self.devices[addr]['alarm']
# print "-- Remove\t", self.devices[addr]['alarm']
# add alarm event to event queue
event = (event_time, self.event_sequence, addr, type, datalen)
self.event_sequence += 1
#print "-- Enqueue\t", event, delay, self.current_time
# print "-- Enqueue\t", event, delay, self.current_time
bisect.insort(self.event_queue, event)
self.devices[addr]['alarm'] = event
self.awake_devices.discard(addr)
if self.current_event and self.current_event[self.EVENT_ADDR] == addr:
#print "Done\t", self.current_event
if (
self.current_event
and self.current_event[self.EVENT_ADDR] == addr
):
# print "Done\t", self.current_event
self.current_event = None
elif type == self.OT_SIM_EVENT_RADIO_RECEIVED:
@@ -294,16 +323,32 @@ class VirtualTime(BaseSimulator):
# add radio receive events event queue
for device in self.devices:
if device != addr and self._is_radio(device):
event = (event_time, self.event_sequence, device, type, datalen, data)
event = (
event_time,
self.event_sequence,
device,
type,
datalen,
data,
)
self.event_sequence += 1
#print "-- Enqueue\t", event
# print "-- Enqueue\t", event
bisect.insort(self.event_queue, event)
self._pcap.append(data, (event_time // 1000000, event_time % 1000000))
self._pcap.append(
data, (event_time // 1000000, event_time % 1000000)
)
self._add_message(addr[1] - self.port, data)
# add radio transmit done events to event queue
event = (event_time, self.event_sequence, addr, type, datalen, data)
event = (
event_time,
self.event_sequence,
addr,
type,
datalen,
data,
)
self.event_sequence += 1
bisect.insort(self.event_queue, event)
@@ -312,10 +357,17 @@ class VirtualTime(BaseSimulator):
elif type == self.OT_SIM_EVENT_RADIO_SPINEL_WRITE:
assert not self._is_radio(addr)
radio_addr = self._to_radio_addr(addr)
if not radio_addr in self.devices:
if radio_addr not in self.devices:
self.awake_devices.add(radio_addr)
event = (event_time, self.event_sequence, radio_addr, self.OT_SIM_EVENT_UART_WRITE, datalen, data)
event = (
event_time,
self.event_sequence,
radio_addr,
self.OT_SIM_EVENT_UART_WRITE,
datalen,
data,
)
self.event_sequence += 1
bisect.insort(self.event_queue, event)
@@ -324,10 +376,17 @@ class VirtualTime(BaseSimulator):
elif type == self.OT_SIM_EVENT_UART_WRITE:
assert self._is_radio(addr)
core_addr = self._to_core_addr(addr)
if not core_addr in self.devices:
if core_addr not in self.devices:
self.awake_devices.add(core_addr)
event = (event_time, self.event_sequence, core_addr, self.OT_SIM_EVENT_RADIO_SPINEL_WRITE, datalen, data)
event = (
event_time,
self.event_sequence,
core_addr,
self.OT_SIM_EVENT_RADIO_SPINEL_WRITE,
datalen,
data,
)
self.event_sequence += 1
bisect.insort(self.event_queue, event)
@@ -362,11 +421,18 @@ class VirtualTime(BaseSimulator):
dbg_print("Pop event: ", event_time, addr, type, datalen)
else:
event_time, sequence, addr, type, datalen, data = event
dbg_print("Pop event: ", event_time, addr, type, datalen, binascii.hexlify(data))
dbg_print(
"Pop event: ",
event_time,
addr,
type,
datalen,
binascii.hexlify(data),
)
self.current_event = event
assert(event_time >= self.current_time)
assert event_time >= self.current_time
self.current_time = event_time
elapsed = event_time - self.devices[addr]['time']
@@ -395,7 +461,9 @@ class VirtualTime(BaseSimulator):
continue
dbg_print('syncing', addr, elapsed)
self.devices[addr]['time'] = self.current_time
message = struct.pack('=QBH', elapsed, self.OT_SIM_EVENT_ALARM_FIRED, 0)
message = struct.pack(
'=QBH', elapsed, self.OT_SIM_EVENT_ALARM_FIRED, 0
)
self._send_message(message, addr)
self.awake_devices.add(addr)
self.receive_events()