card.py

from copy import copy
import numpy as np

class BaseCard(object):
    """docstring for Card"""

    CARD_NUMBER = xrange(2, 15)
    CARD_SHAPE =  ['H', 'S', 'C', 'D'] # H(红心), S(黑桃), C(草花), D(方块)

    PIG = ('S', 12)
    STAR = ('H', 2) # - H14


    def __init__(self):
        super(Card, self).__init__()

        self.num_func = lambda x: x[1]

        self._cards_db = []
        for num in self.CARD_NUMBER:
            for shape in self.CARD_SHAPE:
                self._cards_db.append((shape, num))

        self._cards_db = np.array(self._cards_db)        

class PlayerCards(BaseCard):
    """docstring for PlayerCard"""
    def __init__(self, cards_list):
        super(PlayerCard, self).__init__()
        self.raw_cards_list = cards_list

        self.FIRST_MASTER = ('C', 2)

        self.ONSHOWN = [('H', 14), ('S', 12), ('C', 10)] #  

        self.current_cards_list = copy(self.raw_cards_list)

        self._flush()

        self.onshown_cards_list = []

    def _flush(self):
        self.current_cards_list.sort(key=self.num_func)

        # get current dict
        self.current_cards_dict = {
            'H': [],
            'S': [],
            'C': [],
            'D': []
        }
        for shape, num in self.current_cards_list:
            self.current_cards_dict[shape].append((shape, num))
        for shape in self.current_cards_dict:
            self.current_cards_dict[shape].sort(self.num_func)

        # get length dict
        self.current_length_list = [(shape, len(self.current_length_dict[shape])) for shape in self.current_cards_dict]
        self.current_length_list.sort(self.num_func)

    def drop_one_card(self, card):
        self.current_cards_list.remove(card)
        self._flush()

    def drop_cards(self, cards):
        for card in cards:
            self.current_cards_list.remove(card)
        self._flush()

    def add_one_card(self, card):
        self.current_cards_list.append(card)
        self._flush()

    def add_cards(self, cards):
        self.current_cards_list.extend(cards)
        self._flush()

    def is_first_master(self):
        return True if self.FIRST_MASTER in self.current_cards_list else False

    def cards_can_show(self):
        return [card for card in self.ONSHOWN if card in self.current_cards_list]

    def add_shown_cards(self, cards):
        self.onshown_cards_list.extend(cards)

    def get_opposite_cards(self, cards_list):
        cards_opposite = copy(self._cards_db)
        for card in cards_list:
            cards_opposite.remove(card)

        return cards_opposite


class MasterCards(BaseCard):
    """docstring for MasterCards"""
    def __init__(self):
        super(MasterCards, self).__init__()

    def deal(self, num_of_player=4):
        np.random.shuffle(self._cards_db)
        self.dealed_cards = np.split(self._cards_db, num_of_player)

    def output_cards(self, round):
        """
            round: 1-4
        """

        return self.dealed_cards[round:] + self.dealed_cards[:round]