mirror of
https://github.com/DJ2LS/FreeDATA
synced 2024-05-14 08:04:33 +00:00
240 lines
9.2 KiB
Python
240 lines
9.2 KiB
Python
#!/usr/bin/python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Created on Tue Dec 22 16:58:45 2020
|
|
|
|
@author: DJ2LS
|
|
|
|
"""
|
|
|
|
|
|
import argparse
|
|
import threading
|
|
import socketserver
|
|
import pyaudio
|
|
import time
|
|
import ujson as json
|
|
import subprocess
|
|
import os
|
|
import static
|
|
import psutil
|
|
import sys
|
|
#PORT = 3001
|
|
#TNCPROCESS = 0
|
|
#TNCSTARTED = False
|
|
|
|
#p = pyaudio.PyAudio()
|
|
#info = p.get_host_api_info_by_index(0)
|
|
#numdevices = info.get('deviceCount')
|
|
#for each audio device, determine if is an input or an output and add it to the appropriate list and dictionary
|
|
#for i in range (0,numdevices):
|
|
# if p.get_device_info_by_host_api_device_index(0,i).get('maxInputChannels')>0:
|
|
# print("Input Device id ", i, " - ", p.get_device_info_by_host_api_device_index(0,i).get('name'))#
|
|
#
|
|
# if p.get_device_info_by_host_api_device_index(0,i).get('maxOutputChannels')>0:
|
|
# print("Output Device id ", i, " - ", p.get_device_info_by_host_api_device_index(0,i).get('name'))
|
|
|
|
def start_daemon():
|
|
|
|
try:
|
|
print("SRV | STARTING TCP/IP SOCKET FOR CMD ON PORT: " + str(PORT))
|
|
socketserver.TCPServer.allow_reuse_address = True # https://stackoverflow.com/a/16641793
|
|
daemon = socketserver.TCPServer(('0.0.0.0', PORT), CMDTCPRequestHandler)
|
|
daemon.serve_forever()
|
|
|
|
finally:
|
|
daemon.server_close()
|
|
|
|
|
|
class CMDTCPRequestHandler(socketserver.BaseRequestHandler):
|
|
|
|
def handle(self):
|
|
print("Client connected...")
|
|
|
|
# loop through socket buffer until timeout is reached. then close buffer
|
|
socketTimeout = time.time() + 3
|
|
while socketTimeout > time.time():
|
|
|
|
time.sleep(0.01)
|
|
encoding = 'utf-8'
|
|
#data = str(self.request.recv(1024), 'utf-8')
|
|
|
|
data = bytes()
|
|
|
|
# we need to loop through buffer until end of chunk is reached or timeout occured
|
|
while True and socketTimeout > time.time():
|
|
chunk = self.request.recv(1024) # .strip()
|
|
data += chunk
|
|
if chunk.startswith(b'{') and chunk.endswith(b'}\n'):
|
|
break
|
|
data = data[:-1] # remove b'\n'
|
|
data = str(data, 'utf-8')
|
|
#print(data)
|
|
|
|
if len(data) > 0:
|
|
socketTimeout = time.time() + 3
|
|
|
|
# convert data to json object
|
|
# we need to do some error handling in case of socket timeout
|
|
|
|
try:
|
|
|
|
received_json = json.loads(data)
|
|
|
|
except ValueError as e:
|
|
print("++++++++++++ START OF JSON ERROR +++++++++++++++++++++++")
|
|
print(e)
|
|
print("-----------------------------------")
|
|
print(data)
|
|
print("++++++++++++ END OF JSON ERROR +++++++++++++++++++++++++")
|
|
received_json = {}
|
|
|
|
|
|
|
|
# GET COMMANDS
|
|
# "command" : "..."
|
|
|
|
# SET COMMANDS
|
|
# "command" : "..."
|
|
# "parameter" : " ..."
|
|
|
|
# DATA COMMANDS
|
|
# "command" : "..."
|
|
# "type" : "..."
|
|
# "dxcallsign" : "..."
|
|
# "data" : "..."
|
|
|
|
# print(received_json)
|
|
#print(received_json["type"])
|
|
#print(received_json["command"])
|
|
try:
|
|
|
|
if received_json["type"] == 'SET' and received_json["command"] == 'STARTTNC' and not static.TNCSTARTED:
|
|
rx_audio = received_json["parameter"][0]["rx_audio"]
|
|
tx_audio = received_json["parameter"][0]["tx_audio"]
|
|
deviceid = received_json["parameter"][0]["deviceid"]
|
|
deviceport = received_json["parameter"][0]["deviceport"]
|
|
serialspeed = received_json["parameter"][0]["serialspeed"]
|
|
ptt = received_json["parameter"][0]["ptt"]
|
|
print("---- STARTING TNC !")
|
|
print(received_json["parameter"][0])
|
|
#os.system("python3 main.py --rx 3 --tx 3 --deviceport /dev/ttyUSB0 --deviceid 2028")
|
|
|
|
|
|
# Start RIGCTLD
|
|
|
|
if ptt == "RTS":
|
|
dtr_state = "OFF"
|
|
else:
|
|
dtr_state = "NONE"
|
|
|
|
if sys.platform == "linux":
|
|
command = "exec ./hamlib/linux/rigctld -r " + str(deviceport) + \
|
|
" -s "+ str(serialspeed) + \
|
|
" -P "+ str(ptt) + \
|
|
" -m "+ str(deviceid) + \
|
|
" --set-conf=dtr_state=" + dtr_state
|
|
try:
|
|
p = subprocess.Popen(command, shell=True)
|
|
time.sleep(0.3)
|
|
except:
|
|
print("hamlib not started")
|
|
sys.exit()
|
|
|
|
elif sys.platform == "darwin":
|
|
print("platform not yet supported")
|
|
sys.exit()
|
|
|
|
elif sys.platform == "win32":
|
|
print("platform not yet supported")
|
|
sys.exit()
|
|
|
|
else:
|
|
print("platform not supported!")
|
|
sys.exit()
|
|
|
|
|
|
|
|
|
|
|
|
if DEBUG:
|
|
p = subprocess.Popen("exec python3 main.py --rx "+ str(rx_audio) +" --tx "+ str(tx_audio) +" --deviceport "+ str(deviceport) +" --deviceid "+ str(deviceid) + " --serialspeed "+ str(serialspeed) + " --ptt "+ str(ptt), shell=True)
|
|
else:
|
|
p = subprocess.Popen("exec ./tnc --rx "+ str(rx_audio) +" --tx "+ str(tx_audio) +" --deviceport "+ str(deviceport) +" --deviceid "+ str(deviceid) + " --serialspeed "+ str(serialspeed) + " --ptt "+ str(ptt), shell=True)
|
|
static.TNCPROCESS = p#.pid
|
|
static.TNCSTARTED = True
|
|
|
|
if received_json["type"] == 'SET' and received_json["command"] == 'STOPTNC':
|
|
parameter = received_json["parameter"]
|
|
static.TNCPROCESS.kill()
|
|
print("KILLING PROCESS ------------")
|
|
#os.kill(static.TNCPROCESS, signal.SIGKILL)
|
|
static.TNCSTARTED = False
|
|
|
|
if received_json["type"] == 'GET' and received_json["command"] == 'DAEMON_STATE':
|
|
|
|
data = {'COMMAND' : 'DAEMON_STATE', 'DAEMON_STATE' : [], 'INPUT_DEVICES': [], 'OUTPUT_DEVICES': [],"CPU": str(psutil.cpu_percent()),"RAM": str(psutil.virtual_memory().percent), "VERSION": "0.1-prototype"}
|
|
|
|
if static.TNCSTARTED:
|
|
data["DAEMON_STATE"].append({"STATUS": "running"})
|
|
else:
|
|
data["DAEMON_STATE"].append({"STATUS": "stopped"})
|
|
|
|
p = pyaudio.PyAudio()
|
|
for i in range(0, p.get_device_count()):
|
|
|
|
maxInputChannels = p.get_device_info_by_host_api_device_index(0,i).get('maxInputChannels')
|
|
maxOutputChannels = p.get_device_info_by_host_api_device_index(0,i).get('maxOutputChannels')
|
|
name = p.get_device_info_by_host_api_device_index(0,i).get('name')
|
|
|
|
if maxInputChannels > 0:
|
|
data["INPUT_DEVICES"].append({"ID": i, "NAME" : str(name)})
|
|
if maxOutputChannels > 0:
|
|
data["OUTPUT_DEVICES"].append({"ID": i, "NAME" : str(name)})
|
|
|
|
|
|
|
|
#print(data)
|
|
jsondata = json.dumps(data)
|
|
self.request.sendall(bytes(jsondata, encoding))
|
|
|
|
|
|
|
|
#exception, if JSON cant be decoded
|
|
#except Exception as e:
|
|
except ValueError as e:
|
|
print("############ START OF ERROR #####################")
|
|
print('DAEMON PROGRAM ERROR: %s' %str(e))
|
|
print("Wrong command")
|
|
print(data)
|
|
print(e)
|
|
exc_type, exc_obj, exc_tb = sys.exc_info()
|
|
fname = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1]
|
|
print(exc_type, fname, exc_tb.tb_lineno)
|
|
print("############ END OF ERROR #######################")
|
|
|
|
print("Client disconnected...")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
|
# --------------------------------------------GET PARAMETER INPUTS
|
|
PARSER = argparse.ArgumentParser(description='Simons TEST TNC')
|
|
PARSER.add_argument('--port', dest="socket_port", default=3001, help="Socket port", type=int)
|
|
PARSER.add_argument('--debug', dest="debug", action='store_true')
|
|
|
|
|
|
ARGS = PARSER.parse_args()
|
|
PORT = ARGS.socket_port
|
|
DEBUG = ARGS.debug
|
|
# --------------------------------------------START CMD SERVER
|
|
|
|
DAEMON_THREAD = threading.Thread(target=start_daemon, name="daemon")
|
|
DAEMON_THREAD.start()
|
|
|
|
|
|
|
|
|
|
|
|
|