database_manager.py

# database_manager.py
import sqlite3

DATABASE_FILE = "subscribers.db"

def initialize_database():
    conn = sqlite3.connect(DATABASE_FILE)
    cursor = conn.cursor()
    cursor.execute("CREATE TABLE IF NOT EXISTS subscribers (contact_name TEXT PRIMARY KEY)")
    conn.commit()
    conn.close()
    print("Banco de dados 'subscribers.db' pronto para uso.")

def add_subscriber(contact_name):
    try:
        conn = sqlite3.connect(DATABASE_FILE)
        cursor = conn.cursor()
        cursor.execute("INSERT OR IGNORE INTO subscribers (contact_name) VALUES (?)", (contact_name,))
        conn.commit()
        conn.close()
        print(f"✅ Assinante '{contact_name}' foi ADICIONADO.")
        return True
    except Exception as e:
        print(f"❌ Erro ao adicionar: {e}")
        return False

def remove_subscriber(contact_name):
    try:
        conn = sqlite3.connect(DATABASE_FILE)
        cursor = conn.cursor()
        cursor.execute("DELETE FROM subscribers WHERE contact_name = ?", (contact_name,))
        conn.commit()
        conn.close()
        print(f"🗑️ Assinante '{contact_name}' foi REMOVIDO.")
        return True
    except Exception as e:
        print(f"❌ Erro ao remover: {e}")
        return False

def get_all_subscribers():
    try:
        conn = sqlite3.connect(DATABASE_FILE)
        cursor = conn.cursor()
        cursor.execute("SELECT contact_name FROM subscribers")
        subscribers = [row[0] for row in cursor.fetchall()]
        conn.close()
        return subscribers
    except Exception as e:
        print(f"❌ Erro ao buscar: {e}")
        return []

if __name__ == "__main__":
    initialize_database()