refactor(sponsorblock): clean up categories

This commit is contained in:
2025-02-25 17:46:30 -05:00
parent 0b3425a658
commit d08744ebb2
3 changed files with 19 additions and 7 deletions

View File

@@ -1,13 +1,21 @@
import hashlib
import json
import aiohttp
from state import sponsorblock_cache
CATEGORY_NAMES = {
"music_offtopic": "non-music",
"sponsor": "sponsored",
}
categories = json.dumps(
[
"interaction",
"intro",
"music_offtopic",
"outro",
"preview",
"selfpromo",
"sponsor",
]
)
async def get_segments(video_id: str):
@@ -18,7 +26,7 @@ async def get_segments(video_id: str):
session = aiohttp.ClientSession()
response = await session.get(
f"https://sponsor.ajay.app/api/skipSegments/{hashPrefix}",
params={"categories": '["sponsor", "music_offtopic"]'},
params={"categories": categories},
)
if response.status == 200 and (
results := list(