Use proper and legal sequential invoice numbers

Replacing previous random invoice numbers. The existing numbers remain
untouched in the DB, but no collision is possible because the number format is
different:

- old: "YYYYMM-RRR-RRR" (RRR-RRR being random numbers)
- new: "YYYY-MM-NNNNNN" (NNNNNN being sequence, local to the month)

fix #4

coin/billing/models.py

@@ -5,11 +5,13 @@ import datetime
 import random
 import uuid
 import os
+import re
 from decimal import Decimal
 
 from django.db import models
 from django.db.models.signals import post_save
 from django.dispatch import receiver
+from django.utils.encoding import python_2_unicode_compatible
 
 from coin.offers.models import OfferSubscription
 from coin.members.models import Member
@@ -33,6 +35,58 @@ def invoice_pdf_filename(instance, filename):
                                       instance.number,
                                       uuid.uuid4())
 
+@python_2_unicode_compatible
+class InvoiceNumber:
+    RE_INVOICE_NUMBER = re.compile(
+        r'(?P<year>\d{4})-(?P<month>\d{2})-(?P<index>\d{6})')
+
+    def __init__(self, date, index):
+        self.date = date
+        self.index = index
+
+    def get_next(self):
+        return InvoiceNumber(self.date, self.index + 1)
+
+    def __str__(self):
+        return '{:%Y-%m}-{:0>6}'.format(self.date, self.index)
+
+    @classmethod
+    def parse(cls, string):
+        m = cls.RE_INVOICE_NUMBER.match(string)
+        if not m:
+            raise ValueError('Not a valid invoice number: "{}"'.format(string))
+
+        return cls(
+            datetime.date(
+                year=int(m.group('year')),
+                month=int(m.group('month')),
+                day=1),
+            int(m.group('index')))
+
+
+class InvoiceQuerySet(models.QuerySet):
+    def get_next_invoice_number(self, date):
+        last_invoice_number_str = self._get_last_invoice_number(date)
+
+        if last_invoice_number_str is None:
+            # It's the first bill of the month
+            invoice_number = InvoiceNumber(date, 1)
+        else:
+            invoice_number = InvoiceNumber.parse(last_invoice_number_str).get_next()
+
+        return str(invoice_number)
+
+    def _get_last_invoice_number(self, date):
+        return self.with_valid_number().aggregate(
+            models.Max('number'))['number__max']
+
+    def with_valid_number(self):
+        """ Used for number format/logic migration
+
+        - from YYYYMM-RRR-RRR (RRR-RRR being random numbers)
+        - to YYYY-MM-NNNNNN (NNNNNN being sequence)
+        """
+        return self.filter(number__regex=InvoiceNumber.RE_INVOICE_NUMBER.pattern)
 
 class Invoice(models.Model):
 
@@ -121,6 +175,10 @@ class Invoice(models.Model):
             self.save()
             self.generate_pdf()
 
+    def save(self, *args, **kwargs):
+        self.number = Invoice.objects.get_next_invoice_number(self.date)
+        super(Invoice, self).save(*args, **kwargs)
+
     def is_pdf_exists(self):
         return (self.validated
                 and bool(self.pdf)
@@ -136,6 +194,7 @@ class Invoice(models.Model):
     class Meta:
         verbose_name = 'facture'
 
+    objects = InvoiceQuerySet().as_manager()
 
 class InvoiceDetail(models.Model):
 

coin/billing/tests.py

@@ -8,7 +8,7 @@ from django.conf import settings
 from django.test import TestCase, Client
 from coin.members.tests import MemberTestsUtils
 from coin.members.models import Member, LdapUser
-from coin.billing.models import Invoice
+from coin.billing.models import Invoice, InvoiceQuerySet
 from coin.offers.models import Offer, OfferSubscription
 from coin.billing.create_subscriptions_invoices import create_member_invoice_for_a_period
 from coin.billing.create_subscriptions_invoices import create_all_members_invoices_for_a_period
@@ -222,3 +222,16 @@ class BillingTests(TestCase):
 
         member_a.delete()
         member_b.delete()
+
+
+class InvoiceQuerySetTests(TestCase):
+    def test_get_first_of_month_invoice_number(self):
+        self.assertEqual(
+            Invoice.objects.get_next_invoice_number(datetime.date(2016,1,1)),
+            '2016-01-000001')
+
+    def test_get_second_of_month_invoice_number(self):
+        Invoice.objects.create(date=datetime.date(2016,1,1))
+        self.assertEqual(
+            Invoice.objects.get_next_invoice_number(datetime.date(2016,1,1)),
+            '2016-01-000002')