from objects import stream from objects import glob # TODO: use *args and **kwargs class streamList: def __init__(self): self.streams = {} def add(self, name): """ Create a new stream list if it doesn't already exist :param name: stream name :return: """ if name not in self.streams: self.streams[name] = stream.stream(name) def remove(self, name): """ Removes an existing stream and kick every user in it :param name: stream name :return: """ if name in self.streams: for i in self.streams[name].clients: if i in glob.tokens.tokens: glob.tokens.tokens[i].leaveStream(name) self.streams.pop(name) def join(self, streamName, client=None, token=None): """ Add a client to a stream :param streamName: stream name :param client: client (osuToken) object :param token: client uuid string :return: """ if streamName not in self.streams: return self.streams[streamName].addClient(client=client, token=token) def leave(self, streamName, client=None, token=None): """ Remove a client from a stream :param streamName: stream name :param client: client (osuToken) object :param token: client uuid string :return: """ if streamName not in self.streams: return self.streams[streamName].removeClient(client=client, token=token) def broadcast(self, streamName, data, but=None): """ Send some data to all clients in a stream :param streamName: stream name :param data: data to send :param but: array of tokens to ignore. Default: None (send to everyone) :return: """ if streamName not in self.streams: return self.streams[streamName].broadcast(data, but) def dispose(self, streamName, *args, **kwargs): """ Call `dispose` on `streamName` :param streamName: name of the stream :param args: :param kwargs: :return: """ if streamName not in self.streams: return self.streams[streamName].dispose(*args, **kwargs)