#!/usr/bin/env python3 # # Telegram bot to play UNO in group chats # Copyright (c) 2016 Jannes Höke # # 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 . from random import shuffle import card as c from card import Card import logging class Deck(object): """ This class represents a deck of cards """ def __init__(self): self.cards = list() self.graveyard = list() self.logger = logging.getLogger(__name__) # Fill deck for color in c.COLORS: for value in c.VALUES: self.cards.append(Card(color, value)) if not value == c.ZERO: self.cards.append(Card(color, value)) for special in c.SPECIALS * 4: self.cards.append(Card(None, None, special=special)) self.logger.debug(self.cards) self.shuffle() def shuffle(self): """ Shuffle the deck """ self.logger.debug("Shuffling Deck") shuffle(self.cards) def draw(self): """ Draw a card from this deck """ try: card = self.cards.pop() if card.special: card = Card(None, None, card.special) self.logger.debug("Drawing card " + str(card)) return card except IndexError: while len(self.graveyard): self.cards.append(self.graveyard.pop()) self.shuffle() return self.draw() def dismiss(self, card): """ All played cards should be returned into the deck """ # if card.special: # card.color = None self.graveyard.append(card)