pep.py/objects/streamList.py

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)