ErrorNoInternet/ErrorNoCord
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

fix: handle missing sponsorblock category names

Browse Source
This commit is contained in:
Ryan 2025-02-26 16:21:29 -05:00
parent 5430f7c632
commit 062676df26
Signed by: ErrorNoInternet
GPG Key ID: 2486BFB7B1E6A4A3
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
commands/voice/sponsorblock.py
View File

@ -28,11 +28,12 @@ async def sponsorblock_command(message):
text = []
for segment in video["segments"]:
begin, end = map(int, segment["segment"])
category_name = SPONSORBLOCK_CATEGORY_NAMES.get(segment["category"])
if (category := segment["category"]) in SPONSORBLOCK_CATEGORY_NAMES:
category = SPONSORBLOCK_CATEGORY_NAMES[category]
current = "**" if progress >= begin and progress < end else ""
text.append(
f"{current}`{audio.utils.format_duration(begin)}` - `{audio.utils.format_duration(end)}`: {category_name if category_name else 'Unknown'}{current}"
f"{current}`{audio.utils.format_duration(begin)}` - `{audio.utils.format_duration(end)}`: {category}{current}"
)
await utils.reply(

1
constants.py
View File

@ -21,6 +21,7 @@ OWNERS = [531392146767347712]
PREFIX = "%"
SPONSORBLOCK_CATEGORY_NAMES = {
"music_offtopic": "non-music",
"selfpromo": "self promotion",
"sponsor": "sponsored",
}
RELOADABLE_MODULES = [