M-Labs/artiq
2
0
Fork 2
mirror of https://github.com/m-labs/artiq.git synced 2024-12-25 11:18:27 +08:00
Code Issues Projects Releases Wiki Activity

pc_rpc: add init magic string

Browse Source
This commit is contained in:
Sebastien Bourdeauducq 2014-11-20 14:21:41 -08:00
parent 914bdd95d5
commit 64866a0f49
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
artiq/management/pc_rpc.py
View File

@ -29,6 +29,9 @@ class IncompatibleServer(Exception):
pass
_init_string = b"ARTIQ pc_rpc\n"
class Client:
"""This class proxies the methods available on the server so that they
can be used as if they were local methods.
@ -59,6 +62,7 @@ class Client:
"""
def __init__(self, host, port, expected_id_type):
self.socket = socket.create_connection((host, port))
self.socket.sendall(_init_string)
self._identify(expected_id_type)
def get_rpc_id(self):
@ -176,6 +180,9 @@ class Server:
@asyncio.coroutine
def _handle_connection_task(self, reader, writer):
try:
line = yield from reader.readline()
if line != _init_string:
return
while True:
line = yield from reader.readline()
if not line: