87 lines
3.3 KiB
Python
87 lines
3.3 KiB
Python
|
from django.db import models
|
||
|
from datetime import date
|
||
|
|
||
|
|
||
|
class Client(models.Model):
|
||
|
class Meta:
|
||
|
verbose_name = 'Client'
|
||
|
verbose_name_plural = 'Clients'
|
||
|
|
||
|
name = models.CharField(max_length=255, verbose_name="Nom")
|
||
|
address = models.CharField(max_length=255, verbose_name="Adresse")
|
||
|
postal_code = models.IntegerField(verbose_name="Code postal")
|
||
|
city = models.CharField(max_length=255, verbose_name="Ville")
|
||
|
contact = models.CharField(max_length=255, verbose_name="Personne de contact")
|
||
|
company_number = models.CharField(max_length=50, verbose_name="N° d'entreprise",
|
||
|
blank=True, null=True)
|
||
|
|
||
|
def __str__(self):
|
||
|
return "%s" % (self.name)
|
||
|
|
||
|
|
||
|
class Contract(models.Model):
|
||
|
class Meta:
|
||
|
verbose_name = 'Contrat'
|
||
|
verbose_name_plural = 'Contrats'
|
||
|
|
||
|
name = models.CharField(max_length=255, verbose_name="Nom")
|
||
|
client = models.ForeignKey(Client, on_delete=models.CASCADE, related_name="get_contract")
|
||
|
advance = models.DecimalField(max_digits=6, decimal_places=2, blank=True,
|
||
|
verbose_name="Acompte", default=0)
|
||
|
reference = models.CharField(max_length=255, verbose_name="Référence", blank=True, null=True)
|
||
|
is_finished = models.BooleanField(default=True, blank=True)
|
||
|
date = models.DateField(auto_now_add=True)
|
||
|
|
||
|
def __str__(self):
|
||
|
return "%s" % (self.name)
|
||
|
|
||
|
# def __generate_reference(self):
|
||
|
# """
|
||
|
# Génère automatiquement la référence du contract
|
||
|
# """
|
||
|
# if self.id is None:
|
||
|
# object_id = "XX"
|
||
|
# elif self.id < 10:
|
||
|
# object_id = "0" + str(self.id)
|
||
|
# else:
|
||
|
# object_id = self.id
|
||
|
# self.reference = str(self.date.year) + "-" + object_id
|
||
|
|
||
|
# def save(self, *args, **kwargs):
|
||
|
# """
|
||
|
# Enregistre l'objet en DB après avoir calculé le montant total.
|
||
|
# """
|
||
|
# super().save(*args, **kwargs)
|
||
|
# self.__generate_reference()
|
||
|
# super().save(*args, **kwargs)
|
||
|
|
||
|
|
||
|
class Prestation(models.Model):
|
||
|
class Meta:
|
||
|
verbose_name = 'Prestations'
|
||
|
verbose_name_plural = 'Prestations'
|
||
|
ordering = ['date']
|
||
|
|
||
|
contract = models.ForeignKey(Contract, on_delete=models.CASCADE, related_name="get_prestation", blank=True, null=True)
|
||
|
date = models.DateField()
|
||
|
label = models.CharField(max_length=255, verbose_name='Libellé')
|
||
|
unit = models.DecimalField(max_digits=5, decimal_places=2, verbose_name="Unité",
|
||
|
default=1)
|
||
|
unit_price = models.DecimalField(max_digits=5, decimal_places=2, verbose_name="Prix unitaire",
|
||
|
default="12,5")
|
||
|
total_amount = models.DecimalField(max_digits=6, decimal_places=2, blank=True,
|
||
|
verbose_name="Prix", default=0)
|
||
|
|
||
|
def __compute_total_amount(self):
|
||
|
"""
|
||
|
Calcule le montant total de la prestation.
|
||
|
"""
|
||
|
self.total_amount = self.unit * self.unit_price
|
||
|
|
||
|
def save(self, *args, **kwargs):
|
||
|
"""
|
||
|
Enregistre l'objet en DB après avoir calculé le montant total.
|
||
|
"""
|
||
|
self.__compute_total_amount()
|
||
|
super().save(*args, **kwargs)
|