mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 12:05:07 +00:00
imSp4rky
ebb4e74318
Supplied IDs resolve directly via a shared resolve_by_ids and never fall back to title search; metadata_providers gains per-kind ordering and a disable_metadata switch stops all automatic lookups while explicit IDs keep working. BREAKING: --tmdb/--imdb/--tvdb are now mutually exclusive and require a usable provider; --enrich also overwrites original language; the imdbapi provider and imdb_api_enabled are replaced by a keyless imdb provider.
81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
"""Original-language parsing for the metadata a supplied ID resolves."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from unshackle.commands.dl import parse_language
|
|
from unshackle.core.providers.imdb import primary_language as imdb_language
|
|
from unshackle.core.providers.omdb import primary_language as omdb_language
|
|
from unshackle.core.providers.tvdb import primary_language as tvdb_language
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("tag", "expected"),
|
|
[
|
|
("ja", "ja"), # TMDB alpha-2
|
|
("jpn", "ja"), # IMDb alpha-3
|
|
("pt-BR", "pt-BR"),
|
|
(None, None),
|
|
("", None),
|
|
("und", None),
|
|
("xx-not-a-tag", None),
|
|
],
|
|
)
|
|
def test_parse_language(tag: str | None, expected: str | None) -> None:
|
|
result = parse_language(tag)
|
|
assert (str(result) if result else None) == expected
|
|
|
|
|
|
def test_imdb_language_takes_the_first_spoken_language() -> None:
|
|
data = {"spokenLanguages": {"spokenLanguages": [{"id": "ja", "text": "Japanese"}, {"id": "en", "text": "English"}]}}
|
|
assert imdb_language(data) == "ja"
|
|
|
|
|
|
def test_imdb_language_is_alpha_2() -> None:
|
|
"""IMDb's GraphQL language ids are alpha-2."""
|
|
data = {"spokenLanguages": {"spokenLanguages": [{"id": "ko", "text": "Korean"}]}}
|
|
assert parse_language(imdb_language(data)) is not None
|
|
assert str(parse_language(imdb_language(data))) == "ko"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"data",
|
|
[
|
|
{},
|
|
{"spokenLanguages": None}, # search hits can come back with the whole block null
|
|
{"spokenLanguages": {}},
|
|
{"spokenLanguages": {"spokenLanguages": []}},
|
|
{"spokenLanguages": {"spokenLanguages": None}},
|
|
],
|
|
)
|
|
def test_imdb_language_missing(data: dict) -> None:
|
|
assert imdb_language(data) is None
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("names", "expected"),
|
|
[
|
|
("Korean", "ko"),
|
|
("Korean, English", "ko"), # OMDb lists the main language first
|
|
("", None),
|
|
("None", None), # OMDb's literal placeholder for unknown
|
|
("Not A Language", None),
|
|
],
|
|
)
|
|
def test_omdb_language_from_english_name(names: str, expected: str | None) -> None:
|
|
assert omdb_language({"Language": names}) == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("data", "expected"),
|
|
[
|
|
({"primary_language": "kor"}, "kor"), # /search
|
|
({"originalLanguage": "kor"}, "kor"), # /series/{id}/extended
|
|
({"primary_language": ""}, None),
|
|
({}, None),
|
|
],
|
|
)
|
|
def test_tvdb_language_key_variants(data: dict, expected: str | None) -> None:
|
|
assert tvdb_language(data) == expected
|