mau_mau_bot/game_manager.py
2016-05-08 14:37:25 +02:00

135 lines
4.3 KiB
Python

#!/usr/bin/env python3
#
# Telegram bot to play UNO in group chats
# Copyright (c) 2016 Jannes Höke <uno@jhoeke.de>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
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)
return 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