[YouTube] Improve fix for ae8ba2c

Thx: https://github.com/yt-dlp/yt-dlp/commit/01aba25
This commit is contained in:
dirkf
2023-06-18 00:52:18 +01:00
parent ae8ba2c319
commit 07af47960f
2 changed files with 13 additions and 12 deletions

View File

@ -940,15 +940,18 @@ class JSInterpreter(object):
def extract_object(self, objname):
_FUNC_NAME_RE = r'''(?:[a-zA-Z$0-9]+|"[a-zA-Z$0-9]+"|'[a-zA-Z$0-9]+')'''
obj = {}
obj_m = re.search(
r'''(?xs)
(?:{0}\s*\.\s*{1}|{1}\s*=\s*\{{\s*
(?P<fields>({2}\s*:\s*function\s*\(.*?\)\s*\{{.*?}}(?:,\s*)?)*)
}}\s*);
'''.format(_NAME_RE, re.escape(objname), _FUNC_NAME_RE),
self.code)
fields = obj_m and obj_m.group('fields')
if fields is None:
fields = None
for obj_m in re.finditer(
r'''(?xs)
{0}\s*\.\s*{1}|{1}\s*=\s*\{{\s*
(?P<fields>({2}\s*:\s*function\s*\(.*?\)\s*\{{.*?}}(?:,\s*)?)*)
}}\s*;
'''.format(_NAME_RE, re.escape(objname), _FUNC_NAME_RE),
self.code):
fields = obj_m.group('fields')
if fields:
break
else:
raise self.Exception('Could not find object ' + objname)
# Currently, it only supports function definitions
fields_m = re.finditer(