mirror of
https://github.com/RafaelSolVargas/Vulkan.git
synced 2025-10-29 16:57:23 +00:00
52 lines
2.0 KiB
Python
52 lines
2.0 KiB
Python
from discord.ext.commands import Context
|
|
from discord import Client
|
|
from Handlers.AbstractHandler import AbstractHandler
|
|
from Handlers.HandlerResponse import HandlerResponse
|
|
from Config.Exceptions import BadCommandUsage
|
|
from Parallelism.ProcessManager import ProcessManager
|
|
|
|
|
|
class LoopHandler(AbstractHandler):
|
|
def __init__(self, ctx: Context, bot: Client) -> None:
|
|
super().__init__(ctx, bot)
|
|
|
|
async def run(self, args: str) -> HandlerResponse:
|
|
# Get the current process of the guild
|
|
processManager = ProcessManager()
|
|
processContext = processManager.getRunningPlayerContext(self.guild)
|
|
if not processContext:
|
|
embed = self.embeds.NOT_PLAYING()
|
|
error = BadCommandUsage()
|
|
return HandlerResponse(self.ctx, embed, error)
|
|
|
|
playlist = processContext.getPlaylist()
|
|
|
|
with processContext.getLock():
|
|
if args == '' or args is None:
|
|
playlist.loop_all()
|
|
embed = self.embeds.LOOP_ALL_ACTIVATED()
|
|
return HandlerResponse(self.ctx, embed)
|
|
|
|
args = args.lower()
|
|
if playlist.getCurrentSong() is None:
|
|
embed = self.embeds.NOT_PLAYING()
|
|
error = BadCommandUsage()
|
|
return HandlerResponse(self.ctx, embed, error)
|
|
|
|
if args == 'one':
|
|
playlist.loop_one()
|
|
embed = self.embeds.LOOP_ONE_ACTIVATED()
|
|
return HandlerResponse(self.ctx, embed)
|
|
elif args == 'all':
|
|
playlist.loop_all()
|
|
embed = self.embeds.LOOP_ALL_ACTIVATED()
|
|
return HandlerResponse(self.ctx, embed)
|
|
elif args == 'off':
|
|
playlist.loop_off()
|
|
embed = self.embeds.LOOP_DISABLE()
|
|
return HandlerResponse(self.ctx, embed)
|
|
else:
|
|
error = BadCommandUsage()
|
|
embed = self.embeds.BAD_LOOP_USE()
|
|
return HandlerResponse(self.ctx, embed, error)
|