migrate2.py

import db

conn = db.get_conn()

conn.execute("""
    CREATE TABLE IF NOT EXISTS traffic (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        username TEXT NOT NULL,
        tun_ip TEXT,
        dst_ip TEXT NOT NULL,
        dst_port INTEGER,
        protocol TEXT,
        domain TEXT,
        ts TEXT DEFAULT CURRENT_TIMESTAMP
    )
""")

conn.execute("CREATE INDEX IF NOT EXISTS idx_traffic_user ON traffic(username)")
conn.execute("CREATE INDEX IF NOT EXISTS idx_traffic_ts ON traffic(ts)")

conn.commit()
conn.close()
print("traffic table ready")