ReservablePricingFormPart

Complexity

Total Complexity

2

Size/Duplication

Total Lines	14
Duplicated Lines	0 %

2 Methods

Rating	Name	Duplication	Size	Complexity
A	form_valid()	0	2	1
A	get_form_defs()	0	7	1

# -*- coding: utf-8 -*-
from __future__ import unicode_literals

from django import forms
from django.db.models import Q
from django.utils.translation import ugettext_lazy as _

from reservable_pricing.models import ReservableProductPrice
from shoop.admin.form_part import FormPart, TemplatedFormDef
from shoop.core.models import ContactGroup, Shop


class ReservablePricingForm(forms.Form):
    def __init__(self, **kwargs):
        self.product = kwargs.pop("product", None)
        super(ReservablePricingForm, self).__init__(**kwargs)
        self.shops = []
        self.groups = []
        if self.product:
            self._build_fields()

    def _build_fields(self):
        self.shops = list(Shop.objects.all())
        self.groups = list(ContactGroup.objects.filter(
            Q(show_pricing=True) |
            Q(
                id__in=ReservableProductPrice.objects.filter(product=self.product)

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

                .values_list("group_id", flat=True).distinct()
            )
        ))
        prices_by_shop_and_group = dict(
            ((shop_id or 0, group_id or 0), price)
            for (shop_id, group_id, price)
            in ReservableProductPrice.objects.filter(product=self.product)

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

            .values_list("shop_id", "group_id", "price_value")
        )

        for group in self.groups:
            for shop in self.shops:
                shop_group_id_tuple = self._get_id_tuple(shop, group)
                name = self._get_field_name(shop_group_id_tuple)
                price = prices_by_shop_and_group.get(shop_group_id_tuple)
                price_field = forms.DecimalField(
                    min_value=0, initial=price,
                    label=_("Price (%s/%s)") % (shop, group), required=False
                )
                self.fields[name] = price_field

The Instance of ReservablePricingForm does not seem to have a member named fields.

    def _get_id_tuple(self, shop, group):
        return (
            shop.id if shop else 0,
            group.id if group else 0
        )

    def _get_field_name(self, id_tuple):
        return "s_%d_g_%d" % id_tuple

    def _process_single_save(self, shop, group):
        shop_group_id_tuple = self._get_id_tuple(shop, group)
        name = self._get_field_name(shop_group_id_tuple)
        value = self.cleaned_data.get(name)

The Instance of ReservablePricingForm does not seem to have a member named cleaned_data.

        clear = (value is None or value < 0)
        if clear:
            ReservableProductPrice.objects.filter(product=self.product, group=group, shop=shop).delete()

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

        else:
            (spp, created) = ReservableProductPrice.objects.get_or_create(

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

                product=self.product, group=group, shop=shop,
                defaults={'price_value': value})
            if not created:
                spp.price_value = value
                spp.save()

    def save(self):
        if not self.has_changed():  # No changes, so no need to do anything.

The Instance of ReservablePricingForm does not seem to have a member named has_changed.

            # (This is required because `.full_clean()` would create an empty `.cleaned_data`,
            #  but short-circuits out if `has_changed()` returns false.
            #  That, in kind, would cause `self.cleaned_data.get(name)` in `_process_single_save`
            #  to return Nones, clearing out all prices. Oops.)
            return

        for group in self.groups:
            for shop in self.shops:
                self._process_single_save(shop, group)

    def get_shop_group_field(self, shop, group):
        shop_group_id_tuple = self._get_id_tuple(shop, group)
        name = self._get_field_name(shop_group_id_tuple)
        return self[name]


class ReservablePricingFormPart(FormPart):
    priority = 10

    def get_form_defs(self):
        yield TemplatedFormDef(
            name="reservable_pricing",
            form_class=ReservablePricingForm,
            template_name="reservable_pricing/admin/form_part.jinja",
            required=False,
            kwargs={"product": self.object}

The Instance of ReservablePricingFormPart does not seem to have a member named object.

        )

    def form_valid(self, form):
        form["reservable_pricing"].save()