Git-Mirrors/TC2-BBS-mesh
1
0
Fork 0
mirror of https://github.com/TheCommsChannel/TC2-BBS-mesh.git synced 2024-10-01 01:15:36 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor and unify inserts to one method

Browse Source
This commit is contained in:
tekstrand 2024-07-17 21:41:24 -05:00
parent 87add16151
commit 71f066f3c9
1 changed files with 35 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

71
js8call_integration.py
View File

@ -89,47 +89,46 @@ class JS8CallClient:
''') ''')
self.logger.info("Database tables created or verified.") self.logger.info("Database tables created or verified.")
def insert_message(self, sender, receiver, message): def insert_message(self, table, sender, recipient, message):
"""
Inserts a message into the specified table in the database.
This method saves a message along with its sender and receiver or group name into the specified table.
If the database connection is not available, it logs an error message.
Parameters:
-----------
table : str
The name of the table where the message should be inserted. It can be 'messages', 'groups', or 'urgent'.
sender : str
The meshtastic node identifier of the sender who issued the command
recipient : str
The identifier of the receiver of the message or the group name.
message : str
The content of the message.
Example Usage:
--------------
client.insert_message('messages', sender='CALLSIGN1', receiver_or_group='CALLSIGN2', message='This is a message.')
client.insert_message('groups', sender='CALLSIGN1', receiver_or_group='GroupName', message='This is a group message.')
client.insert_message('urgent', sender='CALLSIGN1', receiver_or_group='UrgentGroupName', message='This is an urgent message.')
"""
if not self.db_conn: if not self.db_conn:
self.logger.error("Database connection is not available.") self.logger.error("Database connection is not available.")
return return
try: try:
with self.db_conn: with self.db_conn:
self.db_conn.execute(''' self.db_conn.execute(f'''
INSERT INTO messages (sender, receiver, message) INSERT INTO {table} (sender, { 'receiver' if table == 'messages' else 'groupname' }, message)
VALUES (?, ?, ?) VALUES (?, ?, ?)
''', (sender, receiver, message)) ''', (sender, receiver_or_group, message))
except sqlite3.Error as e: except sqlite3.Error as e:
self.logger.error(f"Failed to insert message into database: {e}") self.logger.error(f"Failed to insert message into {table} table: {e}")
def insert_group(self, sender, groupname, message):
if not self.db_conn:
self.logger.error("Database connection is not available.")
return
try:
with self.db_conn:
self.db_conn.execute('''
INSERT INTO groups (sender, groupname, message)
VALUES (?, ?, ?)
''', (sender, groupname, message))
except sqlite3.Error as e:
self.logger.error(f"Failed to insert group message into database: {e}")
def insert_urgent(self, sender, groupname, message):
if not self.db_conn:
self.logger.error("Database connection is not available.")
return
try:
with self.db_conn:
self.db_conn.execute('''
INSERT INTO urgent (sender, groupname, message)
VALUES (?, ?, ?)
''', (sender, groupname, message))
except sqlite3.Error as e:
self.logger.error(f"Failed to insert urgent message into database: {e}")
def process(self, message): def process(self, message):
typ = message.get('type', '') typ = message.get('type', '')
@ -161,13 +160,13 @@ class JS8CallClient:
self.logger.info(f"Received JS8Call message: {sender} to {receiver} - {msg}") self.logger.info(f"Received JS8Call message: {sender} to {receiver} - {msg}")
if receiver in self.js8urgent: if receiver in self.js8urgent:
self.insert_urgent(sender, receiver, msg) self.insert_urgent('urgent', sender, receiver, msg)
notification_message = f"💥 URGENT JS8Call Message Received 💥\nFrom: {sender}\nCheck BBS for message" notification_message = f"💥 URGENT JS8Call Message Received 💥\nFrom: {sender}\nCheck BBS for message"
send_message(notification_message, BROADCAST_NUM, self.interface) send_message(notification_message, BROADCAST_NUM, self.interface)
elif receiver in self.js8groups: elif receiver in self.js8groups:
self.insert_group(sender, receiver, msg) self.insert_message('groups', sender, receiver, msg)
elif self.store_messages: elif self.store_messages:
self.insert_message(sender, receiver, msg) self.insert_message('messages', sender, receiver, msg)
else: else:
pass pass