# -*- coding: utf-8 -*-
# This file is part of Shuup.
#
# Copyright (c) 2012-2017, Shoop Commerce Ltd. All rights reserved.
#
# This source code is licensed under the OSL-3.0 license found in the
# LICENSE file in the root directory of this source tree.
from django.conf import settings
from shuup.core.models import Product, StockBehavior
from shuup.core.stocks import ProductStockStatus
from shuup.core.suppliers import BaseSupplierModule
from shuup.core.suppliers.enums import StockAdjustmentType
from shuup.simple_supplier.utils import get_current_stock_value
from .models import StockAdjustment, StockCount
[docs]class SimpleSupplierModule(BaseSupplierModule):
identifier = "simple_supplier"
name = "Simple Supplier"
[docs] def get_stock_statuses(self, product_ids):
stock_counts = (
StockCount.objects
.filter(supplier=self.supplier, product_id__in=product_ids)
.values_list("product_id", "physical_count", "logical_count")
)
values = dict(
(product_id, (physical_count, logical_count))
for (product_id, physical_count, logical_count)
in stock_counts
)
null = (0, 0)
stati = [ProductStockStatus(
product_id=product_id,
physical_count=values.get(product_id, null)[0],
logical_count=values.get(product_id, null)[1]
) for product_id in product_ids]
return dict((pss.product_id, pss) for pss in stati)
[docs] def adjust_stock(self, product_id, delta, purchase_price=0, created_by=None,
type=StockAdjustmentType.INVENTORY):
adjustment = StockAdjustment.objects.create(
supplier=self.supplier,
product_id=product_id,
delta=delta,
purchase_price_value=purchase_price,
created_by=created_by,
type=type
)
self.update_stock(product_id)
return adjustment
[docs] def update_stock(self, product_id):
supplier_id = self.supplier.pk
# TODO: Consider whether this should be done without a cache table
values = get_current_stock_value(supplier_id=supplier_id, product_id=product_id)
sv, _ = StockCount.objects.get_or_create(supplier_id=supplier_id, product_id=product_id)
sv.logical_count = values["logical_count"]
sv.physical_count = values["physical_count"]
latest_event = (
StockAdjustment.objects
.filter(supplier=supplier_id, product=product_id, type=StockAdjustmentType.INVENTORY)
.last())
if latest_event:
sv.stock_value_value = latest_event.purchase_price_value * sv.logical_count
if "shuup.notify" in settings.INSTALLED_APPS:
if sv.alert_limit and sv.physical_count < sv.alert_limit:
product = Product.objects.filter(id=product_id).first()
if product and product.stock_behavior == StockBehavior.STOCKED:
from .notify_events import AlertLimitReached
AlertLimitReached(supplier=self.supplier, product=product).run()
sv.save(update_fields=("logical_count", "physical_count", "stock_value_value"))