scripts/assign_interval_to_annotator.py

# scripts/assign_interval_to_annotator.py

from utils.database import get_db
from data.repository.annotator_repo import AnnotatorRepo
from data.repository.annotation_interval_repo import AnnotationIntervalRepo
from utils.logger import Logger

log = Logger()

START_IDX = 101
END_IDX = 120
ANNOTATOR_NAME = "vargha"

with get_db() as db:
    annot_repo = AnnotatorRepo(db)
    interval_repo = AnnotationIntervalRepo(db)

    annotator = annot_repo.get_annotator_by_name(ANNOTATOR_NAME)
    if not annotator:
        log.info(f"Annotator '{annotator}' not found;")

    try:
        interval = interval_repo.assign_interval_to_annotator(
            annotator_id=annotator.id,
            start_idx=START_IDX,
            end_idx=END_IDX,
        )
        log.info(
            f"Interval [{interval.start_index}, {interval.end_index}] "
            f"successfully assigned to '{annotator.name}' (id={annotator.id})"
        )
    except ValueError as err:
        log.error(f"Could not assign interval: {err}")