46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
from yt_dlp import DateRange
|
|
|
|
from ytdl_subscribe.downloaders.youtube_downloader import YoutubeDownloader
|
|
from ytdl_subscribe.subscriptions.subscription import Subscription
|
|
from ytdl_subscribe.validators.config.source_options.youtube_validators import (
|
|
YoutubeChannelSourceValidator,
|
|
)
|
|
from ytdl_subscribe.validators.config.source_options.youtube_validators import (
|
|
YoutubePlaylistSourceValidator,
|
|
)
|
|
from ytdl_subscribe.validators.config.source_options.youtube_validators import (
|
|
YoutubeVideoSourceValidator,
|
|
)
|
|
|
|
|
|
class YoutubePlaylistSubscription(Subscription[YoutubePlaylistSourceValidator]):
|
|
def _extract_info(self):
|
|
entries = self.get_downloader(YoutubeDownloader).download_playlist(
|
|
playlist_id=self.source_options.playlist_id.value
|
|
)
|
|
|
|
for entry in entries:
|
|
self.post_process_entry(entry)
|
|
|
|
|
|
class YoutubeChannelSubscription(Subscription[YoutubeChannelSourceValidator]):
|
|
def _extract_info(self):
|
|
source_ytdl_options = {}
|
|
source_date_range = self.source_options.get_date_range()
|
|
if source_date_range:
|
|
source_ytdl_options["daterange"] = source_ytdl_options
|
|
|
|
downloader = self.get_downloader(YoutubeDownloader, source_ytdl_options=source_ytdl_options)
|
|
entries = downloader.download_channel(channel_id=self.source_options.channel_id.value)
|
|
|
|
for entry in entries:
|
|
self.post_process_entry(entry)
|
|
|
|
|
|
class YoutubeVideoSubscription(Subscription[YoutubeVideoSourceValidator]):
|
|
def _extract_info(self):
|
|
entry = self.get_downloader(YoutubeDownloader).download_video(
|
|
video_id=self.source_options.video_id.value
|
|
)
|
|
|
|
self.post_process_entry(entry)
|