root/trunk/djedna/store/forms.py

# (c) Copyright 2008 Thomas Bohmbach, Jr.
# 
# This file is part of DJ Edna.
# 
# DJ Edna is free software: you can redistribute it and/or modify it under the
# terms of the GNU General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option)
# any later version.
# 
# DJ Edna is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
# or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
# more details.
# 
# You should have received a copy of the GNU General Public License along with
# DJ Edna.  If not, see <http://www.gnu.org/licenses/>.

from django import newforms as forms
from django.contrib.contenttypes.models import ContentType

from models import FixedPrice, FloatingPrice


class PriceForm(forms.Form):
    def __init__(self, pricing_object, *args, **kwargs):
        super(PriceForm, self).__init__(*args, **kwargs)
        self.pricing_object = pricing_object
    
    def _get_final_price(self):
        return self.pricing_object.price
    final_price = property(_get_final_price)
    

class FloatingPriceForm(PriceForm):
    desired_price = forms.DecimalField(required=True, max_digits=7, decimal_places=2)
    
    def __init__(self, pricing_object, *args, **kwargs):
        super(FloatingPriceForm, self).__init__(pricing_object, *args, **kwargs)
        self.fields['desired_price'].min_value = pricing_object.minimum
    
    def _get_final_price(self):
        return self.cleaned_data['desired_price']
    final_price = property(_get_final_price)
    

def create_price_form(product, data=None):
    price_model_type = product.pricing_type
    if price_model_type == ContentType.objects.get_for_model(FixedPrice):
        return PriceForm(product.pricing_object, data, prefix=product.id)
    elif price_model_type == ContentType.objects.get_for_model(FloatingPrice):
        return FloatingPriceForm(product.pricing_object, data, prefix=product.id,
                                 initial={'desired_price' : product.pricing_object.recommended})

def create_price_forms(products, data=None, selected=None):
    product_forms = []
    selected_form = None
    for product in products:
        if product == selected:
            selected_form = create_price_form(product, data=data)
            product_forms.append((product, selected_form))
        else:
            product_forms.append((product, create_price_form(product)))
    return (product_forms, selected_form)