utils/json_store.py

import json
import os

SESSION_FILE = "data/session_data.json"

# Ensure the file exists
def init_session_data():
    if not os.path.exists("data"):
        os.makedirs("data")
    if not os.path.exists(SESSION_FILE):
        with open(SESSION_FILE, "w") as f:
            json.dump({}, f)

def read_session_data():
    init_session_data()
    with open(SESSION_FILE, "r") as f:
        return json.load(f)

def write_session_data(data):
    with open(SESSION_FILE, "w") as f:
        json.dump(data, f, indent=4)

def update_session_data(key, value):
    data = read_session_data()
    data[key] = value
    write_session_data(data)

def get_session_value(key, default=None):
    data = read_session_data()
    return data.get(key, default)

def clear_session_data():
    write_session_data({})