github/ha-core
1
Fork 0
mirror of https://github.com/home-assistant/core synced 2024-08-02 23:40:32 +02:00
Code Releases Activity

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

Browse Source 
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
1 changed files with 14 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
homeassistant/components/system_bridge/media_source.py
View File

@ -124,7 +124,10 @@ def _build_base_url(
entry: ConfigEntry,
) -> str:
"""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(
@ -191,17 +194,19 @@ def _build_media_item(
media_file: MediaFile,
) -> BrowseMediaSource:
"""Build individual media item."""
ext = (
f"~~{media_file.mime_type}"
if media_file.is_file and media_file.mime_type is not None
else ""
)
ext = ""
if media_file.is_file and media_file.mime_type is not None:
ext = f"~~{media_file.mime_type}"
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(
domain=DOMAIN,
identifier=f"{path}/{media_file.name}{ext}",
media_class=MEDIA_CLASS_DIRECTORY
if media_file.is_directory or media_file.mime_type is None
else MEDIA_CLASS_MAP[media_file.mime_type.split("/", 1)[0]],
media_class=media_class,
media_content_type=media_file.mime_type,
title=media_file.name,
can_play=media_file.is_file,