Archives/tubearchivist
2
0
Fork 0
mirror of https://github.com/tubearchivist/tubearchivist synced 2024-11-19 15:25:51 +00:00
Code Issues Projects Releases Wiki Activity

fix sponsorblock segments parser, #397

Browse Source
This commit is contained in:
simon 2022-12-31 11:17:11 +07:00
parent 14e0429758
commit 34ed6ebc63
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4
1 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tubearchivist/home/src/index/video.py
View File

@ -73,20 +73,14 @@ class SponsorBlock:
def _get_sponsor_dict(self, all_segments): def _get_sponsor_dict(self, all_segments):
"""format and process response""" """format and process response"""
has_unlocked = False _ = [i.pop("description", None) for i in all_segments]
cleaned_segments = [] has_unlocked = not any(i.get("locked") for i in all_segments)
for segment in all_segments:
if not segment["locked"]:
has_unlocked = True
del segment["userID"]
del segment["description"]
cleaned_segments.append(segment)
sponsor_dict = { sponsor_dict = {
"last_refresh": self.last_refresh, "last_refresh": self.last_refresh,
"has_unlocked": has_unlocked, "has_unlocked": has_unlocked,
"is_enabled": True, "is_enabled": True,
"segments": cleaned_segments, "segments": all_segments,
} }
return sponsor_dict return sponsor_dict