mirror of
https://github.com/RafaelSolVargas/Vulkan.git
synced 2025-10-29 16:57:23 +00:00
63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
from multiprocessing import Queue, Lock
|
|
from multiprocessing.managers import BaseManager, NamespaceProxy
|
|
from typing import Dict
|
|
from Config.Singleton import Singleton
|
|
from discord import Guild, Client
|
|
from discord.ext.commands import Context
|
|
from Parallelism.PlayerProcess import PlayerProcess
|
|
from Music.Playlist import Playlist
|
|
from Parallelism.ProcessContext import ProcessContext
|
|
|
|
|
|
class ProcessManager(Singleton):
|
|
def __init__(self, bot: Client = None) -> None:
|
|
if not super().created:
|
|
Manager.register('Playlist', Playlist)
|
|
self.__manager = Manager()
|
|
self.__manager.start()
|
|
if bot is not None:
|
|
self.__bot: Client = bot
|
|
self.__playersProcess: Dict[Guild, ProcessContext] = {}
|
|
|
|
def setPlayerContext(self, guild: Guild, context: ProcessContext):
|
|
self.__playersProcess[guild] = context
|
|
|
|
def getPlayerContext(self, guild: Guild, context: Context) -> ProcessContext:
|
|
try:
|
|
if guild not in self.__playersProcess.keys():
|
|
self.__playersProcess[guild] = self.__createProcess(context)
|
|
else:
|
|
if not self.__playersProcess[guild].getProcess().is_alive():
|
|
self.__playersProcess[guild] = self.__createProcess(context)
|
|
|
|
return self.__playersProcess[guild]
|
|
except Exception as e:
|
|
print(f'[Error In GetPlayerContext] -> {e}')
|
|
|
|
def getRunningPlayerContext(self, guild: Guild) -> ProcessContext:
|
|
if guild not in self.__playersProcess.keys():
|
|
return None
|
|
|
|
return self.__playersProcess[guild]
|
|
|
|
def __createProcess(self, context: Context):
|
|
guildID: int = context.guild.id
|
|
textID: int = context.channel.id
|
|
voiceID: int = context.author.voice.channel.id
|
|
authorID: int = context.author.id
|
|
|
|
playlist: Playlist = self.__manager.Playlist()
|
|
lock = Lock()
|
|
queue = Queue()
|
|
process = PlayerProcess(playlist, lock, queue, guildID, textID, voiceID, authorID)
|
|
processContext = ProcessContext(process, queue, playlist, lock)
|
|
return processContext
|
|
|
|
|
|
class Manager(BaseManager):
|
|
pass
|
|
|
|
|
|
class ProxyBase(NamespaceProxy):
|
|
_exposed_ = ('__getattribute__', '__setattr__', '__delattr__')
|