diff options
| -rw-r--r-- | gr-network/python/network/qa_socket_pdu.py | 6 | ||||
| -rw-r--r-- | gr-network/python/network/tcp_source.py | 4 |
2 files changed, 7 insertions, 3 deletions
diff --git a/gr-network/python/network/qa_socket_pdu.py b/gr-network/python/network/qa_socket_pdu.py index 136f7e4cc6..075f7aaba1 100644 --- a/gr-network/python/network/qa_socket_pdu.py +++ b/gr-network/python/network/qa_socket_pdu.py @@ -117,8 +117,10 @@ class qa_socket_pdu (gr_unittest.TestCase): pdu_msg = pmt.cons(pmt.PMT_NIL, data) self.pdu_source = blocks.message_strobe(pdu_msg, 500) - self.pdu_send = network.socket_pdu("TCP_SERVER", "localhost", port, mtu) - self.pdu_recv = network.socket_pdu("TCP_CLIENT", "localhost", port, mtu) + self.pdu_send = network.socket_pdu( + "TCP_SERVER", "localhost", port, mtu) + self.pdu_recv = network.socket_pdu( + "TCP_CLIENT", "localhost", port, mtu) self.pdu_sink = blocks.message_debug() self.tb.msg_connect(self.pdu_source, "strobe", self.pdu_send, "pdus") diff --git a/gr-network/python/network/tcp_source.py b/gr-network/python/network/tcp_source.py index 4e07a68497..89efa74ce2 100644 --- a/gr-network/python/network/tcp_source.py +++ b/gr-network/python/network/tcp_source.py @@ -23,6 +23,7 @@ import socket import os from gnuradio import gr, blocks + def _get_sock_fd(addr, port, server): """ Get the file descriptor for the socket. @@ -74,9 +75,10 @@ def _get_sock_fd(addr, port, server): sock.connect((addr, port)) return os.dup(sock.fileno()) + class tcp_source(gr.hier_block2): def __init__(self, itemsize, addr, port, server=True): - #init hier block + # init hier block gr.hier_block2.__init__( self, 'tcp_source', gr.io_signature(0, 0, 0), |