from inference.core.active_learning.entities import (
    ActiveLearningConfiguration,
    BatchReCreationInterval,
)
from inference.core.active_learning.utils import (
    generate_start_timestamp_for_this_month,
    generate_start_timestamp_for_this_week,
    generate_today_timestamp,
)

RECREATION_INTERVAL2TIMESTAMP_GENERATOR = {
    BatchReCreationInterval.DAILY: generate_today_timestamp,
    BatchReCreationInterval.WEEKLY: generate_start_timestamp_for_this_week,
    BatchReCreationInterval.MONTHLY: generate_start_timestamp_for_this_month,
}


def generate_batch_name(configuration: ActiveLearningConfiguration) -> str:
    batch_name = configuration.batches_name_prefix
    if configuration.batch_recreation_interval is BatchReCreationInterval.NEVER:
        return batch_name
    timestamp_generator = RECREATION_INTERVAL2TIMESTAMP_GENERATOR[
        configuration.batch_recreation_interval
    ]
    timestamp = timestamp_generator()
    return f"{batch_name}_{timestamp}"