ReservablePricingModule

Complexity

Total Complexity

8

Size/Duplication

Total Lines	54
Duplicated Lines	0 %

2 Methods

Rating	Name	Duplication	Size	Complexity
B	get_price_info()	0	35	5
A	get_context_from_request()	0	11	3

# -*- coding: utf-8 -*-
import six
from django.db.models import Q
from django.utils.translation import ugettext_lazy as _

from shoop.core.models import ShopProduct
from shoop.core.pricing import PriceInfo, PricingContext, PricingModule

from .models import ReservableProductPrice


class ReservablePricingContext(PricingContext):
    REQUIRED_VALUES = ("customer_group_ids", "shop")
    customer_group_ids = ()
    shop = None


class ReservablePricingModule(PricingModule):
    identifier = "reservable_pricing"
    name = _("Reservable Pricing")

    pricing_context_class = ReservablePricingContext

    def get_context_from_request(self, request):
        customer = getattr(request, "customer", None)

        if not customer or customer.is_anonymous:
            customer_group_ids = []
        else:
            customer_group_ids = sorted(customer.groups.all().values_list("id", flat=True))

        return self.pricing_context_class(
            shop=request.shop,
            customer_group_ids=customer_group_ids
        )

    def get_price_info(self, context, product, quantity=1):
        shop = context.shop

        if isinstance(product, six.integer_types):
            product_id = product
            shop_product = ShopProduct.objects.get(product_id=product_id, shop=shop)
        else:
            shop_product = product.get_shop_instance(shop)
            product_id = product.pk

        default_price = (shop_product.default_price_value or 0)

        if context.customer_group_ids:
            filter = Q(

This seems to re-define the built-in filter.

                price_value__gt=0, product=product_id, shop=shop,
                group__in=context.customer_group_ids)
            result = (
                ReservableProductPrice.objects.filter(filter)

The Class ReservableProductPrice does not seem to have a member named objects.

                .order_by("price_value")[:1]
                .values_list("price_value", flat=True)
            )
        else:
            result = None

        if result:
            price = result[0]
            if default_price > 0:
                price = min([default_price, price])
        else:
            price = default_price

        return PriceInfo(
            price=shop.create_price(price * quantity),
            base_price=shop.create_price(price * quantity),
            quantity=quantity,
        )