pep.py/objects/streamList.py

from objects import stream
from objects import glob

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):
		"""
		Send some data to all clients in a stream

		:param streamName: stream name
		:param data: data to send
		:return:
		"""
		if streamName not in self.streams:
			return
		self.streams[streamName].broadcast(data)

	'''def getClients(self, streamName):
		"""
		Get all clients in a stream

		:param streamName: name of the stream
		:return:
		"""
		if streamName not in self.streams:
			return
		return self.streams[streamName].clients'''