mirrors/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2025-08-14 08:28:29 +00:00
Code Issues Projects Releases Wiki Activity

traverse_obj: allow traversal of dataclasses (for protobug)

Browse Source
This commit is contained in:
garret1317 2025-08-11 08:20:24 +01:00
parent bf366517ef
commit a9df16db47
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
yt_dlp/utils/traversal.py
View File

@ -3,6 +3,7 @@
import collections import collections
import collections.abc import collections.abc
import contextlib import contextlib
import dataclasses
import functools import functools
import http.cookies import http.cookies
import inspect import inspect
@ -233,6 +234,8 @@ def apply_specials(element):
result = list(map(apply_specials, obj.iterfind(xpath))) result = list(map(apply_specials, obj.iterfind(xpath)))
else: else:
result = apply_specials(obj) result = apply_specials(obj)
elif dataclasses.is_dataclass(obj):
result = getattr(obj, key)
return branching, result if branching else (result,) return branching, result if branching else (result,)