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