mau_mau_bot/game.py

140 lines
4.4 KiB
Python
Raw Normal View History

2016-05-08 20:37:25 +08:00
#!/usr/bin/env python3
2016-05-20 05:15:46 +08:00
# -*- coding: utf-8 -*-
2016-05-08 20:37:25 +08:00
#
# 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/>.
2016-02-29 19:16:12 +08:00
import logging
from config import ADMIN_LIST, OPEN_LOBBY, DEFAULT_GAMEMODE, ENABLE_TRANSLATIONS
2016-04-30 19:27:41 +08:00
from datetime import datetime
2016-02-29 06:57:24 +08:00
from deck import Deck
import card as c
class Game(object):
2016-03-08 09:50:24 +08:00
""" This class represents a game of UNO """
2016-02-29 06:57:24 +08:00
current_player = None
reversed = False
choosing_color = False
started = False
draw_counter = 0
2016-05-23 01:21:51 +08:00
players_won = 0
starter = None
mode = DEFAULT_GAMEMODE
job = None
owner = ADMIN_LIST
open = OPEN_LOBBY
translate = ENABLE_TRANSLATIONS
2016-02-29 06:57:24 +08:00
2016-04-24 08:11:37 +08:00
def __init__(self, chat):
self.chat = chat
2016-05-22 00:56:27 +08:00
self.last_card = None
2016-04-19 06:41:48 +08:00
self.deck = Deck()
2016-04-19 06:41:48 +08:00
2016-02-29 19:16:12 +08:00
self.logger = logging.getLogger(__name__)
2016-02-29 06:57:24 +08:00
@property
def players(self):
"""Returns a list of all players in this game"""
players = list()
if not self.current_player:
return players
current_player = self.current_player
itplayer = current_player.next
players.append(current_player)
while itplayer and itplayer is not current_player:
players.append(itplayer)
itplayer = itplayer.next
return players
def start(self):
if self.mode == None or self.mode != "wild":
self.deck._fill_classic_()
else:
self.deck._fill_wild_()
self._first_card_()
self.started = True
def set_mode(self, mode):
self.mode = mode
2016-02-29 06:57:24 +08:00
def reverse(self):
"""Reverses the direction of game"""
2016-02-29 06:57:24 +08:00
self.reversed = not self.reversed
def turn(self):
"""Marks the turn as over and change the current player"""
2016-03-01 08:25:26 +08:00
self.logger.debug("Next Player")
2016-02-29 06:57:24 +08:00
self.current_player = self.current_player.next
2016-03-01 08:25:26 +08:00
self.current_player.drew = False
2016-04-30 19:27:41 +08:00
self.current_player.turn_started = datetime.now()
self.choosing_color = False
2016-02-29 06:57:24 +08:00
def _first_card_(self):
# In case that the player did not select a game mode
if not self.deck.cards:
self.set_mode(DEFAULT_GAMEMODE)
# The first card should not be a special card
while not self.last_card or self.last_card.special:
self.last_card = self.deck.draw()
# If the card drawn was special, return it to the deck and loop again
if self.last_card.special:
self.deck.dismiss(self.last_card)
self.play_card(self.last_card)
2016-02-29 06:57:24 +08:00
def play_card(self, card):
"""
Plays a card and triggers its effects.
Should be called only from Player.play or on game start to play the
first card
"""
2016-02-29 06:57:24 +08:00
self.deck.dismiss(self.last_card)
self.last_card = card
2016-02-29 19:16:12 +08:00
self.logger.info("Playing card " + repr(card))
if card.value == c.SKIP:
2016-03-01 08:25:26 +08:00
self.turn()
2016-02-29 19:16:12 +08:00
elif card.special == c.DRAW_FOUR:
2016-02-29 06:57:24 +08:00
self.draw_counter += 4
2016-02-29 19:16:12 +08:00
self.logger.debug("Draw counter increased by 4")
elif card.value == c.DRAW_TWO:
2016-02-29 06:57:24 +08:00
self.draw_counter += 2
2016-02-29 19:16:12 +08:00
self.logger.debug("Draw counter increased by 2")
elif card.value == c.REVERSE:
2016-03-08 09:50:24 +08:00
# Special rule for two players
if self.current_player is self.current_player.next.next:
self.turn()
else:
self.reverse()
2016-02-29 06:57:24 +08:00
2016-03-08 09:50:24 +08:00
# Don't turn if the current player has to choose a color
2016-02-29 06:57:24 +08:00
if card.special not in (c.CHOOSE, c.DRAW_FOUR):
2016-03-01 08:25:26 +08:00
self.turn()
2016-02-29 06:57:24 +08:00
else:
2016-02-29 19:16:12 +08:00
self.logger.debug("Choosing Color...")
2016-02-29 06:57:24 +08:00
self.choosing_color = True
def choose_color(self, color):
"""Carries out the color choosing and turns the game"""
2016-02-29 06:57:24 +08:00
self.last_card.color = color
2016-03-01 08:25:26 +08:00
self.turn()