database.py

import sqlite3

DB_NAME = "filtration.db"

def init_db():
    conn = sqlite3.connect(DB_NAME)
    c = conn.cursor()
    
    c.execute('''
        CREATE TABLE IF NOT EXISTS filtration_data (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            timestamp TEXT,
            turbidity REAL,
            pressure REAL,
            flow_rate REAL,
            flux REAL
        )
    ''')
    
    conn.commit()
    conn.close()

def insert_data(timestamp, turbidity, pressure, flow_rate, flux):
    conn = sqlite3.connect(DB_NAME)
    c = conn.cursor()
    
    c.execute('''
        INSERT INTO filtration_data (timestamp, turbidity, pressure, flow_rate, flux)
        VALUES (?, ?, ?, ?, ?)
    ''', (timestamp, turbidity, pressure, flow_rate, flux))
    
    conn.commit()
    conn.close()

def fetch_latest(limit=50):
    conn = sqlite3.connect(DB_NAME)
    c = conn.cursor()
    
    c.execute('''
        SELECT timestamp, turbidity, pressure, flow_rate, flux
        FROM filtration_data
        ORDER BY id DESC
        LIMIT ?
    ''', (limit,))
    
    data = c.fetchall()
    conn.close()
    return data[::-1]