database.py

import sqlite3
from datetime import datetime

def init_db():
    conn = sqlite3.connect('blocked_traffic.db')
    c = conn.cursor()
    c.execute('''
            CREATE TABLE IF NOT EXISTS blocked_requests (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                url TEXT,
                method TEXT,
                headers TEXT,
                content TEXT,
                blocked_at TIMESTAMP,
                reason TEXT
            )
        ''')
    conn.commit()
    conn.close()

def add_blocked_request(url, method, headers, content, reason):
    conn = sqlite3.connect('blocked_traffic.db')
    c = conn.cursor()
    c.execute('''
        INSERT INTO blocked_requests (url, method, headers, content, blocked_at, reason)
        VALUES (?, ?, ?, ?, ?, ?)
    ''', (url, method, str(headers), content, datetime.now(), reason))
    conn.commit()
    conn.close()

def get_blocked_requests():
    conn = sqlite3.connect('blocked_traffic.db')
    c = conn.cursor()
    c.execute('SELECT * FROM blocked_requests ORDER BY blocked_at DESC')
    rows = c.fetchall()
    conn.close()
    return rows