import logging
from collections import OrderedDict
from decimal import Decimal as D
from django.conf import settings
from django.core.signing import BadSignature, Signer
from django.db import models
from django.db.models import Sum
from django.utils import timezone
from django.utils.crypto import constant_time_compare
from django.utils.timezone import now
from django.utils.translation import gettext_lazy as _
from django.utils.translation import pgettext_lazy
from oscar.apps.order.signals import (
order_line_status_changed, order_status_changed)
from oscar.core.compat import AUTH_USER_MODEL
from oscar.core.loading import get_model
from oscar.core.utils import get_default_currency
from oscar.models.fields import AutoSlugField
from . import exceptions
logger = logging.getLogger('oscar.order')
[docs]class AbstractOrder(models.Model):
"""
The main order model
"""
number = models.CharField(
_("Order number"), max_length=128, db_index=True, unique=True)
# We track the site that each order is placed within
site = models.ForeignKey(
'sites.Site', verbose_name=_("Site"), null=True,
on_delete=models.SET_NULL)
basket = models.ForeignKey(
'basket.Basket', verbose_name=_("Basket"),
null=True, blank=True, on_delete=models.SET_NULL)
# Orders can be placed without the user authenticating so we don't always
# have a customer ID.
user = models.ForeignKey(
AUTH_USER_MODEL, related_name='orders', null=True, blank=True,
verbose_name=_("User"), on_delete=models.SET_NULL)
# Billing address is not always required (eg paying by gift card)
billing_address = models.ForeignKey(
'order.BillingAddress', null=True, blank=True,
verbose_name=_("Billing Address"),
on_delete=models.SET_NULL)
# Total price looks like it could be calculated by adding up the
# prices of the associated lines, but in some circumstances extra
# order-level charges are added and so we need to store it separately
currency = models.CharField(
_("Currency"), max_length=12, default=get_default_currency)
total_incl_tax = models.DecimalField(
_("Order total (inc. tax)"), decimal_places=2, max_digits=12)
total_excl_tax = models.DecimalField(
_("Order total (excl. tax)"), decimal_places=2, max_digits=12)
# Shipping charges
shipping_incl_tax = models.DecimalField(
_("Shipping charge (inc. tax)"), decimal_places=2, max_digits=12,
default=0)
shipping_excl_tax = models.DecimalField(
_("Shipping charge (excl. tax)"), decimal_places=2, max_digits=12,
default=0)
# Not all lines are actually shipped (such as downloads), hence shipping
# address is not mandatory.
shipping_address = models.ForeignKey(
'order.ShippingAddress', null=True, blank=True,
verbose_name=_("Shipping Address"),
on_delete=models.SET_NULL)
shipping_method = models.CharField(
_("Shipping method"), max_length=128, blank=True)
# Identifies shipping code
shipping_code = models.CharField(blank=True, max_length=128, default="")
# Use this field to indicate that an order is on hold / awaiting payment
status = models.CharField(_("Status"), max_length=100, blank=True)
guest_email = models.EmailField(_("Guest email address"), blank=True)
# Index added to this field for reporting
date_placed = models.DateTimeField(db_index=True)
#: Order status pipeline. This should be a dict where each (key, value) #:
#: corresponds to a status and a list of possible statuses that can follow
#: that one.
pipeline = getattr(settings, 'OSCAR_ORDER_STATUS_PIPELINE', {})
#: Order status cascade pipeline. This should be a dict where each (key,
#: value) pair corresponds to an *order* status and the corresponding
#: *line* status that needs to be set when the order is set to the new
#: status
cascade = getattr(settings, 'OSCAR_ORDER_STATUS_CASCADE', {})
[docs] @classmethod
def all_statuses(cls):
"""
Return all possible statuses for an order
"""
return list(cls.pipeline.keys())
[docs] def available_statuses(self):
"""
Return all possible statuses that this order can move to
"""
return self.pipeline.get(self.status, ())
[docs] def set_status(self, new_status):
"""
Set a new status for this order.
If the requested status is not valid, then ``InvalidOrderStatus`` is
raised.
"""
if new_status == self.status:
return
old_status = self.status
if new_status not in self.available_statuses():
raise exceptions.InvalidOrderStatus(
_("'%(new_status)s' is not a valid status for order %(number)s"
" (current status: '%(status)s')")
% {'new_status': new_status,
'number': self.number,
'status': self.status})
self.status = new_status
if new_status in self.cascade:
for line in self.lines.all():
line.status = self.cascade[self.status]
line.save()
self.save()
# Send signal for handling status changed
order_status_changed.send(sender=self,
order=self,
old_status=old_status,
new_status=new_status,
)
self._create_order_status_change(old_status, new_status)
set_status.alters_data = True
def _create_order_status_change(self, old_status, new_status):
# Not setting the status on the order as that should be handled before
self.status_changes.create(old_status=old_status, new_status=new_status)
@property
def is_anonymous(self):
# It's possible for an order to be placed by a customer who then
# deletes their profile. Hence, we need to check that a guest email is
# set.
return self.user is None and bool(self.guest_email)
@property
def basket_total_before_discounts_incl_tax(self):
"""
Return basket total including tax but before discounts are applied
"""
result = self.lines.aggregate(total=Sum('line_price_before_discounts_incl_tax'))
return result['total']
@property
def basket_total_before_discounts_excl_tax(self):
"""
Return basket total excluding tax but before discounts are applied
"""
result = self.lines.aggregate(total=Sum('line_price_before_discounts_excl_tax'))
return result['total']
@property
def basket_total_incl_tax(self):
"""
Return basket total including tax
"""
return self.total_incl_tax - self.shipping_incl_tax - self.surcharge_incl_tax
@property
def basket_total_excl_tax(self):
"""
Return basket total excluding tax
"""
return self.total_excl_tax - self.shipping_excl_tax - self.surcharge_excl_tax
@property
def total_before_discounts_incl_tax(self):
return (self.basket_total_before_discounts_incl_tax
+ self.shipping_incl_tax)
@property
def total_before_discounts_excl_tax(self):
return (self.basket_total_before_discounts_excl_tax
+ self.shipping_excl_tax)
@property
def total_discount_incl_tax(self):
"""
The amount of discount this order received
"""
discount = D('0.00')
for line in self.lines.all():
discount += line.discount_incl_tax
return discount
@property
def total_discount_excl_tax(self):
discount = D('0.00')
for line in self.lines.all():
discount += line.discount_excl_tax
return discount
@property
def total_tax(self):
return self.total_incl_tax - self.total_excl_tax
@property
def surcharge_excl_tax(self):
return sum(charge.excl_tax for charge in self.surcharges.all())
@property
def surcharge_incl_tax(self):
return sum(charge.incl_tax for charge in self.surcharges.all())
@property
def num_lines(self):
return self.lines.count()
@property
def num_items(self):
"""
Returns the number of items in this order.
"""
num_items = 0
for line in self.lines.all():
num_items += line.quantity
return num_items
@property
def shipping_tax(self):
return self.shipping_incl_tax - self.shipping_excl_tax
@property
def shipping_status(self):
"""Return the last complete shipping event for this order."""
# As safeguard against identical timestamps, also sort by the primary
# key. It's not recommended to rely on this behaviour, but in practice
# reasonably safe if PKs are not manually set.
events = self.shipping_events.order_by('-date_created', '-pk').all()
if not len(events):
return ''
# Collect all events by event-type
event_map = OrderedDict()
for event in events:
event_name = event.event_type.name
if event_name not in event_map:
event_map[event_name] = []
event_map[event_name].extend(list(event.line_quantities.all()))
# Determine last complete event
status = _("In progress")
for event_name, event_line_quantities in event_map.items():
if self._is_event_complete(event_line_quantities):
return event_name
return status
@property
def has_shipping_discounts(self):
return len(self.shipping_discounts) > 0
@property
def shipping_before_discounts_incl_tax(self):
# We can construct what shipping would have been before discounts by
# adding the discounts back onto the final shipping charge.
total = D('0.00')
for discount in self.shipping_discounts:
total += discount.amount
return self.shipping_incl_tax + total
def _is_event_complete(self, event_quantities):
# Form map of line to quantity
event_map = {}
for event_quantity in event_quantities:
line_id = event_quantity.line_id
event_map.setdefault(line_id, 0)
event_map[line_id] += event_quantity.quantity
for line in self.lines.all():
if event_map.get(line.pk, 0) != line.quantity:
return False
return True
class Meta:
abstract = True
app_label = 'order'
ordering = ['-date_placed']
verbose_name = _("Order")
verbose_name_plural = _("Orders")
def __str__(self):
return "#%s" % (self.number,)
def verification_hash(self):
signer = Signer(salt='oscar.apps.order.Order')
return signer.sign(self.number)
[docs] def check_verification_hash(self, hash_to_check):
"""
Checks the received verification hash against this order number.
Returns False if the verification failed, True otherwise.
"""
signer = Signer(salt='oscar.apps.order.Order')
try:
signed_number = signer.unsign(hash_to_check)
except BadSignature:
return False
return constant_time_compare(signed_number, self.number)
@property
def email(self):
if not self.user:
return self.guest_email
return self.user.email
@property
def basket_discounts(self):
# This includes both offer- and voucher- discounts. For orders we
# don't need to treat them differently like we do for baskets.
return self.discounts.filter(
category=AbstractOrderDiscount.BASKET)
@property
def shipping_discounts(self):
return self.discounts.filter(
category=AbstractOrderDiscount.SHIPPING)
@property
def post_order_actions(self):
return self.discounts.filter(
category=AbstractOrderDiscount.DEFERRED)
def set_date_placed_default(self):
if self.date_placed is None:
self.date_placed = now()
[docs] def save(self, *args, **kwargs):
# Ensure the date_placed field works as it auto_now_add was set. But
# this gives us the ability to set the date_placed explicitly (which is
# useful when importing orders from another system).
self.set_date_placed_default()
super().save(*args, **kwargs)
[docs]class AbstractOrderNote(models.Model):
"""
A note against an order.
This are often used for audit purposes too. IE, whenever an admin
makes a change to an order, we create a note to record what happened.
"""
order = models.ForeignKey(
'order.Order',
on_delete=models.CASCADE,
related_name="notes",
verbose_name=_("Order"))
# These are sometimes programatically generated so don't need a
# user everytime
user = models.ForeignKey(
AUTH_USER_MODEL,
on_delete=models.CASCADE,
null=True,
verbose_name=_("User"))
# We allow notes to be classified although this isn't always needed
INFO, WARNING, ERROR, SYSTEM = 'Info', 'Warning', 'Error', 'System'
note_type = models.CharField(_("Note Type"), max_length=128, blank=True)
message = models.TextField(_("Message"))
date_created = models.DateTimeField(_("Date Created"), auto_now_add=True)
date_updated = models.DateTimeField(_("Date Updated"), auto_now=True)
# Notes can only be edited for 5 minutes after being created
editable_lifetime = 300
class Meta:
abstract = True
app_label = 'order'
ordering = ['-date_updated']
verbose_name = _("Order Note")
verbose_name_plural = _("Order Notes")
def __str__(self):
return "'%s' (%s)" % (self.message[0:50], self.user)
def is_editable(self):
if self.note_type == self.SYSTEM:
return False
delta = timezone.now() - self.date_updated
return delta.seconds < self.editable_lifetime
class AbstractOrderStatusChange(models.Model):
order = models.ForeignKey(
'order.Order',
on_delete=models.CASCADE,
related_name='status_changes',
verbose_name=_('Order Status Changes')
)
old_status = models.CharField(_('Old Status'), max_length=100, blank=True)
new_status = models.CharField(_('New Status'), max_length=100, blank=True)
date_created = models.DateTimeField(_('Date Created'), auto_now_add=True, db_index=True)
class Meta:
abstract = True
app_label = 'order'
verbose_name = _('Order Status Change')
verbose_name_plural = _('Order Status Changes')
ordering = ['-date_created']
def __str__(self):
return _("%(order)s has changed status from %(old_status)s to %(new_status)s") \
% {'order': self.order, 'old_status': self.old_status, 'new_status': self.new_status, }
[docs]class AbstractCommunicationEvent(models.Model):
"""
An order-level event involving a communication to the customer, such
as an confirmation email being sent.
"""
order = models.ForeignKey(
'order.Order',
on_delete=models.CASCADE,
related_name="communication_events",
verbose_name=_("Order"))
event_type = models.ForeignKey(
'communication.CommunicationEventType',
on_delete=models.CASCADE,
verbose_name=_("Event Type"))
date_created = models.DateTimeField(_("Date"), auto_now_add=True, db_index=True)
class Meta:
abstract = True
app_label = 'order'
verbose_name = _("Communication Event")
verbose_name_plural = _("Communication Events")
ordering = ['-date_created']
def __str__(self):
return _("'%(type)s' event for order #%(number)s") \
% {'type': self.event_type.name, 'number': self.order.number}
# LINES
[docs]class AbstractLine(models.Model):
"""
An order line
"""
order = models.ForeignKey(
'order.Order',
on_delete=models.CASCADE,
related_name='lines',
verbose_name=_("Order"))
# PARTNER INFORMATION
# -------------------
# We store the partner and various detail their SKU and the title for cases
# where the product has been deleted from the catalogue (but we still need
# the data for reporting). We also store the partner name in case the
# partner gets deleted at a later date.
partner = models.ForeignKey(
'partner.Partner', related_name='order_lines', blank=True, null=True,
on_delete=models.SET_NULL, verbose_name=_("Partner"))
partner_name = models.CharField(
_("Partner name"), max_length=128, blank=True)
partner_sku = models.CharField(_("Partner SKU"), max_length=128)
# A line reference is the ID that a partner uses to represent this
# particular line (it's not the same as a SKU).
partner_line_reference = models.CharField(
_("Partner reference"), max_length=128, blank=True,
help_text=_("This is the item number that the partner uses "
"within their system"))
partner_line_notes = models.TextField(
_("Partner Notes"), blank=True)
# We keep a link to the stockrecord used for this line which allows us to
# update stocklevels when it ships
stockrecord = models.ForeignKey(
'partner.StockRecord', on_delete=models.SET_NULL, blank=True,
null=True, verbose_name=_("Stock record"))
# PRODUCT INFORMATION
# -------------------
# We don't want any hard links between orders and the products table so we
# allow this link to be NULLable.
product = models.ForeignKey(
'catalogue.Product', on_delete=models.SET_NULL, blank=True, null=True,
verbose_name=_("Product"))
title = models.CharField(
pgettext_lazy("Product title", "Title"), max_length=255)
# UPC can be null because it's usually set as the product's UPC, and that
# can be null as well
upc = models.CharField(_("UPC"), max_length=128, blank=True, null=True)
quantity = models.PositiveIntegerField(_("Quantity"), default=1)
# REPORTING INFORMATION
# ---------------------
# Price information (these fields are actually redundant as the information
# can be calculated from the LinePrice models
line_price_incl_tax = models.DecimalField(
_("Price (inc. tax)"), decimal_places=2, max_digits=12)
line_price_excl_tax = models.DecimalField(
_("Price (excl. tax)"), decimal_places=2, max_digits=12)
# Price information before discounts are applied
line_price_before_discounts_incl_tax = models.DecimalField(
_("Price before discounts (inc. tax)"),
decimal_places=2, max_digits=12)
line_price_before_discounts_excl_tax = models.DecimalField(
_("Price before discounts (excl. tax)"),
decimal_places=2, max_digits=12)
# Normal site price for item (without discounts)
unit_price_incl_tax = models.DecimalField(
_("Unit Price (inc. tax)"), decimal_places=2, max_digits=12,
blank=True, null=True)
unit_price_excl_tax = models.DecimalField(
_("Unit Price (excl. tax)"), decimal_places=2, max_digits=12,
blank=True, null=True)
# Partners often want to assign some status to each line to help with their
# own business processes.
status = models.CharField(_("Status"), max_length=255, blank=True)
#: Order status pipeline. This should be a dict where each (key, value)
#: corresponds to a status and the possible statuses that can follow that
#: one.
pipeline = getattr(settings, 'OSCAR_LINE_STATUS_PIPELINE', {})
class Meta:
abstract = True
app_label = 'order'
# Enforce sorting in order of creation.
ordering = ['pk']
verbose_name = _("Order Line")
verbose_name_plural = _("Order Lines")
def __str__(self):
if self.product:
title = self.product.title
else:
title = _('<missing product>')
return _("Product '%(name)s', quantity '%(qty)s'") % {
'name': title, 'qty': self.quantity}
[docs] @classmethod
def all_statuses(cls):
"""
Return all possible statuses for an order line
"""
return list(cls.pipeline.keys())
[docs] def available_statuses(self):
"""
Return all possible statuses that this order line can move to
"""
return self.pipeline.get(self.status, ())
[docs] def set_status(self, new_status):
"""
Set a new status for this line
If the requested status is not valid, then ``InvalidLineStatus`` is
raised.
"""
if new_status == self.status:
return
old_status = self.status
if new_status not in self.available_statuses():
raise exceptions.InvalidLineStatus(
_("'%(new_status)s' is not a valid status (current status:"
" '%(status)s')")
% {'new_status': new_status, 'status': self.status})
self.status = new_status
self.save()
# Send signal for handling status changed
order_line_status_changed.send(sender=self,
line=self,
old_status=old_status,
new_status=new_status,
)
set_status.alters_data = True
@property
def description(self):
"""
Returns a description of this line including details of any
line attributes.
"""
desc = self.title
ops = []
for attribute in self.attributes.all():
ops.append("%s = '%s'" % (attribute.type, attribute.value))
if ops:
desc = "%s (%s)" % (desc, ", ".join(ops))
return desc
@property
def discount_incl_tax(self):
return self.line_price_before_discounts_incl_tax \
- self.line_price_incl_tax
@property
def discount_excl_tax(self):
return self.line_price_before_discounts_excl_tax \
- self.line_price_excl_tax
@property
def line_price_tax(self):
return self.line_price_incl_tax - self.line_price_excl_tax
@property
def unit_price_tax(self):
return self.unit_price_incl_tax - self.unit_price_excl_tax
# Shipping status helpers
@property
def shipping_status(self):
"""
Returns a string summary of the shipping status of this line
"""
status_map = self.shipping_event_breakdown
if not status_map:
return ''
events = []
last_complete_event_name = None
for event_dict in reversed(list(status_map.values())):
if event_dict['quantity'] == self.quantity:
events.append(event_dict['name'])
last_complete_event_name = event_dict['name']
else:
events.append("%s (%d/%d items)" % (
event_dict['name'], event_dict['quantity'],
self.quantity))
if last_complete_event_name == list(status_map.values())[0]['name']:
return last_complete_event_name
return ', '.join(events)
[docs] def is_shipping_event_permitted(self, event_type, quantity):
"""
Test whether a shipping event with the given quantity is permitted
This method should normally be overridden to ensure that the
prerequisite shipping events have been passed for this line.
"""
# Note, this calculation is simplistic - normally, you will also need
# to check if previous shipping events have occurred. Eg, you can't
# return lines until they have been shipped.
current_qty = self.shipping_event_quantity(event_type)
return (current_qty + quantity) <= self.quantity
[docs] def shipping_event_quantity(self, event_type):
"""
Return the quantity of this line that has been involved in a shipping
event of the passed type.
"""
result = self.shipping_event_quantities.filter(
event__event_type=event_type).aggregate(Sum('quantity'))
if result['quantity__sum'] is None:
return 0
else:
return result['quantity__sum']
[docs] def has_shipping_event_occurred(self, event_type, quantity=None):
"""
Test whether this line has passed a given shipping event
"""
if not quantity:
quantity = self.quantity
return self.shipping_event_quantity(event_type) == quantity
[docs] def get_event_quantity(self, event):
"""
Fetches the ShippingEventQuantity instance for this line
Exists as a separate method so it can be overridden to avoid
the DB query that's caused by get().
"""
return event.line_quantities.get(line=self)
@property
def shipping_event_breakdown(self):
"""
Returns a dict of shipping events that this line has been through
"""
status_map = OrderedDict()
for event in self.shipping_events.all():
event_type = event.event_type
event_name = event_type.name
event_quantity = self.get_event_quantity(event).quantity
if event_name in status_map:
status_map[event_name]['quantity'] += event_quantity
else:
status_map[event_name] = {
'event_type': event_type,
'name': event_name,
'quantity': event_quantity
}
return status_map
# Payment event helpers
[docs] def is_payment_event_permitted(self, event_type, quantity):
"""
Test whether a payment event with the given quantity is permitted.
Allow each payment event type to occur only once per quantity.
"""
current_qty = self.payment_event_quantity(event_type)
return (current_qty + quantity) <= self.quantity
[docs] def payment_event_quantity(self, event_type):
"""
Return the quantity of this line that has been involved in a payment
event of the passed type.
"""
result = self.payment_event_quantities.filter(
event__event_type=event_type).aggregate(Sum('quantity'))
if result['quantity__sum'] is None:
return 0
else:
return result['quantity__sum']
@property
def is_product_deleted(self):
return self.product is None
[docs] def is_available_to_reorder(self, basket, strategy):
"""
Test if this line can be re-ordered using the passed strategy and
basket
"""
if not self.product:
return False, (_("'%(title)s' is no longer available") %
{'title': self.title})
try:
basket_line = basket.lines.get(product=self.product)
except basket.lines.model.DoesNotExist:
desired_qty = self.quantity
else:
desired_qty = basket_line.quantity + self.quantity
result = strategy.fetch_for_product(self.product)
is_available, reason = result.availability.is_purchase_permitted(
quantity=desired_qty)
if not is_available:
return False, reason
return True, None
[docs]class AbstractLineAttribute(models.Model):
"""
An attribute of a line
"""
line = models.ForeignKey(
'order.Line',
on_delete=models.CASCADE,
related_name='attributes',
verbose_name=_("Line"))
option = models.ForeignKey(
'catalogue.Option', null=True, on_delete=models.SET_NULL,
related_name="line_attributes", verbose_name=_("Option"))
type = models.CharField(_("Type"), max_length=128)
value = models.CharField(_("Value"), max_length=255)
class Meta:
abstract = True
app_label = 'order'
verbose_name = _("Line Attribute")
verbose_name_plural = _("Line Attributes")
def __str__(self):
return "%s = %s" % (self.type, self.value)
[docs]class AbstractLinePrice(models.Model):
"""
For tracking the prices paid for each unit within a line.
This is necessary as offers can lead to units within a line
having different prices. For example, one product may be sold at
50% off as it's part of an offer while the remainder are full price.
"""
order = models.ForeignKey(
'order.Order',
on_delete=models.CASCADE,
related_name='line_prices',
verbose_name=_("Option"))
line = models.ForeignKey(
'order.Line',
on_delete=models.CASCADE,
related_name='prices',
verbose_name=_("Line"))
quantity = models.PositiveIntegerField(_("Quantity"), default=1)
price_incl_tax = models.DecimalField(
_("Price (inc. tax)"), decimal_places=2, max_digits=12)
price_excl_tax = models.DecimalField(
_("Price (excl. tax)"), decimal_places=2, max_digits=12)
shipping_incl_tax = models.DecimalField(
_("Shiping (inc. tax)"), decimal_places=2, max_digits=12, default=0)
shipping_excl_tax = models.DecimalField(
_("Shipping (excl. tax)"), decimal_places=2, max_digits=12, default=0)
class Meta:
abstract = True
app_label = 'order'
ordering = ('id',)
verbose_name = _("Line Price")
verbose_name_plural = _("Line Prices")
def __str__(self):
return _("Line '%(number)s' (quantity %(qty)d) price %(price)s") % {
'number': self.line,
'qty': self.quantity,
'price': self.price_incl_tax}
# PAYMENT EVENTS
[docs]class AbstractPaymentEventType(models.Model):
"""
Payment event types are things like 'Paid', 'Failed', 'Refunded'.
These are effectively the transaction types.
"""
name = models.CharField(_("Name"), max_length=128, unique=True)
code = AutoSlugField(_("Code"), max_length=128, unique=True,
populate_from='name')
class Meta:
abstract = True
app_label = 'order'
verbose_name = _("Payment Event Type")
verbose_name_plural = _("Payment Event Types")
ordering = ('name', )
def __str__(self):
return self.name
[docs]class AbstractPaymentEvent(models.Model):
"""
A payment event for an order
For example:
* All lines have been paid for
* 2 lines have been refunded
"""
order = models.ForeignKey(
'order.Order',
on_delete=models.CASCADE,
related_name='payment_events',
verbose_name=_("Order"))
amount = models.DecimalField(
_("Amount"), decimal_places=2, max_digits=12)
# The reference should refer to the transaction ID of the payment gateway
# that was used for this event.
reference = models.CharField(
_("Reference"), max_length=128, blank=True)
lines = models.ManyToManyField(
'order.Line', through='PaymentEventQuantity',
verbose_name=_("Lines"))
event_type = models.ForeignKey(
'order.PaymentEventType',
on_delete=models.CASCADE,
verbose_name=_("Event Type"))
# Allow payment events to be linked to shipping events. Often a shipping
# event will trigger a payment event and so we can use this FK to capture
# the relationship.
shipping_event = models.ForeignKey(
'order.ShippingEvent',
null=True,
on_delete=models.CASCADE,
related_name='payment_events')
date_created = models.DateTimeField(_("Date created"), auto_now_add=True, db_index=True)
class Meta:
abstract = True
app_label = 'order'
verbose_name = _("Payment Event")
verbose_name_plural = _("Payment Events")
ordering = ['-date_created']
def __str__(self):
return _("Payment event for order %s") % self.order
def num_affected_lines(self):
return self.lines.all().count()
[docs]class PaymentEventQuantity(models.Model):
"""
A "through" model linking lines to payment events
"""
event = models.ForeignKey(
'order.PaymentEvent',
on_delete=models.CASCADE,
related_name='line_quantities',
verbose_name=_("Event"))
line = models.ForeignKey(
'order.Line',
on_delete=models.CASCADE,
related_name="payment_event_quantities",
verbose_name=_("Line"))
quantity = models.PositiveIntegerField(_("Quantity"))
class Meta:
app_label = 'order'
verbose_name = _("Payment Event Quantity")
verbose_name_plural = _("Payment Event Quantities")
unique_together = ('event', 'line')
# SHIPPING EVENTS
[docs]class AbstractShippingEvent(models.Model):
"""
An event is something which happens to a group of lines such as
1 item being dispatched.
"""
order = models.ForeignKey(
'order.Order',
on_delete=models.CASCADE,
related_name='shipping_events',
verbose_name=_("Order"))
lines = models.ManyToManyField(
'order.Line', related_name='shipping_events',
through='ShippingEventQuantity', verbose_name=_("Lines"))
event_type = models.ForeignKey(
'order.ShippingEventType',
on_delete=models.CASCADE,
verbose_name=_("Event Type"))
notes = models.TextField(
_("Event notes"), blank=True,
help_text=_("This could be the dispatch reference, or a "
"tracking number"))
date_created = models.DateTimeField(_("Date Created"), auto_now_add=True, db_index=True)
class Meta:
abstract = True
app_label = 'order'
verbose_name = _("Shipping Event")
verbose_name_plural = _("Shipping Events")
ordering = ['-date_created']
def __str__(self):
return _("Order #%(number)s, type %(type)s") % {
'number': self.order.number,
'type': self.event_type}
def num_affected_lines(self):
return self.lines.count()
[docs]class ShippingEventQuantity(models.Model):
"""
A "through" model linking lines to shipping events.
This exists to track the quantity of a line that is involved in a
particular shipping event.
"""
event = models.ForeignKey(
'order.ShippingEvent',
on_delete=models.CASCADE,
related_name='line_quantities',
verbose_name=_("Event"))
line = models.ForeignKey(
'order.Line',
on_delete=models.CASCADE,
related_name="shipping_event_quantities",
verbose_name=_("Line"))
quantity = models.PositiveIntegerField(_("Quantity"))
class Meta:
app_label = 'order'
verbose_name = _("Shipping Event Quantity")
verbose_name_plural = _("Shipping Event Quantities")
unique_together = ('event', 'line')
[docs] def save(self, *args, **kwargs):
# Default quantity to full quantity of line
if not self.quantity:
self.quantity = self.line.quantity
# Ensure we don't violate quantities constraint
if not self.line.is_shipping_event_permitted(
self.event.event_type, self.quantity):
raise exceptions.InvalidShippingEvent
super().save(*args, **kwargs)
def __str__(self):
return _("%(product)s - quantity %(qty)d") % {
'product': self.line.product,
'qty': self.quantity}
[docs]class AbstractShippingEventType(models.Model):
"""
A type of shipping/fulfilment event
E.g.: 'Shipped', 'Cancelled', 'Returned'
"""
# Name is the friendly description of an event
name = models.CharField(_("Name"), max_length=255, unique=True)
# Code is used in forms
code = AutoSlugField(_("Code"), max_length=128, unique=True,
populate_from='name')
class Meta:
abstract = True
app_label = 'order'
verbose_name = _("Shipping Event Type")
verbose_name_plural = _("Shipping Event Types")
ordering = ('name', )
def __str__(self):
return self.name
# DISCOUNTS
[docs]class AbstractOrderDiscount(models.Model):
"""
A discount against an order.
Normally only used for display purposes so an order can be listed with
discounts displayed separately even though in reality, the discounts are
applied at the line level.
This has evolved to be a slightly misleading class name as this really
track benefit applications which aren't necessarily discounts.
"""
order = models.ForeignKey(
'order.Order',
on_delete=models.CASCADE,
related_name="discounts",
verbose_name=_("Order"))
# We need to distinguish between basket discounts, shipping discounts and
# 'deferred' discounts.
BASKET, SHIPPING, DEFERRED = "Basket", "Shipping", "Deferred"
CATEGORY_CHOICES = (
(BASKET, _(BASKET)),
(SHIPPING, _(SHIPPING)),
(DEFERRED, _(DEFERRED)),
)
category = models.CharField(
_("Discount category"), default=BASKET, max_length=64,
choices=CATEGORY_CHOICES)
offer_id = models.PositiveIntegerField(
_("Offer ID"), blank=True, null=True)
offer_name = models.CharField(
_("Offer name"), max_length=128, db_index=True, blank=True)
voucher_id = models.PositiveIntegerField(
_("Voucher ID"), blank=True, null=True)
voucher_code = models.CharField(
_("Code"), max_length=128, db_index=True, blank=True)
frequency = models.PositiveIntegerField(_("Frequency"), null=True)
amount = models.DecimalField(
_("Amount"), decimal_places=2, max_digits=12, default=0)
# Post-order offer applications can return a message to indicate what
# action was taken after the order was placed.
message = models.TextField(blank=True)
@property
def is_basket_discount(self):
return self.category == self.BASKET
@property
def is_shipping_discount(self):
return self.category == self.SHIPPING
@property
def is_post_order_action(self):
return self.category == self.DEFERRED
class Meta:
abstract = True
app_label = 'order'
ordering = ['pk']
verbose_name = _("Order Discount")
verbose_name_plural = _("Order Discounts")
[docs] def save(self, **kwargs):
if self.offer_id and not self.offer_name:
offer = self.offer
if offer:
self.offer_name = offer.name
if self.voucher_id and not self.voucher_code:
voucher = self.voucher
if voucher:
self.voucher_code = voucher.code
super().save(**kwargs)
def __str__(self):
return _("Discount of %(amount)r from order %(order)s") % {
'amount': self.amount, 'order': self.order}
@property
def offer(self):
Offer = get_model('offer', 'ConditionalOffer')
try:
return Offer.objects.get(id=self.offer_id)
except Offer.DoesNotExist:
return None
@property
def voucher(self):
Voucher = get_model('voucher', 'Voucher')
try:
return Voucher.objects.get(id=self.voucher_id)
except Voucher.DoesNotExist:
return None
def description(self):
if self.voucher_code:
return self.voucher_code
return self.offer_name or ""
class AbstractSurcharge(models.Model):
order = models.ForeignKey(
'order.Order',
on_delete=models.CASCADE,
related_name="surcharges",
verbose_name=_("Surcharges"))
name = models.CharField(
_("Surcharge"), max_length=128
)
code = models.CharField(
_("Surcharge code"), max_length=128
)
incl_tax = models.DecimalField(
_("Surcharge (inc. tax)"), decimal_places=2, max_digits=12,
default=0)
excl_tax = models.DecimalField(
_("Surcharge (excl. tax)"), decimal_places=2, max_digits=12,
default=0)
@property
def tax(self):
return self.incl_tax - self.excl_tax
class Meta:
abstract = True
app_label = 'order'
ordering = ['pk']