simulation.py

import random

from faker import Faker
fake = Faker()

from patient import Patient

from enums import Severity

CONDITIONS = {
    Severity.MINOR:    ["sprained ankle", "mild headache", "small laceration", "common cold"],
    Severity.MODERATE: ["broken arm", "deep laceration", "high fever", "mild concussion"],
    Severity.SERIOUS:  ["pneumonia", "appendicitis", "severe infection", "fractured rib"],
    Severity.CRITICAL: ["heart attack", "stroke", "severe trauma", "internal bleeding"],
    Severity.FATAL:    ["cardiac arrest", "massive stroke", "critical head trauma"],
}

def generate_patient():
    name = fake.name()
    age = random.randint(18, 90)
    severity = random.choices(
        list(Severity),
        weights=[30, 25, 20, 15, 10]
    )[0]
    condition = random.choice(CONDITIONS[severity])
    return Patient(name, age, condition, severity)