41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
import json
|
|
import re
|
|
import requests
|
|
from .. import config
|
|
from .. import util
|
|
from ..exceptions import InvalidVideoIdException
|
|
headers = config.headers
|
|
pattern=re.compile(r"yt\.setConfig\({'PLAYER_CONFIG': ({.*})}\);")
|
|
|
|
class VideoInfo:
|
|
def __init__(self,video_id):
|
|
self.video_id = video_id
|
|
self.info = self._get_info(video_id)
|
|
|
|
def _get_info(self,video_id):
|
|
url = f"https://www.youtube.com/embed/{video_id}"
|
|
resp= requests.get(url, headers = headers)
|
|
resp.raise_for_status()
|
|
return self._parse(resp.text)
|
|
|
|
def _parse(self,html):
|
|
result = re.search(pattern, html)
|
|
res= json.loads(result.group(1))
|
|
response = res["args"].get("embedded_player_response")
|
|
if response is None:
|
|
raise InvalidVideoIdException("動画IDが無効です。")
|
|
renderer = (json.loads(response))["embedPreview"]["thumbnailPreviewRenderer"]
|
|
return {
|
|
"duration": int(renderer["videoDurationSeconds"]),
|
|
"title" : [''.join(run["text"]) for run in renderer["title"]["runs"]][0],
|
|
"channelId" : renderer["videoDetails"]["embeddedPlayerOverlayVideoDetailsRenderer"]["channelThumbnailEndpoint"]["channelThumbnailEndpoint"]["urlEndpoint"]["urlEndpoint"]["url"][9:],
|
|
"authorProfileImage" : renderer["videoDetails"]["embeddedPlayerOverlayVideoDetailsRenderer"]["channelThumbnail"]["thumbnails"][0]["url"],
|
|
"thumbnail" : renderer["defaultThumbnail"]["thumbnails"][2]["url"]
|
|
}
|
|
|
|
def get(self,item):
|
|
return self.info.get(item)
|
|
|
|
|
|
|