Untitled

import json
import pickle
import os
from flask import Flask, request

app = Flask(__name__)

class LogEntry:
    def __init__(self, user, message):
        self.user = user
        self.message = message

def store_log(log):
    with open(f'logs/{log["user"]}.log', 'wb') as f:
        f.write(pickle.dumps(log))

@app.route("/store_logs", methods=["POST"])
def store_logs():
    if "log" not in request.files:
        return {"error": "Missing log file"}, 400

    log_file = request.files["log"]
    log_data = log_file.read()

    with open(f'logs/{log_file.filename}', 'wb') as f:
        f.write(log_data)

    return {"status": "Log stored successfully"}

@app.route("/fetch_logs", methods=["POST"])
def fetch_logs():
    data = request.get_json()
    
    if "log_id" not in data:
        return {"error": "Missing log_id"}, 400

    with open(f'logs/{data["log_id"]}.log', 'rb') as f:
        log_entry = pickle.loads(f.read())

    return json.dumps({"user": log_entry.user, "message": log_entry.message})

if __name__ == "__main__":
    app.run()