extractors/{collection,media}: Extract *_expect_extractor "partial" method

master
Felix Stupp 3 years ago
parent ab61adbf38
commit 4d5f5ea4d4
Signed by: zocker
GPG Key ID: 93E1BD26F6B02FB7

@ -16,9 +16,15 @@ COLLECTION_EXTRACTORS: Dict[str, CollectionExtractor] = {
"youtube": YouTubeCollectionExtractor(),
}
def collection_expect_extractor(uri: str) -> CollectionExtractor:
return expect_suitable_extractor(
extractor_list = COLLECTION_EXTRACTORS.values(),
uri = uri,
)
def collection_extract_uri(uri: str) -> MediaCollection:
elem: MediaCollection = CollectionExtractor.check_uri(uri)
ex = expect_suitable_extractor(COLLECTION_EXTRACTORS.values(), uri)
ex = collection_expect_extractor(uri)
if not elem:
elem = ex.extract_and_store(uri)
else:

@ -12,8 +12,15 @@ MEDIA_EXTRACTORS: Dict[str, MediaExtractor] = {
"ytdl": YtdlMediaExtractor(),
}
def media_expect_extractor(uri: str) -> MediaExtractor:
return expect_suitable_extractor(
extractor_list = MEDIA_EXTRACTORS.values(),
uri = uri,
)
def media_extract_uri(uri: str) -> MediaElement:
elem: MediaElement = MediaExtractor.check_uri(uri)
if not elem:
elem = expect_suitable_extractor(MEDIA_EXTRACTORS.values(), uri).extract_and_store(uri)
ex = media_expect_extractor(uri)
elem = ex.extract_and_store(uri)
return elem

Loading…
Cancel
Save