Fix extracting video info

This commit is contained in:
taizan-hokuto
2020-08-21 23:16:12 +09:00
parent b02a2badf7
commit 5640c45af6
2 changed files with 12 additions and 3 deletions

View File

@@ -7,3 +7,9 @@ class InvalidVideoIdException(Exception):
class UnknownConnectionError(Exception): class UnknownConnectionError(Exception):
pass pass
class VideoInfoParseException(Exception):
'''
thrown when failed to parse video info
'''

View File

@@ -4,9 +4,9 @@ import json
import re import re
import requests import requests
from . import util from . import util
from . exceptions import InvalidVideoIdException from . exceptions import InvalidVideoIdException, VideoInfoParseException
pattern = re.compile(r"yt\.setConfig\({'PLAYER_CONFIG': ({.*})}\);") pattern = re.compile(r"'PLAYER_CONFIG': ({.*}}})")
item_channel_id = [ item_channel_id = [
"videoDetails", "videoDetails",
@@ -98,7 +98,10 @@ class VideoInfo:
def _parse(self, text): def _parse(self, text):
result = re.search(pattern, text) result = re.search(pattern, text)
self._res = json.loads(result.group(1)) gr = result.group(1)
if gr is None:
raise VideoInfoParseException("Failed to parse video info.")
self._res = json.loads(gr[:-1])
response = self._get_item(self._res, item_response) response = self._get_item(self._res, item_response)
if response is None: if response is None:
self._check_video_is_private(self._res.get("args")) self._check_video_is_private(self._res.get("args"))