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
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)