parker/guava
parker/guava
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Only search YouTube when enabled

Browse Source
This commit is contained in:
Parker M. 2024-11-29 00:09:31 -06:00
parent e73db927c1
commit 8b231a1b54
Signed by: parker
GPG Key ID: 505ED36FC12B5D5E
1 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
code/utils/custom_sources.py
View File

@ -6,6 +6,8 @@ from lavalink import (
PlaylistInfo, PlaylistInfo,
) )
from utils.config import YOUTUBE_SUPPORT
class LoadError( class LoadError(
Exception Exception
@ -32,21 +34,22 @@ class CustomAudioTrack(DeferredAudioTrack):
LoadType.EMPTY, LoadType.EMPTY,
LoadType.ERROR, LoadType.ERROR,
): ):
ytmsearch = f"ytmsearch:{self.title} {self.author}" if YOUTUBE_SUPPORT:
results = await client.get_tracks(ytmsearch) ytmsearch = f"ytmsearch:{self.title} {self.author}"
results = await client.get_tracks(ytmsearch)
if not results.tracks or results.load_type in (
LoadType.EMPTY,
LoadType.ERROR,
):
ytsearch = f"ytsearch:{self.title} {self.author} audio"
results = await client.get_tracks(ytsearch)
if not results.tracks or results.load_type in ( if not results.tracks or results.load_type in (
LoadType.EMPTY, LoadType.EMPTY,
LoadType.ERROR, LoadType.ERROR,
): ):
raise LoadError ytsearch = f"ytsearch:{self.title} {self.author} audio"
results = await client.get_tracks(ytsearch)
if not results.tracks or results.load_type in (
LoadType.EMPTY,
LoadType.ERROR,
):
raise LoadError
first_track = results.tracks[ first_track = results.tracks[
0 0