Migrate invoices and lot of eboutic improvements

eboutic/models.py

@@ -1,8 +1,9 @@
 from django.db import models, DataError
 from django.utils.translation import ugettext_lazy as _
+from django.conf import settings
 
 from accounting.models import CurrencyField
-from counter.models import Counter, Product, Customer
+from counter.models import Counter, Product, Customer, Selling, Refilling
 from core.models import User
 
 class Basket(models.Model):
@@ -15,7 +16,7 @@ class Basket(models.Model):
     def add_product(self, p, q = 1):
         item = self.items.filter(product_id=p.id).first()
         if item is None:
-            BasketItem(basket=self, product_id=p.id, product_name=p.name, type=p.product_type.name,
+            BasketItem(basket=self, product_id=p.id, product_name=p.name, type_id=p.product_type.id,
                     quantity=q, product_unit_price=p.selling_price).save()
         else:
             item.quantity += q
@@ -44,10 +45,11 @@ class Invoice(models.Model):
     """
     user = models.ForeignKey(User, related_name='invoices', verbose_name=_('user'), blank=False)
     date = models.DateTimeField(_('date'), auto_now=True)
-    payment_method = models.CharField(choices=[('CREDIT_CARD', _('Credit card')), ('SITH_ACCOUNT', _('Sith account'))],
-            max_length=20, verbose_name=_('payment method'))
     validated = models.BooleanField(_("validated"), default=False)
 
+    def __str__(self):
+        return "%s - %s - %s" % (self.user, self.get_total(), self.date)
+
     def get_total(self):
         total = 0
         for i in self.items.all():
@@ -59,23 +61,44 @@ class Invoice(models.Model):
             raise DataError(_("Invoice already validated"))
         from counter.models import Customer
         if not Customer.objects.filter(user=self.user).exists():
-            number = Customer.objects.last().account_id[:-1]
+            number = Customer.objects.count() + 1
             Customer(user=self.user, account_id=Customer.generate_account_id(number), amount=0).save()
-        if self.payment_method == "SITH_ACCOUNT":
-            self.user.customer.amount -= self.get_total()
-            self.user.customer.save()
-        else:
-            for i in self.items.filter(type="REFILLING").all():
-                self.user.customer.amount += i.product_unit_price * i.quantity
-            self.user.customer.save()
-
+        eboutic = Counter.objects.filter(type="EBOUTIC").first()
+        for i in self.items.all():
+            if i.type_id == settings.SITH_COUNTER_PRODUCTTYPE_REFILLING:
+                new = Refilling(
+                        counter=eboutic,
+                        customer=self.user.customer,
+                        operator=self.user,
+                        amount=i.product_unit_price * i.quantity,
+                        payment_method="CARD",
+                        bank="OTHER",
+                        date=self.date,
+                        )
+                new.save()
+            else:
+                product = Product.objects.filter(id=i.product_id).first()
+                new = Selling(
+                        label=i.product_name,
+                        counter=eboutic,
+                        club=product.club,
+                        product=product,
+                        seller=self.user,
+                        customer=self.user.customer,
+                        unit_price=i.product_unit_price,
+                        quantity=i.quantity,
+                        payment_method="CARD",
+                        is_validated=True,
+                        date=self.date,
+                        )
+                new.save()
         self.validated = True
         self.save()
 
 class AbstractBaseItem(models.Model):
     product_id = models.IntegerField(_('product id'))
     product_name = models.CharField(_('product name'), max_length=255)
-    type = models.CharField(_('product type'), max_length=255)
+    type_id = models.IntegerField(_('product type id'))
     product_unit_price = CurrencyField(_('unit price'))
     quantity = models.IntegerField(_('quantity'))