import os
import json
from g4f.cookies import get_cookies_dir

def get_logs(log_dir):
    try:
        return [f for f in os.listdir(log_dir) if os.path.isfile(os.path.join(log_dir, f))]
    except OSError:
        return None

for part in (".logging", ".usage"):
    log_dir = os.path.join(get_cookies_dir(), part)
    save_dir = os.path.join(".", part[1:])
    for filename in get_logs(log_dir):
        with open(os.path.join(log_dir, filename), "rb") as file:
            with open(os.path.join(save_dir, filename), "w") as save:
                for line in file:
                    line = json.loads(line)
                    if "origin" in line:
                        line.pop("origin")
                    if "user" in line:
                        line.pop("user")
                    save.write(json.dumps(line) + "\n")