mirror of
https://github.com/TheCommsChannel/TC2-BBS-mesh.git
synced 2024-10-01 01:15:36 -04:00
Added database admin script
This adds a simple database admin script
This commit is contained in:
parent
15d836db27
commit
5a2db4b5f5
201
db_admin.py
Normal file
201
db_admin.py
Normal file
@ -0,0 +1,201 @@
|
|||||||
|
import logging
|
||||||
|
import sqlite3
|
||||||
|
import threading
|
||||||
|
import uuid
|
||||||
|
from datetime import datetime
|
||||||
|
import os
|
||||||
|
|
||||||
|
# Import functions from meshtastic and utils as needed
|
||||||
|
|
||||||
|
thread_local = threading.local()
|
||||||
|
|
||||||
|
def get_db_connection():
|
||||||
|
if not hasattr(thread_local, 'connection'):
|
||||||
|
thread_local.connection = sqlite3.connect('bulletins.db')
|
||||||
|
return thread_local.connection
|
||||||
|
|
||||||
|
def initialize_database():
|
||||||
|
conn = get_db_connection()
|
||||||
|
c = conn.cursor()
|
||||||
|
c.execute('''CREATE TABLE IF NOT EXISTS bulletins (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
board TEXT NOT NULL,
|
||||||
|
sender_short_name TEXT NOT NULL,
|
||||||
|
date TEXT NOT NULL,
|
||||||
|
subject TEXT NOT NULL,
|
||||||
|
content TEXT NOT NULL,
|
||||||
|
unique_id TEXT NOT NULL
|
||||||
|
)''')
|
||||||
|
c.execute('''CREATE TABLE IF NOT EXISTS mail (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
sender TEXT NOT NULL,
|
||||||
|
sender_short_name TEXT NOT NULL,
|
||||||
|
recipient TEXT NOT NULL,
|
||||||
|
date TEXT NOT NULL,
|
||||||
|
subject TEXT NOT NULL,
|
||||||
|
content TEXT NOT NULL,
|
||||||
|
unique_id TEXT NOT NULL
|
||||||
|
);''')
|
||||||
|
c.execute('''CREATE TABLE IF NOT EXISTS channels (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
name TEXT NOT NULL,
|
||||||
|
url TEXT NOT NULL
|
||||||
|
);''')
|
||||||
|
conn.commit()
|
||||||
|
|
||||||
|
def list_bulletins():
|
||||||
|
conn = get_db_connection()
|
||||||
|
c = conn.cursor()
|
||||||
|
c.execute("SELECT id, board, sender_short_name, date, subject, unique_id FROM bulletins")
|
||||||
|
bulletins = c.fetchall()
|
||||||
|
if bulletins:
|
||||||
|
print_bold("Bulletins:")
|
||||||
|
for bulletin in bulletins:
|
||||||
|
print_bold(f"(ID: {bulletin[0]}, Board: {bulletin[1]}, Poster: {bulletin[2]}, Subject: {bulletin[4]})")
|
||||||
|
else:
|
||||||
|
print_bold("No bulletins found.")
|
||||||
|
print_separator()
|
||||||
|
return bulletins
|
||||||
|
|
||||||
|
def list_mail():
|
||||||
|
conn = get_db_connection()
|
||||||
|
c = conn.cursor()
|
||||||
|
c.execute("SELECT id, sender, sender_short_name, recipient, date, subject, unique_id FROM mail")
|
||||||
|
mail = c.fetchall()
|
||||||
|
if mail:
|
||||||
|
print_bold("Mail:")
|
||||||
|
for mail in mail:
|
||||||
|
print_bold(f"(ID: {mail[0]}, Sender: {mail[2]}, Recipient: {mail[3]}, Subject: {mail[5]})")
|
||||||
|
else:
|
||||||
|
print_bold("No mail found.")
|
||||||
|
print_separator()
|
||||||
|
return mail
|
||||||
|
|
||||||
|
def list_channels():
|
||||||
|
conn = get_db_connection()
|
||||||
|
c = conn.cursor()
|
||||||
|
c.execute("SELECT id, name, url FROM channels")
|
||||||
|
channels = c.fetchall()
|
||||||
|
if channels:
|
||||||
|
print_bold("Channels:")
|
||||||
|
for channel in channels:
|
||||||
|
print_bold(f"(ID: {channel[0]}, Name: {channel[1]}, URL: {channel[2]})")
|
||||||
|
else:
|
||||||
|
print_bold("No channels found.")
|
||||||
|
print_separator()
|
||||||
|
return channels
|
||||||
|
|
||||||
|
def delete_bulletin():
|
||||||
|
bulletins = list_bulletins()
|
||||||
|
if bulletins:
|
||||||
|
bulletin_ids = input_bold("Enter the bulletin ID(s) to delete (comma-separated) or 'X' to cancel: ").split(',')
|
||||||
|
if 'X' in [id.strip().upper() for id in bulletin_ids]:
|
||||||
|
print_bold("Deletion cancelled.")
|
||||||
|
print_separator()
|
||||||
|
return
|
||||||
|
conn = get_db_connection()
|
||||||
|
c = conn.cursor()
|
||||||
|
for bulletin_id in bulletin_ids:
|
||||||
|
c.execute("DELETE FROM bulletins WHERE id = ?", (bulletin_id.strip(),))
|
||||||
|
conn.commit()
|
||||||
|
print_bold(f"Bulletin(s) with ID(s) {', '.join(bulletin_ids)} deleted.")
|
||||||
|
print_separator()
|
||||||
|
|
||||||
|
def delete_mail():
|
||||||
|
mail = list_mail()
|
||||||
|
if mail:
|
||||||
|
mail_ids = input_bold("Enter the mail ID(s) to delete (comma-separated) or 'X' to cancel: ").split(',')
|
||||||
|
if 'X' in [id.strip().upper() for id in mail_ids]:
|
||||||
|
print_bold("Deletion cancelled.")
|
||||||
|
print_separator()
|
||||||
|
return
|
||||||
|
conn = get_db_connection()
|
||||||
|
c = conn.cursor()
|
||||||
|
for mail_id in mail_ids:
|
||||||
|
c.execute("DELETE FROM mail WHERE id = ?", (mail_id.strip(),))
|
||||||
|
conn.commit()
|
||||||
|
print_bold(f"Mail with ID(s) {', '.join(mail_ids)} deleted.")
|
||||||
|
print_separator()
|
||||||
|
|
||||||
|
def delete_channel():
|
||||||
|
channels = list_channels()
|
||||||
|
if channels:
|
||||||
|
channel_ids = input_bold("Enter the channel ID(s) to delete (comma-separated) or 'X' to cancel: ").split(',')
|
||||||
|
if 'X' in [id.strip().upper() for id in channel_ids]:
|
||||||
|
print_bold("Deletion cancelled.")
|
||||||
|
print_separator()
|
||||||
|
return
|
||||||
|
conn = get_db_connection()
|
||||||
|
c = conn.cursor()
|
||||||
|
for channel_id in channel_ids:
|
||||||
|
c.execute("DELETE FROM channels WHERE id = ?", (channel_id.strip(),))
|
||||||
|
conn.commit()
|
||||||
|
print_bold(f"Channel(s) with ID(s) {', '.join(channel_ids)} deleted.")
|
||||||
|
print_separator()
|
||||||
|
|
||||||
|
def display_menu():
|
||||||
|
print("Menu:")
|
||||||
|
print("1. List Bulletins")
|
||||||
|
print("2. List Mail")
|
||||||
|
print("3. List Channels")
|
||||||
|
print("4. Delete Bulletin")
|
||||||
|
print("5. Delete Mail")
|
||||||
|
print("6. Delete Channel")
|
||||||
|
print("7. Exit")
|
||||||
|
|
||||||
|
def display_banner():
|
||||||
|
banner = """
|
||||||
|
████████╗ ██████╗██████╗ ██████╗ ██████╗ ███████╗
|
||||||
|
╚══██╔══╝██╔════╝╚════██╗ ██╔══██╗██╔══██╗██╔════╝
|
||||||
|
██║ ██║ █████╔╝█████╗██████╔╝██████╔╝███████╗
|
||||||
|
██║ ██║ ██╔═══╝ ╚════╝██╔══██╗██╔══██╗╚════██║
|
||||||
|
██║ ╚██████╗███████╗ ██████╔╝██████╔╝███████║
|
||||||
|
╚═╝ ╚═════╝╚══════╝ ╚═════╝ ╚═════╝ ╚══════╝
|
||||||
|
Database Administrator
|
||||||
|
"""
|
||||||
|
print_bold(banner)
|
||||||
|
print_separator()
|
||||||
|
|
||||||
|
def clear_screen():
|
||||||
|
# Clear the console screen
|
||||||
|
os.system('cls' if os.name == 'nt' else 'clear')
|
||||||
|
|
||||||
|
def input_bold(prompt):
|
||||||
|
print("\033[1m") # ANSI escape code for bold text
|
||||||
|
response = input(prompt)
|
||||||
|
print("\033[0m") # ANSI escape code to reset text
|
||||||
|
return response
|
||||||
|
|
||||||
|
def print_bold(message):
|
||||||
|
print("\033[1m" + message + "\033[0m") # Bold text
|
||||||
|
|
||||||
|
def print_separator():
|
||||||
|
print_bold("========================")
|
||||||
|
|
||||||
|
def main():
|
||||||
|
display_banner()
|
||||||
|
initialize_database()
|
||||||
|
while True:
|
||||||
|
display_menu()
|
||||||
|
choice = input_bold("Enter your choice: ")
|
||||||
|
clear_screen()
|
||||||
|
if choice == '1':
|
||||||
|
list_bulletins()
|
||||||
|
elif choice == '2':
|
||||||
|
list_mail()
|
||||||
|
elif choice == '3':
|
||||||
|
list_channels()
|
||||||
|
elif choice == '4':
|
||||||
|
delete_bulletin()
|
||||||
|
elif choice == '5':
|
||||||
|
delete_mail()
|
||||||
|
elif choice == '6':
|
||||||
|
delete_channel()
|
||||||
|
elif choice == '7':
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
print_bold("Invalid choice. Please try again.")
|
||||||
|
print_separator()
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
Loading…
Reference in New Issue
Block a user