Add products on billings

apps/treasury/__init__.py

@@ -0,0 +1,4 @@
+# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+default_app_config = 'treasury.apps.TreasuryConfig'

apps/treasury/admin.py

@@ -0,0 +1,16 @@
+# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from django.contrib import admin
+
+from treasury.models import Billing, Product
+
+
+@admin.register(Billing)
+class BillingAdmin(admin.ModelAdmin):
+    list_display = ('id', 'name', 'subject', 'acquitted', )
+
+
+@admin.register(Product)
+class ProductAdmin(admin.ModelAdmin):
+    list_display = ('designation', 'quantity', 'amount', )

apps/treasury/forms.py

@@ -0,0 +1,29 @@
+# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from crispy_forms.helper import FormHelper
+from django import forms
+
+from .models import Billing, Product
+
+
+class BillingForm(forms.ModelForm):
+    class Meta:
+        model = Billing
+        fields = '__all__'
+
+
+ProductFormSet = forms.inlineformset_factory(
+    Billing,
+    Product,
+    fields='__all__',
+    extra=1,
+)
+
+class ProductFormSetHelper(FormHelper):
+    def __init__(self, form=None):
+        super().__init__(form)
+        self.form_tag = False
+        self.form_method = 'POST'
+        self.form_class = 'form-inline'
+        self.template = 'bootstrap4/table_inline_formset.html'

apps/treasury/models.py

@@ -2,6 +2,7 @@
 # SPDX-License-Identifier: GPL-3.0-or-later
 
 from django.db import models
+from django.urls import reverse_lazy
 from django.utils.translation import gettext_lazy as _
 
 
@@ -122,6 +123,14 @@ class Product(models.Model):
         verbose_name=_("Unit price")
     )
 
+    @property
+    def amount_euros(self):
+        return self.amount / 100
+
     @property
     def total(self):
         return self.quantity * self.amount
+
+    @property
+    def total_euros(self):
+        return self.total / 100

apps/treasury/tables.py

@@ -14,7 +14,10 @@ class BillingTable(tables.Table):
                                args=[A("pk")],
                                accessor="pk",
                                text="",
-                               attrs={'a': {'class': 'fa fa-file-pdf-o'}})
+                               attrs={
+                                   'a': {'class': 'fa fa-file-pdf-o'},
+                                   'td': {'data-turbolinks': 'false'}
+                               })
 
     class Meta:
         attrs = {

apps/treasury/views.py

@@ -6,15 +6,20 @@ import shutil
 import subprocess
 from tempfile import mkdtemp
 
+from crispy_forms.helper import FormHelper
 from django.contrib.auth.mixins import LoginRequiredMixin
+from django.db.models import Q
 from django.http import HttpResponse
 from django.template.loader import render_to_string
+from django.urls import reverse_lazy
 from django.views.generic import CreateView, UpdateView
 from django.views.generic.base import View
 from django_tables2 import SingleTableView
+
 from note_kfet.settings.base import BASE_DIR
 
-from .models import Billing
+from .forms import BillingForm, ProductFormSet, ProductFormSetHelper
+from .models import Billing, Product
 from .tables import BillingTable
 
 
@@ -23,8 +28,36 @@ class BillingCreateView(LoginRequiredMixin, CreateView):
     Create Billing
     """
     model = Billing
-    fields = '__all__'
-    # form_class = ClubForm
+    form_class = BillingForm
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        form = context['form']
+        form.helper = FormHelper()
+        form.helper.form_tag = False
+        form_set = ProductFormSet(instance=form.instance)
+        context['formset'] = form_set
+        context['helper'] = ProductFormSetHelper()
+        context['no_cache'] = True
+
+        return context
+
+    def form_valid(self, form):
+        ret = super().form_valid(form)
+
+        formset = ProductFormSet(self.request.POST, instance=form.instance)
+        if formset.is_valid():
+            for f in formset:
+                if f.is_valid() and f.instance.designation:
+                    f.save()
+                    f.instance.save()
+                else:
+                    f.instance = None
+
+        return ret
+
+    def get_success_url(self):
+        return reverse_lazy('treasury:billing')
 
 
 class BillingListView(LoginRequiredMixin, SingleTableView):
@@ -40,8 +73,42 @@ class BillingUpdateView(LoginRequiredMixin, UpdateView):
     Create Billing
     """
     model = Billing
-    fields = '__all__'
-    # form_class = BillingForm
+    form_class = BillingForm
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        form = context['form']
+        form.helper = FormHelper()
+        form.helper.form_tag = False
+        form_set = ProductFormSet(instance=form.instance)
+        context['formset'] = form_set
+        context['helper'] = ProductFormSetHelper()
+        context['no_cache'] = True
+
+        return context
+
+    def form_valid(self, form):
+        ret = super().form_valid(form)
+
+        formset = ProductFormSet(self.request.POST, instance=form.instance)
+        saved = []
+        if formset.is_valid():
+            for f in formset:
+                if f.is_valid() and f.instance.designation:
+                    if type(f.instance.pk) == 'number' and f.instance.pk <= 0:
+                        f.instance.pk = None
+                    f.save()
+                    f.instance.save()
+                    saved.append(f.instance.pk)
+                else:
+                    f.instance = None
+
+        Product.objects.filter(~Q(pk__in=saved), billing=form.instance).delete()
+
+        return ret
+
+    def get_success_url(self):
+        return reverse_lazy('treasury:billing')
 
 
 class BillingRenderView(LoginRequiredMixin, View):
@@ -52,10 +119,11 @@ class BillingRenderView(LoginRequiredMixin, View):
     def get(self, request, **kwargs):
         pk = kwargs["pk"]
         billing = Billing.objects.get(pk=pk)
+        products = Product.objects.filter(billing=billing).all()
 
         billing.description = billing.description.replace("\n", "\\newline\n")
         billing.address = billing.address.replace("\n", "\\newline\n")
-        tex = render_to_string("treasury/billing_sample.tex", dict(obj=billing))
+        tex = render_to_string("treasury/billing_sample.tex", dict(obj=billing, products=products))
         try:
             os.mkdir(BASE_DIR + "/tmp")
         except FileExistsError: