secstate/vrobbler
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[videos] Add check if youtube fails

Browse Source
This commit is contained in:
Colin Powell
2025-01-27 01:30:41 -05:00
parent 300a2ae6aa
commit a36efa3b1d
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
vrobbler/apps/videos/sources/youtube.py
View File

@ -1,6 +1,9 @@
import pendulum import pendulum
from meta_yt import Video, YouTube from meta_yt import Video, YouTube
from videos.metadata import VideoMetadata, VideoType from videos.metadata import VideoMetadata, VideoType
import logging
logger = logging.getLogger(__name__)
YOUTUBE_VIDEO_URL = "https://www.youtube.com/watch?v=" YOUTUBE_VIDEO_URL = "https://www.youtube.com/watch?v="
YOUTUBE_CHANNEL_URL = "https://www.youtube.com/channel/" YOUTUBE_CHANNEL_URL = "https://www.youtube.com/channel/"
@ -9,12 +12,16 @@ YOUTUBE_CHANNEL_URL = "https://www.youtube.com/channel/"
def lookup_video_from_youtube(youtube_id: str) -> VideoMetadata: def lookup_video_from_youtube(youtube_id: str) -> VideoMetadata:
from videos.models import Channel from videos.models import Channel
yt_metadata: Optional[Video] = YouTube( video_metadata = VideoMetadata(youtube_id=youtube_id)
YOUTUBE_VIDEO_URL + youtube_id try:
).video yt_metadata: Optional[Video] = YouTube(
YOUTUBE_VIDEO_URL + youtube_id
).video
except KeyError:
logger.warning("Cannot access youtube, probably thinks you're a bot")
return video_metadata
if yt_metadata: if yt_metadata:
video_metadata = VideoMetadata(youtube_id=youtube_id)
if yt_metadata.channel: if yt_metadata.channel:
channel, _created = Channel.objects.get_or_create( channel, _created = Channel.objects.get_or_create(