Address late review of system bridge media source (#77032)

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
This commit is contained in:
Aidan Timson 2022-08-19 23:27:33 +01:00 committed by GitHub
parent 98c9399ff0
commit a076d3faa0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -124,7 +124,10 @@ def _build_base_url(
entry: ConfigEntry, entry: ConfigEntry,
) -> str: ) -> str:
"""Build base url for System Bridge media.""" """Build base url for System Bridge media."""
return f"http://{entry.data[CONF_HOST]}:{entry.data[CONF_PORT]}/api/media/file/data?apiKey={entry.data[CONF_API_KEY]}" return (
f"http://{entry.data[CONF_HOST]}:{entry.data[CONF_PORT]}"
f"/api/media/file/data?apiKey={entry.data[CONF_API_KEY]}"
)
def _build_root_paths( def _build_root_paths(
@ -191,17 +194,19 @@ def _build_media_item(
media_file: MediaFile, media_file: MediaFile,
) -> BrowseMediaSource: ) -> BrowseMediaSource:
"""Build individual media item.""" """Build individual media item."""
ext = ( ext = ""
f"~~{media_file.mime_type}" if media_file.is_file and media_file.mime_type is not None:
if media_file.is_file and media_file.mime_type is not None ext = f"~~{media_file.mime_type}"
else ""
) if media_file.is_directory or media_file.mime_type is None:
media_class = MEDIA_CLASS_DIRECTORY
else:
media_class = MEDIA_CLASS_MAP[media_file.mime_type.split("/", 1)[0]]
return BrowseMediaSource( return BrowseMediaSource(
domain=DOMAIN, domain=DOMAIN,
identifier=f"{path}/{media_file.name}{ext}", identifier=f"{path}/{media_file.name}{ext}",
media_class=MEDIA_CLASS_DIRECTORY media_class=media_class,
if media_file.is_directory or media_file.mime_type is None
else MEDIA_CLASS_MAP[media_file.mime_type.split("/", 1)[0]],
media_content_type=media_file.mime_type, media_content_type=media_file.mime_type,
title=media_file.name, title=media_file.name,
can_play=media_file.is_file, can_play=media_file.is_file,