1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
import os
import subprocess
import threading
import socket
import time
HEADER = 64
PORT = 6011
SERVER = 'localhost'
ADDR = (SERVER, PORT)
FORMAT = 'utf-8'
DISCONNECT_MESSAGE = b'!DISSCONNECT'
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind(ADDR)
def establish_connection(mutable = {}):
server.listen()
conn, addr = server.accept()
mutable['conn'] = conn
mutable['addr'] = addr
return conn, addr
def send_msg(conn, msg):
data = bytes(msg, FORMAT)
data += b' ' * (HEADER - len(data))
conn.send(data)
def recv_msg(conn):
msg = conn.recv(HEADER).strip()
return msg
def execute_source_file(file_name):
ext = file_name.split('.')
if ext[1] == 'py': cmd = 'Python'
elif ext[1] == 'r': cmd = 'Rscript'
subprocess.run(
f'{cmd} {file_name}',
cwd = os.getcwd(),
start_new_session = True
)
print(f'{file_name} executed')
def launch_and_connect(file_name, CONN):
connect_ = threading.Thread(target=establish_connection, args=[CONN])
launch_ = threading.Thread(target=execute_source_file, args=[file_name])
connect_.start()
launch_.start()
connect_.join()
conn = CONN['conn']
addr = CONN['addr']
CONN['launch_'] = launch_
return conn, addr
def disconnect_and_close(CONN):
conn = CONN['conn']
launch_ = CONN['launch_']
send_msg(conn, '!DISCONNECT')
print(f'closing {CONN["addr"]}')
launch_.join()
print(f'closed {CONN["addr"]}')
conn.close()
CONN_R = {}
CONN_PY = {}
src_fil_r = threading.Thread(target=launch_and_connect, args=['source_file.r', CONN_R])
src_fil_py = threading.Thread(target=launch_and_connect, args=['source_file.py', CONN_PY])
disc_r = threading.Thread(target=disconnect_and_close, args=[CONN_R])
disc_py = threading.Thread(target=disconnect_and_close, args=[CONN_PY])
src_fil_r.run()
src_fil_py.run()
conn_r = CONN_R['conn']
conn_py = CONN_PY['conn']
msg = recv_msg(conn_r)
print(msg)
msg = recv_msg(conn_py)
print(msg)
for i in range(255):
msg = f'message {i}'
send_msg(conn_r, msg)
send_msg(conn_py, msg)
disc_r.start()
disc_py.start()
disc_r.join()
disc_py.join()
print("end launcher")
|
