2021-11-25 06:44:12 +00:00
|
|
|
# Python imports
|
2021-11-28 04:03:00 +00:00
|
|
|
import threading, socket, time
|
2021-11-25 06:44:12 +00:00
|
|
|
from multiprocessing.connection import Listener, Client
|
|
|
|
|
2021-12-07 04:44:39 +00:00
|
|
|
# Lib imports
|
2021-11-25 06:44:12 +00:00
|
|
|
|
|
|
|
# Application imports
|
|
|
|
|
|
|
|
|
|
|
|
def threaded(fn):
|
|
|
|
def wrapper(*args, **kwargs):
|
2021-12-25 05:58:57 +00:00
|
|
|
threading.Thread(target=fn, args=args, kwargs=kwargs, daemon=True).start()
|
2021-11-25 06:44:12 +00:00
|
|
|
return wrapper
|
|
|
|
|
|
|
|
|
2021-12-07 04:44:39 +00:00
|
|
|
|
|
|
|
|
2022-01-20 02:14:44 +00:00
|
|
|
class IPCServerMixin:
|
2021-11-25 06:44:12 +00:00
|
|
|
|
|
|
|
@threaded
|
|
|
|
def create_ipc_server(self):
|
2022-01-24 16:24:55 +00:00
|
|
|
listener = Listener((self.ipc_address, self.ipc_port), authkey=self.ipc_authkey)
|
2021-11-25 06:44:12 +00:00
|
|
|
self.is_ipc_alive = True
|
2021-12-25 05:58:57 +00:00
|
|
|
while True:
|
2021-11-28 04:03:00 +00:00
|
|
|
conn = listener.accept()
|
|
|
|
start_time = time.time()
|
|
|
|
|
2021-11-25 06:44:12 +00:00
|
|
|
print(f"New Connection: {listener.last_accepted}")
|
|
|
|
while True:
|
|
|
|
msg = conn.recv()
|
2021-11-25 08:21:10 +00:00
|
|
|
if debug:
|
|
|
|
print(msg)
|
2021-11-25 06:44:12 +00:00
|
|
|
|
|
|
|
if "FILE|" in msg:
|
|
|
|
file = msg.split("FILE|")[1].strip()
|
2021-11-25 08:21:10 +00:00
|
|
|
if file:
|
2022-01-31 06:13:43 +00:00
|
|
|
event_system.push_gui_event([None, "handle_file_from_ipc", (file,)])
|
2021-11-25 08:21:10 +00:00
|
|
|
|
2021-11-25 06:44:12 +00:00
|
|
|
conn.close()
|
|
|
|
break
|
|
|
|
|
|
|
|
|
|
|
|
if msg == 'close connection':
|
|
|
|
conn.close()
|
|
|
|
break
|
|
|
|
if msg == 'close server':
|
|
|
|
conn.close()
|
|
|
|
break
|
2021-11-28 04:03:00 +00:00
|
|
|
|
|
|
|
# NOTE: Not perfect but insures we don't lockup the connection for too long.
|
|
|
|
end_time = time.time()
|
2022-01-24 16:24:55 +00:00
|
|
|
if (end - start) > self.ipc_timeout:
|
2021-11-28 04:03:00 +00:00
|
|
|
conn.close()
|
|
|
|
|
2021-11-25 06:44:12 +00:00
|
|
|
listener.close()
|
|
|
|
|
|
|
|
|
|
|
|
def send_ipc_message(self, message="Empty Data..."):
|
|
|
|
try:
|
2022-01-24 16:24:55 +00:00
|
|
|
conn = Client((self.ipc_address, self.ipc_port), authkey=self.ipc_authkey)
|
2021-11-25 06:44:12 +00:00
|
|
|
conn.send(message)
|
|
|
|
conn.send('close connection')
|
|
|
|
except Exception as e:
|
|
|
|
print(repr(e))
|