pep.py/objects/matchList.py

65 lines
2.2 KiB
Python
Raw Normal View History

2016-05-18 17:12:46 +00:00
from objects import match
from objects import glob
from constants import serverPackets
from common.log import logUtils as log
2016-04-19 17:40:59 +00:00
2016-09-02 15:45:10 +00:00
class matchList:
2016-04-19 17:40:59 +00:00
def __init__(self):
"""Initialize a matchList object"""
self.matches = {}
self.lastID = 1
2017-08-01 22:22:57 +00:00
def createMatch(self, matchName, matchPassword, beatmapID, beatmapName, beatmapMD5, gameMode, hostUserID, isTourney=False):
2016-04-19 17:40:59 +00:00
"""
Add a new match to matches list
:param matchName: match name, string
:param matchPassword: match md5 password. Leave empty for no password
:param beatmapID: beatmap ID
:param beatmapName: beatmap name, string
:param beatmapMD5: beatmap md5 hash, string
:param gameMode: game mode ID. See gameModes.py
:param hostUserID: user id of who created the match
:return: match ID
2016-04-19 17:40:59 +00:00
"""
# Add a new match to matches list and create its stream
2016-04-19 17:40:59 +00:00
matchID = self.lastID
self.lastID+=1
2017-08-01 22:22:57 +00:00
self.matches[matchID] = match.match(matchID, matchName, matchPassword, beatmapID, beatmapName, beatmapMD5, gameMode, hostUserID, isTourney)
2016-04-19 17:40:59 +00:00
return matchID
def disposeMatch(self, matchID):
2016-04-19 17:40:59 +00:00
"""
Destroy match object with id = matchID
2016-04-19 17:40:59 +00:00
:param matchID: ID of match to dispose
:return:
2016-04-19 17:40:59 +00:00
"""
# Make sure the match exists
if matchID not in self.matches:
2016-04-19 17:40:59 +00:00
return
# Get match and disconnect all players
_match = self.matches[matchID]
for slot in _match.slots:
_token = glob.tokens.getTokenFromUserID(slot.userID, ignoreIRC=True)
if _token is None:
continue
_match.userLeft(_token, disposeMatch=False) # don't dispose the match twice when we remove all players
# Delete chat channel
glob.channels.removeChannel("#multi_{}".format(_match.matchID))
# Send matchDisposed packet before disposing streams
glob.streams.broadcast(_match.streamName, serverPackets.disposeMatch(_match.matchID))
# Dispose all streams
2016-12-26 09:33:05 +00:00
glob.streams.dispose(_match.streamName)
glob.streams.dispose(_match.playingStreamName)
glob.streams.remove(_match.streamName)
glob.streams.remove(_match.playingStreamName)
2016-04-19 17:40:59 +00:00
# Send match dispose packet to everyone in lobby
glob.streams.broadcast("lobby", serverPackets.disposeMatch(matchID))
del self.matches[matchID]
log.info("MPROOM{}: Room disposed manually".format(_match.matchID))