Wizzard
/
youtube-alert-and-download
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added logging

This commit is contained in:
Wizzard 2024-02-05 00:27:33 -05:00
parent f32f0e7d1e
commit e2321915e0
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
main.py
View File

@ -5,6 +5,10 @@ import yt_dlp as youtube_dl
from config import DISCORD_BOT_TOKEN, DISCORD_CHANNEL_ID, YOUTUBE_CHANNEL_IDS from config import DISCORD_BOT_TOKEN, DISCORD_CHANNEL_ID, YOUTUBE_CHANNEL_IDS
import os import os
import logging
logging.basicConfig(level=logging.INFO)
intents = discord.Intents.default() intents = discord.Intents.default()
intents.messages = True intents.messages = True
intents.guilds = True intents.guilds = True
@ -20,6 +24,7 @@ ydl_opts = {
} }
downloaded_videos_file = 'downloaded_videos.txt' downloaded_videos_file = 'downloaded_videos.txt'
def get_downloaded_videos(): def get_downloaded_videos():
if not os.path.exists(downloaded_videos_file): if not os.path.exists(downloaded_videos_file):
return set() return set()
@ -38,8 +43,10 @@ def get_latest_video_url(channel_id):
@tasks.loop(minutes=10) @tasks.loop(minutes=10)
async def check_new_videos(): async def check_new_videos():
logging.info("Checking new videos...")
downloaded_videos = get_downloaded_videos() downloaded_videos = get_downloaded_videos()
for channel_id in YOUTUBE_CHANNEL_IDS: for channel_id in YOUTUBE_CHANNEL_IDS:
logging.info(f"Checking channel: {channel_id}")
video_url, video_id = get_latest_video_url(channel_id) video_url, video_id = get_latest_video_url(channel_id)
if video_url and video_id not in downloaded_videos: if video_url and video_id not in downloaded_videos:
with youtube_dl.YoutubeDL(ydl_opts) as ydl: with youtube_dl.YoutubeDL(ydl_opts) as ydl: