proekton.com/users/forms.py

from django import forms
import itertools
import pydash as _; _.map = _.map_; _.filter = _.filter_

from .models import User, UserFinancialInfo, Team, GENDERS
from common.models import Location
from projects.models import Project, Realty, BuildingClassfication, ConstructionType
from specializations.models import Specialization


class TeamForm(forms.ModelForm):

    class Meta:
        model = Team
        fields = (
            'name',
            'owner',
        )


class UserProfileEditForm(forms.ModelForm):
    gender = forms.ChoiceField(
        choices=GENDERS,
        widget=forms.RadioSelect,
    )
    
    class Meta:
        model = User
        
        fields = (
            'avatar',
            'contractor_specializations',
            'cro',
            'date_of_birth',
            'first_name',
            'gender',
            'last_name',
            'location',
            'patronym',
            'phone',
            'skype',
            'website',
        )
        
        widgets = {
            'date_of_birth': forms.SelectDateWidget(years=range(1940, 2015), attrs={'class': 'selectpicker'}),
        }
    
    def __init__(self, *args, **kwargs):
        self.request = kwargs.pop('request')
        super().__init__(*args, **kwargs)


class UserFinancialInfoEditForm(forms.ModelForm):
    class Meta:
        model = UserFinancialInfo
        
        fields = (
            'credit_card_number',
            'date_of_birth',
            'fio',
            'inn',
            'legal_status',
            'passport_issue_date',
            'passport_issued_by',
            'passport_number',
            'passport_scan',
            'passport_series',
            'phone',
            'residency',
            'subdivision_code',
            'yandex_money',
        )
        
        widgets = {
            'residency': forms.RadioSelect(),
            'legal_status': forms.RadioSelect(),
        }
    
    def __init__(self, *args, **kwargs):
        self.request = kwargs.pop('request')
        super().__init__(*args, **kwargs)
        
        self.fields['residency'].choices = self.fields['residency'].choices[1:]
        self.fields['legal_status'].choices = self.fields['legal_status'].choices[1:]
        
        # self.fields['residency'].empty_label = None
        # self.fields['residency'].widget.choices = self.fields['residency'].choices


class ContractorFilterForm(forms.Form):
    CONTRACTOR_ORDER_CHOICES = (
        ('', 'Сортировать по...'),
        ('name', 'Названию'),
        ('created', 'Дате размещения'),
        ('views', 'Просмотрам'),
        ('reviews', 'Отзывам'),
        ('rating', 'Рейтингу'),
        # ('project_number', 'Кол-ву выполн. проектов'),
    )
    
    PARTY_TYPES = (
        ('all', 'Все'),
        ('teams', 'Группы'),
        ('contractors', 'Исполнители'),
    )
    
    order_by = forms.ChoiceField(required=False, choices=CONTRACTOR_ORDER_CHOICES)
    last_order_by = forms.ChoiceField(required=False, choices=CONTRACTOR_ORDER_CHOICES)
    reverse_order = forms.BooleanField(required=False)
    
    specialization = forms.ModelChoiceField(
        queryset=Specialization.objects.root_nodes()[0].get_descendants(),
        # queryset=Specialization.objects, # Migrate with this enabled
        required=False,
    )
    
    location = forms.ModelChoiceField(
        queryset=Location.objects.root_nodes()[0].get_descendants(),
        # queryset=Location.objects, # Migrate with this enabled
        required=False,
    )
    
    building_classification = forms.ModelChoiceField(
        queryset=BuildingClassfication.objects,
        widget=forms.Select(attrs={'class': 'selectpicker'}),
        required=False,
        empty_label='',
    )
    
    work_type = forms.ChoiceField(
        choices=tuple(itertools.chain((('',''),), Project.WORK_TYPES)),
        widget=forms.Select(attrs={'class': 'selectpicker'}),
        required=False,
    )
    
    construction_type = forms.ModelChoiceField(
        queryset=ConstructionType.objects,
        widget=forms.Select(attrs={'class': 'selectpicker'}),
        required=False,
        empty_label='',
    )
    
    cro = forms.BooleanField(required=False)
    
    party_types = forms.ChoiceField(
        choices=PARTY_TYPES,
        required=False,
    )
    
    last_party_types = forms.ChoiceField(
        choices=PARTY_TYPES,
        required=False,
    )
    
    def __init__(self, *args, **kwargs):
        self.request = kwargs.pop('request')
        super().__init__(*args, **kwargs)


class CustomerProfileProjectRealtyForm(forms.Form):
    def __init__(self, *args, **kwargs):
        self.request = kwargs.pop('request')
        self.customer = kwargs.pop('customer')
        
        super().__init__(*args, **kwargs)
        
        realties = _.uniq(tuple(p.realty for p in self.customer.projects.all()))
        
        self.fields['realty'] = forms.ChoiceField(
            widget=forms.Select(attrs={
                'class': 'selectpicker',
                'onchange': "$(this).closest('form').submit()",
            }),
            
            choices=(('', 'Все объекты'),) + tuple((r.pk, r.name) for r in realties),
            required=False,
        )


# import code; code.interact(local=dict(globals(), **locals()))