From c6a435eba01e196ab72bf86f5e63b43f8954b586 Mon Sep 17 00:00:00 2001
From: Oliver Walters <oliver.henry.walters@gmail.com>
Date: Tue, 3 Sep 2019 09:34:32 +1000
Subject: [PATCH] Add currency field to SupplierPriceBreak

---
 .../0006_supplierpricebreak_currency.py       | 20 +++++++++++++++++++
 InvenTree/company/models.py                   | 16 ++++++++++++++-
 2 files changed, 35 insertions(+), 1 deletion(-)
 create mode 100644 InvenTree/company/migrations/0006_supplierpricebreak_currency.py

diff --git a/InvenTree/company/migrations/0006_supplierpricebreak_currency.py b/InvenTree/company/migrations/0006_supplierpricebreak_currency.py
new file mode 100644
index 0000000000..94f533cf66
--- /dev/null
+++ b/InvenTree/company/migrations/0006_supplierpricebreak_currency.py
@@ -0,0 +1,20 @@
+# Generated by Django 2.2.4 on 2019-09-02 23:34
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('common', '0003_auto_20190902_2310'),
+        ('company', '0005_auto_20190525_2356'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='supplierpricebreak',
+            name='currency',
+            field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, to='common.Currency'),
+        ),
+    ]
diff --git a/InvenTree/company/models.py b/InvenTree/company/models.py
index 9d09557f17..eb8d7e5215 100644
--- a/InvenTree/company/models.py
+++ b/InvenTree/company/models.py
@@ -19,6 +19,7 @@ from django.conf import settings
 from django.contrib.staticfiles.templatetags.staticfiles import static
 
 from InvenTree.status_codes import OrderStatus
+from common.models import Currency
 
 
 def rename_company_image(instance, filename):
@@ -310,7 +311,7 @@ class SupplierPart(models.Model):
             # If this price-break quantity is the largest so far, use it!
             if pb.quantity > pb_quantity:
                 pb_quantity = pb.quantity
-                pb_cost = pb.cost
+                pb_cost = pb.get_cost()
 
         if pb_found:
             cost = pb_cost * quantity
@@ -369,6 +370,7 @@ class SupplierPriceBreak(models.Model):
         part: Link to a SupplierPart object that this price break applies to
         quantity: Quantity required for price break
         cost: Cost at specified quantity
+        currency: Reference to the currency of this pricebreak (leave empty for base currency)
     """
 
     part = models.ForeignKey(SupplierPart, on_delete=models.CASCADE, related_name='pricebreaks')
@@ -377,6 +379,18 @@ class SupplierPriceBreak(models.Model):
 
     cost = models.DecimalField(max_digits=10, decimal_places=5, validators=[MinValueValidator(0)])
 
+    currency = models.ForeignKey(Currency, blank=True, null=True, on_delete=models.SET_NULL)
+
+    def get_cost(self):
+        """ Return the cost of this price break, converted to the base currency """
+
+        scaler = 1.0
+
+        if self.currency:
+            scaler = self.currency.value
+
+        return self.cost * scaler
+
     class Meta:
         unique_together = ("part", "quantity")