mirror of
https://github.com/inventree/InvenTree
synced 2024-08-30 18:33:04 +00:00
c62ba5eb12
- DRF does not by deault run validate_unique on the model - Need to check if we are "creating" or "updating" a model - Catch and re-throw errors in the correct format - Unit tests
79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
# Generated by Django 3.2.4 on 2021-06-21 07:24
|
|
|
|
from django.db import migrations
|
|
|
|
|
|
def extract_purchase_price(apps, schema_editor):
|
|
"""
|
|
Find instances of StockItem which do *not* have a purchase price set,
|
|
but which point to a PurchaseOrder where there *is* a purchase price set.
|
|
|
|
Then, assign *that* purchase price to original StockItem.
|
|
|
|
This is to address an issue where older versions of InvenTree
|
|
did not correctly copy purchase price information cross to the StockItem objects.
|
|
|
|
Current InvenTree version (as of 2021-06-21) copy this information across correctly,
|
|
so this one-time data migration should suffice.
|
|
"""
|
|
|
|
# Required database models
|
|
StockItem = apps.get_model('stock', 'stockitem')
|
|
PurchaseOrder = apps.get_model('order', 'purchaseorder')
|
|
PurchaseOrderLineItem = apps.get_model('order', 'purchaseorderlineitem')
|
|
Part = apps.get_model('part', 'part')
|
|
|
|
# Find all the StockItem objects without a purchase_price which point to a PurchaseOrder
|
|
items = StockItem.objects.filter(purchase_price=None).exclude(purchase_order=None)
|
|
|
|
if items.count() > 0:
|
|
print(f"Found {items.count()} stock items with missing purchase price information")
|
|
|
|
update_count = 0
|
|
|
|
for item in items:
|
|
|
|
part_id = item.part
|
|
|
|
po = item.purchase_order
|
|
|
|
# Look for a matching PurchaseOrderLineItem (with a price)
|
|
lines = PurchaseOrderLineItem.objects.filter(part__part=part_id, order=po)
|
|
|
|
if lines.exists():
|
|
|
|
for line in lines:
|
|
if line.purchase_price is not None:
|
|
|
|
# Copy pricing information across
|
|
item.purchase_price = line.purchase_price
|
|
item.purchases_price_currency = line.purchase_price_currency
|
|
|
|
print(f"- Updating supplier price for {item.part.name} - {item.purchase_price} {item.purchase_price_currency}")
|
|
|
|
update_count += 1
|
|
|
|
item.save()
|
|
|
|
break
|
|
|
|
if update_count > 0:
|
|
print(f"Updated pricing for {update_count} stock items")
|
|
|
|
def reverse_operation(apps, schema_editor):
|
|
"""
|
|
DO NOTHING!
|
|
"""
|
|
pass
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
dependencies = [
|
|
('stock', '0063_auto_20210511_2343'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.RunPython(extract_purchase_price, reverse_code=reverse_operation)
|
|
]
|