You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
1005 B
Python
40 lines
1005 B
Python
from __future__ import annotations
|
|
|
|
from typing import Dict
|
|
|
|
|
|
from ...models import MediaElement
|
|
from ..helpers import expect_suitable_extractor
|
|
from .base import MediaExtractor
|
|
from .youtube import YoutubeMediaExtractor
|
|
from .ytdl import YtdlMediaExtractor
|
|
|
|
|
|
MEDIA_EXTRACTORS: Dict[str, MediaExtractor] = {
|
|
"youtube": YoutubeMediaExtractor(),
|
|
"ytdl": YtdlMediaExtractor(),
|
|
}
|
|
|
|
|
|
def media_expect_extractor(uri: str) -> MediaExtractor:
|
|
return expect_suitable_extractor(
|
|
extractor_list=MEDIA_EXTRACTORS.values(),
|
|
uri=uri,
|
|
)
|
|
|
|
|
|
def media_update(element: MediaElement, check_cache_expired: bool = True) -> None:
|
|
ex = media_expect_extractor(element.uri)
|
|
ex.update_object(
|
|
object=element,
|
|
check_cache_expired=check_cache_expired,
|
|
)
|
|
|
|
|
|
def media_extract_uri(uri: str) -> MediaElement:
|
|
elem: MediaElement = MediaExtractor.check_uri(uri)
|
|
if not elem:
|
|
ex = media_expect_extractor(uri)
|
|
elem = ex.extract_and_store(uri)
|
|
return elem
|