server.py

from flask import Flask, jsonify, request
import hashlib
import random
from datetime import datetime

app = Flask(__name__)

# Mock Database
judges = [
    {"id": 1, "name": "Judge A", "conflicts": ["CorpX"], "wealth_reports": []},
    {"id": 2, "name": "Judge B", "conflicts": [], "wealth_reports": []},
    {"id": 3, "name": "Judge C", "conflicts": ["PolPartyY"], "wealth_reports": []},
]

cases = [
    {"id": 101, "defendant": "CorpX", "category": "corporate"},
    {"id": 102, "defendant": "John Doe", "category": "criminal"},
    {"id": 103, "defendant": "PolPartyY", "category": "political"},
]

assignments = []

# Helper: Hash wealth reports for immutability
def hash_wealth_report(judge_id, year, assets):
    data = f"{judge_id}-{year}-{assets}"
    return hashlib.sha256(data.encode()).hexdigest()

# API 1: Assign a judge to a case (conflict-check + random selection)
@app.route('/assign', methods=['POST'])
def assign_judge():
    case_id = request.json.get('case_id')
    case = next((c for c in cases if c['id'] == case_id), None)
    
    if not case:
        return jsonify({"error": "Case not found"}), 404

    # Filter judges with no conflicts
    eligible_judges = [
        j for j in judges 
        if case['defendant'] not in j['conflicts']
    ]
    
    if not eligible_judges:
        return jsonify({"error": "No eligible judges"}), 400

    # Random assignment
    selected_judge = random.choice(eligible_judges)
    assignment = {
        "case_id": case_id,
        "judge_id": selected_judge['id'],
        "timestamp": datetime.now().isoformat()
    }
    assignments.append(assignment)
    
    return jsonify(assignment)

# API 2: Submit a wealth report (stored as hash)
@app.route('/wealth', methods=['POST'])
def submit_wealth():
    judge_id = request.json.get('judge_id')
    year = request.json.get('year')
    assets = request.json.get('assets')
    
    judge = next((j for j in judges if j['id'] == judge_id), None)
    if not judge:
        return jsonify({"error": "Judge not found"}), 404
    
    # Store hashed report
    report_hash = hash_wealth_report(judge_id, year, assets)
    judge['wealth_reports'].append({
        "year": year,
        "assets": assets,
        "hash": report_hash
    })
    
    return jsonify({"status": "success", "hash": report_hash})

# API 3: Get all data for dashboard
@app.route('/data', methods=['GET'])
def get_data():
    return jsonify({
        "judges": judges,
        "cases": cases,
        "assignments": assignments
    })

if __name__ == '__main__':
    app.run(debug=True)