FreeDATA/modem/command_message_send.py

58 lines
2.4 KiB
Python
Raw Normal View History

2024-01-18 10:35:44 +00:00
from command import TxCommand
import api_validations
import base64
from queue import Queue
from arq_session_iss import ARQSessionISS
from message_p2p import MessageP2P
2024-01-30 08:30:13 +00:00
from arq_data_type_handler import ARQ_SESSION_TYPES
2024-01-27 11:07:07 +00:00
from message_system_db_manager import DatabaseManager
from message_system_db_messages import DatabaseManagerMessages
2024-01-18 10:35:44 +00:00
class SendMessageCommand(TxCommand):
"""Command to send a P2P message using an ARQ transfer session
"""
def set_params_from_api(self, apiParams):
origin = f"{self.config['STATION']['mycall']}-{self.config['STATION']['myssid']}"
self.message = MessageP2P.from_api_params(origin, apiParams)
2024-02-24 20:49:53 +00:00
DatabaseManagerMessages(self.event_manager).add_message(self.message.to_dict(), statistics={}, direction='transmit', status='queued')
2024-01-18 10:35:44 +00:00
def transmit(self, modem):
2024-01-27 11:07:07 +00:00
2024-02-02 18:37:02 +00:00
if self.state_manager.getARQ():
self.log("Modem busy, waiting until ready...")
return
2024-01-27 11:07:07 +00:00
2024-03-07 14:14:07 +00:00
if not modem:
self.log("Modem not running...", isWarning=True)
return
first_queued_message = DatabaseManagerMessages(self.event_manager).get_first_queued_message()
2024-02-02 18:37:02 +00:00
if not first_queued_message:
self.log("No queued message in database.")
return
2024-02-21 16:05:28 +00:00
try:
self.log(f"Queued message found: {first_queued_message['id']}")
DatabaseManagerMessages(self.event_manager).update_message(first_queued_message["id"], update_data={'status': 'transmitting'})
message_dict = DatabaseManagerMessages(self.event_manager).get_message_by_id(first_queued_message["id"])
message = MessageP2P.from_api_params(message_dict['origin'], message_dict)
# Convert JSON string to bytes (using UTF-8 encoding)
payload = message.to_payload().encode('utf-8')
json_bytearray = bytearray(payload)
data, data_type = self.arq_data_type_handler.prepare(json_bytearray, ARQ_SESSION_TYPES.p2pmsg_lzma)
iss = ARQSessionISS(self.config,
modem,
self.message.destination,
self.state_manager,
data,
data_type
)
self.state_manager.register_arq_iss_session(iss)
iss.start()
except Exception as e:
self.log(f"Error starting ARQ session: {e}", isWarning=True)