extr/coll,media/youtube: Move setting URLs to media extractor

master
Felix Stupp 2 years ago
parent 4e62a3fe70
commit 96e12f6fe8
Signed by: zocker
GPG Key ID: 93E1BD26F6B02FB7

@ -100,15 +100,10 @@ class YouTubeCollectionExtractor(CollectionExtractor[Dict]):
video_list = reversed(video_list)
for index, video in enumerate(video_list):
video_url = f"https://www.youtube.com/watch?v={video['id']}"
other_urls = [
f"https://youtube.com/watch?v={video['id']}",
f"https://youtu.be/{video['id']}",
]
element = self._add_episode(
collection=object,
uri=video_url,
episode=index + 1,
)
if element:
element.add_uris(other_urls)
orm.commit() # so progress is stored

@ -112,3 +112,10 @@ class YoutubeMediaExtractor(MediaExtractor[YoutubeVideoData]):
data.get("uploadDate") or data["publishDate"], "%Y-%m-%d"
)
object.length = int(data["duration"]["secondsText"])
object.uri = f"https://www.youtube.com/watch?v={data['id']}"
object.add_uris(
(
f"https://youtu.be/{data['id']}",
f"https://youtube.com/watch?v={data['id']}",
)
)

Loading…
Cancel
Save