util.py

Go to the documentation of this file.

"""Small utility functions for the dlna_dms integration."""
 
from __future__ import annotations
 
from homeassistant.core import HomeAssistant
from homeassistant.util import slugify
 
from .const import CONF_SOURCE_ID, DOMAIN
 
 
def generate_source_id(hass: HomeAssistant, name: str) -> str:
    """Generate a unique source ID."""
    other_entries = hass.config_entries.async_entries(DOMAIN)
    other_source_ids: set[str] = {
        other_source_id
        for entry in other_entries
        if (other_source_id := entry.data.get(CONF_SOURCE_ID))
    }
 
    source_id_base = slugify(name)
    if source_id_base not in other_source_ids:
        return source_id_base
 
    tries = 1
    while (suggested_source_id := f"{source_id_base}_{tries}") in other_source_ids:
        tries += 1
 
    return suggested_source_id