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

[videos] Youtube using the actual API

Browse Source
This commit is contained in:
Colin Powell
2025-01-27 14:51:56 -05:00
parent ef1fcd4026
commit 66c34942e6
1 changed files with 45 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,50 +1,69 @@
import pendulum import json
from meta_yt import Video, YouTube
from videos.metadata import VideoMetadata, VideoType
import logging import logging
import pendulum
import requests
from django.conf import settings
from videos.metadata import VideoMetadata, VideoType
logger = logging.getLogger(__name__) 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/"
API_KEY = settings.GOOGLE_API_KEY
GOOGLE_VIDEOS_URL = "https://www.googleapis.com/youtube/v3/videos?part=snippet,contentDetails&id={youtube_id}&key={key}"
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
video_metadata = VideoMetadata(youtube_id=youtube_id) video_metadata = VideoMetadata(youtube_id=youtube_id)
try:
yt_metadata: Optional[Video] = YouTube( url = GOOGLE_VIDEOS_URL.format(youtube_id=youtube_id, key=API_KEY)
YOUTUBE_VIDEO_URL + youtube_id headers = {"User-Agent": "Vrobbler 0.11.12"}
).video response = requests.get(url, headers=headers)
except KeyError:
logger.warning("Cannot access youtube, probably thinks you're a bot") if response.status_code != 200:
logger.warning(
"Bad response from Google", extra={"response": response}
)
return video_metadata return video_metadata
if yt_metadata: yt_metadata = (
json.loads(response.content).get("items", [None])[0].get("snippet")
)
duration_iso8601 = (
json.loads(response.content)
.get("items", [None])[0]
.get("contentDetails", {})
.get("duration")
)
minutes = duration_iso8601.split("PT")[1].split("M")[0]
seconds = duration_iso8601.split("PT")[1].split("M")[1].replace("S", "")
if yt_metadata.channel: duration = (int(minutes) * 60) + int(seconds)
channel, _created = Channel.objects.get_or_create(
youtube_id=yt_metadata.channel_id, if yt_metadata:
name=yt_metadata.channel, if yt_metadata.get("channelId"):
channel, _ = Channel.objects.get_or_create(
youtube_id=yt_metadata.get("channelId"),
name=yt_metadata.get("channelTitle"),
) )
video_metadata.channel_id = channel.id video_metadata.channel_id = channel.id
video_metadata.title = yt_metadata.title video_metadata.title = yt_metadata.get("title", "")
video_metadata.run_time_seconds = yt_metadata.duration video_metadata.run_time_seconds = duration
video_metadata.video_type = VideoType.YOUTUBE.value video_metadata.video_type = VideoType.YOUTUBE.value
video_metadata.youtube_id = yt_metadata.video_id video_metadata.youtube_id = youtube_id
video_metadata.cover_url = yt_metadata.thumbnail video_metadata.cover_url = (
video_metadata.genres = yt_metadata.keywords yt_metadata.get("thumbnails", {}).get("high", {}).get("url", {})
video_metadata.overview = yt_metadata.metadata.get("videoDetails").get(
"shortDescription"
) )
video_metadata.genres = yt_metadata.get("tags")
video_metadata.overview = yt_metadata.get("description")
date_str = ( date_str = yt_metadata.get("publishedAt")
yt_metadata.metadata.get("microformat")
.get("playerMicroformatRenderer")
.get("uploadDate")
)
if date_str: if date_str:
video_metadata.upload_date = pendulum.parse(date_str).date() video_metadata.upload_date = pendulum.parse(date_str).date()
video_metadata.year = video_metadata.upload_date.year video_metadata.year = video_metadata.upload_date.year