models with prox

gistfile1.txt

# -*- coding: utf-8 -*-

from django.db import models
import uuid

ANIMALS = (('DO', 'Dog'), ('CA', 'Cat'))


class Animal(models.Model):
    tipo = models.CharField(max_length=2, choices=ANIMALS)
    nome = models.CharField(max_length=50)
    idade = models.CharField(max_length=50)
    codigo = models.CharField(max_length=50, unique=True, null=True, blank=True)

    def __unicode__(self):
        return self.nome

    def save(self, *args, **kwargs):
        self.codigo = uuid.uuid1()
        super(Animal, self).save(*args, **kwargs)

    def falar(self):
        pass


class Gato(Animal):
    felino = models.CharField(max_length=2)

    def __unicode__(self):
        return self.nome

    def falar(self):
        print "%s disse: Miau!" % self.nome
        return "%s disse: Miau!" % self.nome


class Cachorro(Animal):
    dog = models.CharField(max_length=2)

    def __unicode__(self):
        return self.nome

    def falar(self):
        print "%s disse: Miau!" % self.nome
        return "%s disse: Miau!" % self.nome


class Vacina(models.Model):
    animal = models.ForeignKey(Animal)
    data = models.DateField(auto_now=True)
    tipo = models.CharField(max_length=50)
    doutor = models.CharField(max_length=50)

    def __unicode__(self):
        return self.tipo


class Doutor(models.Model):
    nome = models.CharField(max_length=50)
    tipo = models.CharField(max_length=50)
    animais = models.ManyToManyField(Animal)

    def __init__(self, nome):
        self.nome = nome
        self.restricao_doutor_e_animal_id = [(u"João",'1'), (u"Carlos",'5')]

    def __unicode__(self):
        return 'Doutor %s' % self.nome

    def tratarAnimal(self, animal):
        primeiro_char_codigo_animal = animal.codigo[0]
        for nome_doutor, _id in self.restricao_doutor_e_animal_id:
            if self.nome == nome_doutor:
                if primeiro_char_codigo_animal != _id:
                    raise Exception(u"Animais tratados com %s devem ter número de registro iniciando em %s" % (self.nome, _id))

class FactoryTratador(object):
    def getTratador(self, animal):
        if animal.__class__ is Gato:
            return Doutor(nome=u"João")