mau_mau_bot/deck.py

67 lines
2.1 KiB
Python
Raw Normal View History

2016-05-08 20:37:25 +08:00
#!/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/>.
2016-02-29 06:57:24 +08:00
from random import shuffle
2016-03-08 09:50:24 +08:00
import card as c
2016-02-29 06:57:24 +08:00
from card import Card
import logging
class Deck(object):
2016-03-08 09:50:24 +08:00
""" This class represents a deck of cards """
2016-02-29 06:57:24 +08:00
def __init__(self):
self.cards = list()
self.graveyard = list()
self.logger = logging.getLogger(__name__)
2016-03-08 09:50:24 +08:00
# Fill deck
for color in c.COLORS:
for value in c.VALUES:
2016-02-29 06:57:24 +08:00
self.cards.append(Card(color, value))
2016-03-08 09:50:24 +08:00
if not value == c.ZERO:
2016-02-29 06:57:24 +08:00
self.cards.append(Card(color, value))
2016-03-08 09:50:24 +08:00
for special in c.SPECIALS * 4:
2016-02-29 06:57:24 +08:00
self.cards.append(Card(None, None, special=special))
self.logger.debug(self.cards)
self.shuffle()
def shuffle(self):
2016-03-08 09:50:24 +08:00
""" Shuffle the deck """
2016-02-29 08:53:59 +08:00
self.logger.debug("Shuffling Deck")
shuffle(self.cards)
2016-02-29 06:57:24 +08:00
def draw(self):
2016-03-08 09:50:24 +08:00
""" Draw a card from this deck """
2016-02-29 06:57:24 +08:00
try:
2016-02-29 08:53:59 +08:00
card = self.cards.pop()
self.logger.debug("Drawing card " + str(card))
return card
2016-02-29 06:57:24 +08:00
except IndexError:
while len(self.graveyard):
self.cards.append(self.graveyard.pop())
self.shuffle()
return self.draw()
def dismiss(self, card):
2016-03-08 09:50:24 +08:00
""" All played cards should be returned into the deck """
2016-02-29 06:57:24 +08:00
self.graveyard.append(card)