from django.db import models   
import datetime

class SkladAddArticle(models.Model):
    date_in = models.DateField(u'Дата поступления')
    name = models.CharField(u'Артикул',maxlength=20,unique=True)
    sumin = models.IntegerField(u'Кол-во')

    def was_published_today(self):
 	return self.date_in.date() == datetime.date.today()

    class Admin:
    	list_display = ('date_in','name','sumin')

    def __unicode__(self):
    	return '%s %d' % (self.name, self.sumin)

class SkladAddSum(models.Model):
    article = models.CharField(u'Артикул', maxlength=20)
    addsum = models.IntegerField(u'Количество добавляемых позциий')
    SkladAddArticle = models.ForeignKey(SkladAddArticle, verbose_name=u"Новый артикул", blank=True, null=True)

    class Admin:
	pass

    def was_published_today(self):
        return self.date_in.date() == datetime.date.today()

    def __unicode__(self):
    	return '%d %d %s %s' % (self.addsum, self.SkladAddArticle.sum, self.article, self.SkladAddArticle.name)

    def save(self):
        b = SkladAddSum.objects.filter(article=SkladAddArticle.name)
        if b:
           p = SkladAddArticle.objects.filter(SkladAddSum__article=name).values(SkladAddArticle.sum)
           self.SkladAddArticle.sumin += p
           r = SkladAddArticle.save()
           return super(SkladAddArticle, self).save
        else:
           return None