inventree/InvenTree
2
0
Fork 0
mirror of https://github.com/inventree/InvenTree.git synced 2025-04-29 12:06:44 +00:00
Code

Code cleanup

Browse Source
This commit is contained in:
Oliver Walters 2017-04-20 22:20:41 +10:00
parent 41485e0e0f
commit 92cbd43f0f
1 changed files with 3 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
InvenTree/stock/models.py
View File

@ -78,27 +78,6 @@ class StockItem(models.Model):
self.stocktake_date = datetime.now().date() self.stocktake_date = datetime.now().date()
self.save() self.save()
def take_stock(self, amount):
""" Take items from stock
This function can be called by initiating a ProjectRun,
or by manually taking the items from the stock location
"""
if self.infinite:
return
amount = int(amount)
if amount < 0:
raise ValueError("Stock amount must be positive")
q = self.quantity - amount
if q < 0:
q = 0
self.quantity = q
self.save()
def add_stock(self, amount): def add_stock(self, amount):
""" Add items to stock """ Add items to stock
This function can be called by initiating a ProjectRun, This function can be called by initiating a ProjectRun,
@ -119,6 +98,9 @@ class StockItem(models.Model):
self.quantity = q self.quantity = q
self.save() self.save()
def take_stock(self, amount):
self.add_stock(-amount)
def __str__(self): def __str__(self):
return "{n} x {part} @ {loc}".format( return "{n} x {part} @ {loc}".format(
n=self.quantity, n=self.quantity,