LibBotUniversal/libbot/i18n/__init__.py

from os import listdir
from pathlib import Path
from typing import Any, Dict

from libbot import config_get, json_read, sync
from libbot.i18n.classes.bot_locale import BotLocale


async def _(key: str, *args: str, locale: str = sync.config_get("locale")) -> Any:
    """Get value of locale string

    ### Args:
        * key (`str`): The last key of the locale's keys path
        * *args (`list`): Path to key like: `dict[args][key]`
        * locale (`str`): Locale to looked up in. Defaults to config's `"locale"` value

    ### Returns:
        * `Any`: Value of provided locale key. Is usually `str`, `dict` or `list`
    """
    if locale is None:
        locale = sync.config_get("locale")

    try:
        this_dict = await json_read(
            Path(f'{await config_get("locale", "locations")}/{locale}.json')
        )
    except FileNotFoundError:
        try:
            this_dict = await json_read(
                Path(
                    f'{await config_get("locale", "locations")}/{await config_get("locale")}.json'
                )
            )
        except FileNotFoundError:
            return f'⚠️ Locale in config is invalid: could not get "{key}" in {args} from locale "{locale}"'

    this_key = this_dict
    for dict_key in args:
        this_key = this_key[dict_key]

    try:
        return this_key[key]
    except KeyError:
        return f'⚠️ Locale in config is invalid: could not get "{key}" in {args} from locale "{locale}"'


async def in_all_locales(key: str, *args: str) -> list:
    """Get value of the provided key and path in all available locales

    ### Args:
        * key (`str`): The last key of the locale's keys path.
        * *args (`list`): Path to key like: `dict[args][key]`.

    ### Returns:
        * `list`: List of values in all locales
    """

    output = []
    files_locales = listdir(await config_get("locale", "locations"))

    valid_locales = [".".join(entry.split(".")[:-1]) for entry in files_locales]
    for lc in valid_locales:
        try:
            this_dict = await json_read(
                Path(f'{await config_get("locale", "locations")}/{lc}.json')
            )
        except FileNotFoundError:
            continue

        this_key = this_dict
        for dict_key in args:
            this_key = this_key[dict_key]

        try:
            output.append(this_key[key])
        except KeyError:
            continue

    return output


async def in_every_locale(key: str, *args: str) -> Dict[str, Any]:
    """Get value of the provided key and path in every available locale with locale tag

    ### Args:
        * key (`str`): The last key of the locale's keys path.
        * *args (`list`): Path to key like: `dict[args][key]`.

    ### Returns:
        * `Dict[str, Any]`: Locale is a key and it's value from locale file is a value
    """

    output = {}
    files_locales = listdir(await config_get("locale", "locations"))

    valid_locales = [".".join(entry.split(".")[:-1]) for entry in files_locales]
    for lc in valid_locales:
        try:
            this_dict = await json_read(
                Path(f'{await config_get("locale", "locations")}/{lc}.json')
            )
        except FileNotFoundError:
            continue

        this_key = this_dict
        for dict_key in args:
            this_key = this_key[dict_key]

        try:
            output[lc] = this_key[key]
        except KeyError:
            continue

    return output