hass-core/script/translations/lokalise.py

"""API for Lokalise."""
import requests

from .const import PROJECT_ID
from .util import get_lokalise_token


def get_api() -> "Lokalise":
    """Get Lokalise API."""
    return Lokalise(PROJECT_ID, get_lokalise_token())


class Lokalise:
    """Lokalise API."""

    def __init__(self, project_id, token):
        """Initialize Lokalise API."""
        self.project_id = project_id
        self.token = token

    def request(self, method, path, data):
        """Make a request to the Lokalise API."""
        method = method.upper()
        kwargs = {"headers": {"x-api-token": self.token}}
        if method == "GET":
            kwargs["params"] = data
        else:
            kwargs["json"] = data

        req = requests.request(
            method,
            f"https://api.lokalise.com/api2/projects/{self.project_id}/{path}",
            **kwargs,
        )
        req.raise_for_status()
        return req.json()

    def keys_list(self, params={}):
        """List keys.

        https://app.lokalise.com/api2docs/curl/#transition-list-all-keys-get
        """
        return self.request("GET", "keys", params)["keys"]

    def keys_delete_multiple(self, key_ids):
        """Delete multiple keys.

        https://app.lokalise.com/api2docs/curl/#transition-delete-multiple-keys-delete
        """
        return self.request("DELETE", "keys", {"keys": key_ids})

    def keys_bulk_update(self, updates):
        """Update multiple keys.

        https://app.lokalise.com/api2docs/curl/#transition-bulk-update-put
        """
        return self.request("PUT", "keys", {"keys": updates})