forked from test/mau_mau_bot
115 lines
3.5 KiB
Python
115 lines
3.5 KiB
Python
import logging
|
|
|
|
from game import Game
|
|
from player import Player
|
|
|
|
|
|
class GameManager(object):
|
|
""" Manages all running games by using a confusing amount of dicts """
|
|
|
|
def __init__(self):
|
|
self.chatid_games = dict()
|
|
self.userid_players = dict()
|
|
self.userid_current = dict()
|
|
self.logger = logging.getLogger(__name__)
|
|
|
|
def new_game(self, chat):
|
|
"""
|
|
Generate a game join link with a unique ID and connect the game to the
|
|
group chat
|
|
"""
|
|
chat_id = chat.id
|
|
|
|
self.logger.info("Creating new game with id " + str(chat_id))
|
|
game = Game(chat)
|
|
|
|
if chat_id not in self.chatid_games:
|
|
self.chatid_games[chat_id] = list()
|
|
|
|
self.chatid_games[chat_id].append(game)
|
|
|
|
def join_game(self, chat_id, user):
|
|
""" Create a player from the Telegram user and add it to the game """
|
|
self.logger.info("Joining game with id " + str(chat_id))
|
|
try:
|
|
game = self.chatid_games[chat_id][-1]
|
|
except (KeyError, IndexError):
|
|
return None
|
|
|
|
if user.id not in self.userid_players:
|
|
self.userid_players[user.id] = list()
|
|
|
|
players = self.userid_players[user.id]
|
|
|
|
# Don not re-add a player and remove the player from previous games in
|
|
# this chat
|
|
for player in players:
|
|
if player in game.players:
|
|
return False
|
|
else:
|
|
self.leave_game(user, chat_id)
|
|
|
|
player = Player(game, user)
|
|
|
|
players.append(player)
|
|
self.userid_current[user.id] = player
|
|
return True
|
|
|
|
def leave_game(self, user, chat_id):
|
|
""" Remove a player from its current game """
|
|
try:
|
|
players = self.userid_players[user.id]
|
|
games = self.chatid_games[chat_id]
|
|
|
|
for player in players:
|
|
for game in games:
|
|
if player in game.players:
|
|
if player is game.current_player:
|
|
game.turn()
|
|
|
|
player.leave()
|
|
players.remove(player)
|
|
|
|
# If this is the selected game, switch to another
|
|
if self.userid_current[user.id] is player:
|
|
if len(players):
|
|
self.userid_current[user.id] = players[0]
|
|
else:
|
|
del self.userid_current[user.id]
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
except KeyError:
|
|
return False
|
|
|
|
def end_game(self, chat_id, user):
|
|
"""
|
|
Generate a game join link with a unique ID and connect the game to the
|
|
group chat
|
|
"""
|
|
|
|
self.logger.info("Game in chat " + str(chat_id) + " ended")
|
|
players = self.userid_players[user.id]
|
|
games = self.chatid_games[chat_id]
|
|
the_game = None
|
|
|
|
# Find the correct game instance to end
|
|
logging.info(str(players))
|
|
logging.info(str(games))
|
|
for player in players:
|
|
for game in games:
|
|
if player in game.players:
|
|
the_game = game
|
|
break
|
|
if the_game:
|
|
break
|
|
else:
|
|
return
|
|
|
|
for player in the_game.players:
|
|
self.userid_players[player.user.id].remove(player)
|
|
if len(self.userid_players[player.user.id]) is 0:
|
|
del self.userid_players[player.user.id]
|
|
self.chatid_games[chat_id].remove(the_game)
|
|
return
|