Mirrors/Dispatcharr
1
0
Fork 1
mirror of https://github.com/Dispatcharr/Dispatcharr.git synced 2026-07-24 02:28:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Updated providers.py

Browse source 
Updated to prefer TMDB artwork for local media.
This commit is contained in:
Dispatcharr 2026-02-22 12:45:01 -06:00
parent 563deb88b9
commit 846c3ba6ef
1 changed files with 69 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

77
apps/media_servers/providers.py
View file

@ -69,6 +69,55 @@ def _find_artwork_in_parent_dirs(
return None, None
def _is_http_url(value: Optional[str]) -> bool:
if not value:
return False
return value.startswith('http://') or value.startswith('https://')
def _prefer_tmdb_artwork(
metadata: dict,
*,
local_poster: Optional[str] = None,
local_backdrop: Optional[str] = None,
) -> tuple[dict, Optional[str], Optional[str]]:
"""
Treat local filesystem artwork as fallback so TMDB URLs can become primary.
"""
poster_fallback = None
backdrop_fallback = None
existing_poster = str(metadata.get('poster_url') or '').strip()
if existing_poster and not _is_http_url(existing_poster):
poster_fallback = existing_poster
metadata.pop('poster_url', None)
existing_backdrop = str(metadata.get('backdrop_url') or '').strip()
if existing_backdrop and not _is_http_url(existing_backdrop):
backdrop_fallback = existing_backdrop
metadata.pop('backdrop_url', None)
if local_poster:
poster_fallback = local_poster
if local_backdrop:
backdrop_fallback = local_backdrop
return metadata, poster_fallback, backdrop_fallback
def _apply_artwork_fallback(
metadata: dict,
*,
poster_fallback: Optional[str] = None,
backdrop_fallback: Optional[str] = None,
) -> dict:
if poster_fallback and not str(metadata.get('poster_url') or '').strip():
metadata['poster_url'] = poster_fallback
if backdrop_fallback and not str(metadata.get('backdrop_url') or '').strip():
metadata['backdrop_url'] = backdrop_fallback
return metadata
@dataclass
class ProviderLibrary:
id: str
@ -1223,15 +1272,21 @@ class LocalClient(BaseMediaServerClient):
local_poster, local_backdrop = find_local_artwork_files(
os.path.dirname(full_path)
)
if local_poster:
metadata['poster_url'] = local_poster
if local_backdrop:
metadata['backdrop_url'] = local_backdrop
metadata, poster_fallback, backdrop_fallback = _prefer_tmdb_artwork(
metadata,
local_poster=local_poster,
local_backdrop=local_backdrop,
)
metadata, _tmdb_error = enrich_movie_metadata_with_tmdb(
metadata,
title=classification.title or os.path.splitext(file_name)[0],
year=classification.year,
)
metadata = _apply_artwork_fallback(
metadata,
poster_fallback=poster_fallback,
backdrop_fallback=backdrop_fallback,
)
title = (
str(metadata.get('title') or '').strip()
or classification.title
@ -1315,15 +1370,21 @@ class LocalClient(BaseMediaServerClient):
os.path.dirname(full_path),
stop_dir=base_path,
)
if local_poster:
series_meta['poster_url'] = local_poster
if local_backdrop:
series_meta['backdrop_url'] = local_backdrop
series_meta, poster_fallback, backdrop_fallback = _prefer_tmdb_artwork(
series_meta,
local_poster=local_poster,
local_backdrop=local_backdrop,
)
series_meta, _tmdb_error = enrich_series_metadata_with_tmdb(
series_meta,
title=series_title,
year=classification.year,
)
series_meta = _apply_artwork_fallback(
series_meta,
poster_fallback=poster_fallback,
backdrop_fallback=backdrop_fallback,
)
external_series_id = hashlib.sha1(
series_key.encode('utf-8')
).hexdigest()