94 lines
2.1 KiB
Python
94 lines
2.1 KiB
Python
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)
|
|
|
|
def getStream(self, streamName):
|
|
"""
|
|
Returns streamName's stream object or None if it doesn't exist
|
|
|
|
:param streamName:
|
|
:return:
|
|
"""
|
|
if streamName in self.streams:
|
|
return self.streams[streamName]
|
|
return None |