lyh пре 3 година
родитељ
комит
bb8cad0e9c
1 измењених фајлова са 15 додато и 0 уклоњено
  1. 15 0
      apps/finance/models.py

+ 15 - 0
apps/finance/models.py

@@ -107,6 +107,21 @@ class dbFinanceIncome(models.Model):
     check_time = models.DateTimeField(verbose_name=u"审核时间", null=True)
     check_status = models.PositiveSmallIntegerField(choices=settings.CHECK_STATUS_CHOICES, verbose_name=u"审核状态", default=settings.DEFAULT)
 
+
+    def save(self, *args, **kwargs):
+        if self.no == None or self.no == '':
+            now = timezone.now()
+            rows = dbFinanceIncome.objects.filter(create_time__gte=now.strftime('%Y-%m-%d'),  type=self.type).order_by('-no')
+            count = rows.count()
+            prefix = dbFinanceIncome.PREFIX_CHOICES[self.type-1][1]
+            if count == 0:
+                self.no = '%s%s%04d' % (prefix, now.strftime('%Y%m%d'), count + 1)
+            else:
+                suffix = (rows[0].no.split('-'))[1]
+                self.no = prefix+str(int(suffix) + 1)
+
+        super(dbFinanceIncome, self).save(*args, **kwargs)
+
     class Meta:
         db_table = "finance_income"
         verbose_name = u"收支管理"