diff options
Diffstat (limited to 'servo/models/invoices.py')
-rw-r--r-- | servo/models/invoices.py | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/servo/models/invoices.py b/servo/models/invoices.py index 2154377..cbcbd66 100644 --- a/servo/models/invoices.py +++ b/servo/models/invoices.py @@ -14,7 +14,7 @@ from servo.models import (User, Customer, Order, Location, class Invoice(models.Model): created_at = models.DateTimeField(editable=False, auto_now_add=True) - created_by = models.ForeignKey(settings.AUTH_USER_MODEL, editable=False) + created_by = models.ForeignKey(settings.AUTH_USER_MODEL, editable=False, on_delete=models.SET_NULL) PAYMENT_METHODS = ( (0, _("No Charge")), @@ -34,12 +34,13 @@ class Invoice(models.Model): is_paid = models.BooleanField(default=False, verbose_name=_("Paid")) paid_at = models.DateTimeField(null=True, editable=False) - order = models.ForeignKey(Order, editable=False) + order = models.ForeignKey(Order, editable=False, on_delete=models.PROTECT) location = models.ForeignKey( Location, null=True, blank=True, - editable=False + editable=False, + on_delete=models.SET_NULL ) customer = models.ForeignKey( Customer, @@ -115,7 +116,7 @@ class Invoice(models.Model): soi.save() def get_absolute_url(self): - from django.core.urlresolvers import reverse + from django.urls import reverse return reverse("invoices-view_invoice", args=[self.pk]) def save(self, *args, **kwargs): @@ -135,7 +136,8 @@ class Invoice(models.Model): class InvoiceItem(AbstractOrderItem): - invoice = models.ForeignKey(Invoice) + """A line item on an invoice.""" + invoice = models.ForeignKey(Invoice, on_delete=models.CASCADE) price = models.DecimalField( max_digits=8, decimal_places=2, @@ -168,7 +170,7 @@ class InvoiceItem(AbstractOrderItem): class Payment(models.Model): - invoice = models.ForeignKey(Invoice) + invoice = models.ForeignKey(Invoice, on_delete=models.CASCADE) METHODS = ( (0, _("No Charge")), (1, _("Cash")), @@ -182,7 +184,7 @@ class Payment(models.Model): default=METHODS[0][0], verbose_name=_("Payment Method") ) - created_by = models.ForeignKey(User) + created_by = models.ForeignKey(User, on_delete=models.SET_NULL) created_at = models.DateTimeField(auto_now_add=True) amount = models.DecimalField(max_digits=8, decimal_places=2) |