|
|
|
@ -3,6 +3,7 @@ from django.db import models
@@ -3,6 +3,7 @@ from django.db import models
|
|
|
|
|
from django.contrib.auth.models import User |
|
|
|
|
import datetime |
|
|
|
|
from decimal import Decimal |
|
|
|
|
from django.db.models.signals import pre_delete, post_delete |
|
|
|
|
|
|
|
|
|
# Create your models here. |
|
|
|
|
class BuyableType(models.Model): |
|
|
|
@ -31,18 +32,6 @@ class Buyable(models.Model):
@@ -31,18 +32,6 @@ class Buyable(models.Model):
|
|
|
|
|
""" Returns True if the item has deposit. """ |
|
|
|
|
return self.deposit > Decimal(0) |
|
|
|
|
|
|
|
|
|
def createPurchase(self, isDeposit=False): |
|
|
|
|
""" Creates a :class:`Purchase` containing this :class:`Buyable`. """ |
|
|
|
|
p = Purchase() |
|
|
|
|
if isDeposit: |
|
|
|
|
p.price = self.deposit |
|
|
|
|
else: |
|
|
|
|
p.price = self.price |
|
|
|
|
p.dateTime = datetime.datetime.now() |
|
|
|
|
p.buyable = self |
|
|
|
|
|
|
|
|
|
return p |
|
|
|
|
|
|
|
|
|
def __unicode__(self): |
|
|
|
|
item = "%s (%.2f EUR" % (self.name, self.price) |
|
|
|
|
if self.hasDeposit(): |
|
|
|
@ -59,30 +48,19 @@ class Buyable(models.Model):
@@ -59,30 +48,19 @@ class Buyable(models.Model):
|
|
|
|
|
class Order(models.Model): |
|
|
|
|
""" Represents an order by the user, consists of several :class:`Purchases <Purchase>`. """ |
|
|
|
|
user = models.ForeignKey(User) |
|
|
|
|
price = models.DecimalField(max_digits=8, decimal_places=2) |
|
|
|
|
dateTime = models.DateTimeField() |
|
|
|
|
|
|
|
|
|
def create(self, user=None): |
|
|
|
|
models.Model.__init__(self) |
|
|
|
|
self.price = Decimal(0) |
|
|
|
|
self.dateTime = datetime.datetime.now() |
|
|
|
|
if user: |
|
|
|
|
self.user = user |
|
|
|
|
price = models.DecimalField(max_digits=8, decimal_places=2, default=Decimal(0)) |
|
|
|
|
dateTime = models.DateTimeField(auto_now_add=True) |
|
|
|
|
|
|
|
|
|
def addItems(self, items): |
|
|
|
|
for item in items: |
|
|
|
|
self.purchase.add(item) |
|
|
|
|
self.price += item.price |
|
|
|
|
|
|
|
|
|
def updatePrice(self, commit=False, updateBalance=False): |
|
|
|
|
self.price = Decimal(0) |
|
|
|
|
for item in self.purchase_set.all(): |
|
|
|
|
self.price += item.price |
|
|
|
|
if commit or updateBalance: |
|
|
|
|
# TROLL MODE ON! |
|
|
|
|
profile = self.user.get_profile() |
|
|
|
|
profile.balance -= self.price |
|
|
|
|
profile.save() |
|
|
|
|
def updatePrice(self, instance, price, saveOrder=True): |
|
|
|
|
""" Update price of order, save Order if saveOrder is True. """ |
|
|
|
|
self.price += price |
|
|
|
|
if self.id and saveOrder: |
|
|
|
|
self.save() |
|
|
|
|
|
|
|
|
|
def __unicode__(self): |
|
|
|
|
return "Price %s, User %s" % (self.price, self.user) |
|
|
|
@ -92,40 +70,86 @@ class Order(models.Model):
@@ -92,40 +70,86 @@ class Order(models.Model):
|
|
|
|
|
for item in self.purchase_set.all(): |
|
|
|
|
l += item.buyable.name + u", " |
|
|
|
|
return l.rstrip(u", ") |
|
|
|
|
|
|
|
|
|
def save(self, *args, **kwargs): |
|
|
|
|
profile = self.user.get_profile() |
|
|
|
|
if self.id == None: |
|
|
|
|
# new item, get it! |
|
|
|
|
profile.balance -= self.price |
|
|
|
|
profile.save() |
|
|
|
|
else: |
|
|
|
|
# get old |
|
|
|
|
oldobj = Order.objects.get(id=self.id) |
|
|
|
|
if oldobj.user == self.user: |
|
|
|
|
profile = self.user.get_profile() |
|
|
|
|
profile.balance -= (self.price - oldobj.price) |
|
|
|
|
profile.save() |
|
|
|
|
else: |
|
|
|
|
oldProfile = oldobj.user.get_profile() |
|
|
|
|
newProfile = self.user.get_profile() |
|
|
|
|
oldProfile.balance += oldobj.price |
|
|
|
|
oldProfile.save() |
|
|
|
|
newprofile.balance -= self.price |
|
|
|
|
self.save() |
|
|
|
|
super(Order, self).save(*args, **kwargs) |
|
|
|
|
|
|
|
|
|
@staticmethod |
|
|
|
|
def pre_delete_signal(sender, instance, **kwargs): |
|
|
|
|
# Normally we would not need this function as for consistency |
|
|
|
|
# reasons and the behaviour of Buyables delete at deletion |
|
|
|
|
# the orders price should be 0. But for the sake of not getting |
|
|
|
|
# inconsistent with the users balance in the strangest situations |
|
|
|
|
# we will check that here. This means (at least) one extra query |
|
|
|
|
# per order deletion, as we don't get the updated object but the one |
|
|
|
|
# where all the buyables have not updated the price |
|
|
|
|
updOrder = Order.objects.get(id=instance.id) |
|
|
|
|
if updOrder.price != Decimal("0"): |
|
|
|
|
profile = updOrder.user.get_profile() |
|
|
|
|
profile.balance += updOrder.price |
|
|
|
|
profile.save() |
|
|
|
|
|
|
|
|
|
#def save(self, *args, **kwargs): |
|
|
|
|
#profile = self.user.get_profile() |
|
|
|
|
#if self.id == None: |
|
|
|
|
## new item, get it! |
|
|
|
|
#profile.balance -= self.price |
|
|
|
|
#profile.save() |
|
|
|
|
#else: |
|
|
|
|
## get old |
|
|
|
|
#super(Order, self).save(*args, **kwargs) |
|
|
|
|
|
|
|
|
|
pre_delete.connect(Order.pre_delete_signal, sender=Order) |
|
|
|
|
|
|
|
|
|
class Purchase(models.Model): |
|
|
|
|
""" Represents a :class:`Buyable` in a :class:`Order`""" |
|
|
|
|
""" Represents a :class:`Buyable` in a :class:`Order`. |
|
|
|
|
|
|
|
|
|
Could be viewed as 'one bought item' or 'one item on the voucher'.""" |
|
|
|
|
order = models.ForeignKey(Order) |
|
|
|
|
price = models.DecimalField(max_digits=8, decimal_places=2) |
|
|
|
|
price = models.DecimalField(max_digits=8, decimal_places=2, default=None) |
|
|
|
|
isDeposit = models.BooleanField() |
|
|
|
|
buyable = models.ForeignKey(Buyable) |
|
|
|
|
|
|
|
|
|
@staticmethod |
|
|
|
|
def create(order, buyable, isDeposit=False): |
|
|
|
|
""" Create a Purchase from a buyable. Still needs to be saved. """ |
|
|
|
|
p = Purchase() |
|
|
|
|
p.order = order |
|
|
|
|
p.isDeposit = isDeposit |
|
|
|
|
if p.isDeposit: |
|
|
|
|
p.price = buyable.deposit |
|
|
|
|
|
|
|
|
|
def updatePrice(self): |
|
|
|
|
if self.isDeposit: |
|
|
|
|
self.price = self.buyable.deposit |
|
|
|
|
else: |
|
|
|
|
p.price = buyable.price |
|
|
|
|
p.dateTime = datetime.datetime.now() |
|
|
|
|
p.buyable = buyable |
|
|
|
|
|
|
|
|
|
return p |
|
|
|
|
|
|
|
|
|
self.price = self.buyable.price |
|
|
|
|
|
|
|
|
|
def __unicode__(self): |
|
|
|
|
return "%s%s" % (self.buyable.name, (self.isDeposit and " (deposit)" or "")) |
|
|
|
|
|
|
|
|
|
def save(self, saveOrder=True, *args, **kwargs): |
|
|
|
|
if self.price == None: |
|
|
|
|
self.updatePrice() |
|
|
|
|
|
|
|
|
|
if self.order: |
|
|
|
|
if self.id == None: |
|
|
|
|
self.order.updatePrice(self, self.price, saveOrder=saveOrder) |
|
|
|
|
else: |
|
|
|
|
# object exists, update. |
|
|
|
|
oldobj = Purchase.objects.get(id=self.id) |
|
|
|
|
if oldobj.order == self.order: |
|
|
|
|
self.order.updatePrice(self, self.price - oldobj.price, saveOrder=saveOrder) |
|
|
|
|
else: |
|
|
|
|
oldobj.order.updatePrice(oldobj, -oldobj.price, saveOrder=saveOrder) |
|
|
|
|
self.order.updatePrice(self, self.price, saveOrder=saveOrder) |
|
|
|
|
super(Purchase, self).save(*args, **kwargs) |
|
|
|
|
|
|
|
|
|
@staticmethod |
|
|
|
|
def pre_delete_signal(sender, instance, **kwargs): |
|
|
|
|
# FIXME: For many items this could get very inefficient |
|
|
|
|
# Find workarround! |
|
|
|
|
print "pre delete from purchase" |
|
|
|
|
instance.order.updatePrice(instance, -instance.price, saveOrder=True) |
|
|
|
|
|
|
|
|
|
pre_delete.connect(Purchase.pre_delete_signal, sender=Purchase) |
|
|
|
|