kodi-jacktook / plugin.video.jacktook / lib / clients / stremio / torrentio.py

import os
from datetime import timedelta

import xbmcgui

from lib.clients.stremio.constants import (
    TORRENTIO_PROVIDERS_KEY,
    all_torrentio_providers,
)
from lib.db.cached import cache
from lib.utils.kodi.utils import (
    ADDON_PATH,
    translation,
)


def torrentio_toggle_providers(params):
    selected_ids = cache.get(TORRENTIO_PROVIDERS_KEY) or ""
    selected_ids = selected_ids.split(",") if selected_ids else []

    options = []
    selected_indexes = []
    for i, (key, name, _logo) in enumerate(all_torrentio_providers):
        item = xbmcgui.ListItem(label=name)
        item.setArt({"icon": os.path.join(ADDON_PATH, "resources", "img", "torrentio.png")})
        options.append(item)
        if key in selected_ids:
            selected_indexes.append(i)

    dialog = xbmcgui.Dialog()

    selected = dialog.multiselect(
        translation(90117), options, preselect=selected_indexes, useDetails=True
    )

    if selected is None:
        return

    new_selected_ids = [all_torrentio_providers[i][0] for i in selected]
    cache.set(TORRENTIO_PROVIDERS_KEY, ",".join(new_selected_ids), timedelta(days=365 * 20))