extractor/collection/{tt_rss,youtube}: Fix extracting media uri's

master
Felix Stupp 3 years ago
parent 37d24b0cee
commit aafa99cbf7
Signed by: zocker
GPG Key ID: 93E1BD26F6B02FB7

@ -9,6 +9,7 @@ from pony import orm # TODO remove
from ...models import MediaCollection
from ..all.tt_rss import HeadlineList, TtRssConnectionParameter, TtRssUri
from ..generic import ExtractedData, ExtractionError, SuitableLevel
from ..media import media_extract_uri
from .base import CollectionExtractor
@ -78,7 +79,7 @@ class TtRssCollectionExtractor(CollectionExtractor[HeadlineList]):
for headline in data:
logging.debug(f"Add to collection {headline.url!r}")
try:
object.add_episode(media_extract_uri("ytdl", headline.url))
object.add_episode(media_extract_uri(headline.url))
orm.commit()
except ExtractionError:
logging.warning(f"Failed while extracting media {headline.url!r}", exc_info=True)

@ -10,6 +10,7 @@ import youtubesearchpython
from ...models import MediaCollection
from ..generic import ExtractedData, ExtractionError, SuitableLevel
from ..media import media_extract_uri
from .base import CollectionExtractor
@ -99,7 +100,7 @@ class YouTubeCollectionExtractor(CollectionExtractor[Dict]):
]
logging.debug(f"[youtube] Add to collection {object.title!r} video {video_url!r} ({index+1} of {len_video_list})")
try:
element = media_extract_uri("ytdl", video_url)
element = media_extract_uri(video_url)
element.add_uris(other_urls)
object.add_episode(element, episode=index+1)
orm.commit() # so progress is stored

Loading…
Cancel
Save