from biz.checkpoint.checkpoint_biz import FrameProcessorWorker as CheckpointWorker
from biz.prison.trajectory02_biz import FrameProcessorWorker as TrajectoryWorker
from biz.prison.supervision_room_biz import FrameProcessorWorker as SupervisionWorker
# ... 其他导入

PROCESSOR_MAP = {
    "checkpoint": CheckpointWorker,
    "trajectory": TrajectoryWorker,
    "supervision_room": SupervisionWorker,
}

def get_processor(processor_type: str):
    if processor_type not in PROCESSOR_MAP:
        raise ValueError(f"Unknown processor type: {processor_type}")
    return PROCESSOR_MAP[processor_type]