mtg-web/analyzer/models.py

import re
from django.db import models
from mtgweb.lib.mtg import mtg

# Create your models here.
class CardType(models.Model):
    name = models.CharField(max_length=200, unique=True, db_index=True)
    def __unicode__(self):
        return self.name
class Attribute(models.Model):
    name = models.CharField(max_length=200, unique=True, db_index=True)
    def __unicode__(self):
        return self.name
class Card(models.Model, mtg.Card):
    name = models.CharField(max_length=200, unique=True)
    type = models.ForeignKey(CardType)
    attributes = models.ManyToManyField(Attribute)
    cost = models.CharField(max_length=80)
    converted_cost = models.IntegerField(default=0)
    power = models.CharField(max_length=10)
    toughness = models.CharField(max_length=10)
    rarity = models.CharField(max_length=1)
    text = models.TextField()

    def __unicode__(self):
        return self.name
class Deck(models.Model):
    name = models.CharField(max_length=80)
    cards = models.ManyToManyField(Card, through='Included')
    def __unicode__(self):
        return self.name
    def colors(self):
        symbols = {}
        for symbol in mtg.Mana.types.keys():
            symbols[symbol] = 0
        total = 0
        for card in self.cards.all():
            # Get symbols from card cost
            cost = mtg.ManaCost(card.cost)
            for color, count in cost.mana.mana.iteritems():
                symbols[color] += count
                total += count

            # Get symbols from abilities
            pattern = '{%s}' % mtg.ManaCost.symbolPattern
            costs = [mtg.ManaCost(cost) for cost in re.findall(pattern, str(card.text))]
            for cost in costs:
                for color, count in cost.mana.mana.iteritems():
                    symbols[color] += count
                    total += count
        return (symbols, total)
class Included(models.Model):
    card = models.ForeignKey(Card)
    deck = models.ForeignKey(Deck)
    count = models.IntegerField(default=0)
    def __unicode__(self):
        return '{0}x {1}'.format(self.count, self.card)