123456789101112131415161718192021222324252627 |
- from django.db import models
- from django.core.validators import MaxValueValidator
- class Payment(models.Model):
- TRANSFERT = 0
- WITHDRAWAL = 1
- PAYMENT_CHOICES = (
- (TRANSFERT, 'Virement'),
- (WITHDRAWAL, 'Prélèvement'),
- )
- amount = models.DecimalField(max_digits=9, decimal_places=2, verbose_name='Montant')
- period = models.PositiveIntegerField(validators=[MaxValueValidator(12)], verbose_name='Période')
- payment_method = models.IntegerField(choices=PAYMENT_CHOICES, verbose_name='Méthode de paiement')
- class Meta:
- verbose_name = 'paiement'
- def __str__(self):
- s = str(self.amount) + '€'
- if self.period:
- s += '%d/mois' % self.period
- if self.payment_method == self.TRANSFERT:
- s += ' (virement)'
- elif self.payment_method == self.WITHDRAWAL:
- s += ' (prélèvement)'
- return s
|