import uuid from django.shortcuts import redirect from django.urls import reverse from django.views.generic import TemplateView, DetailView from django.contrib.auth.mixins import LoginRequiredMixin from django.core.paginator import Paginator from django.db.models import Q from django.forms import formset_factory from django.db import IntegrityError from django.db.models import ProtectedError from ..models import Stock, Component, ComponentParameter, DistributorNum from ..forms import * from .component_import import import_components_from_csv from .generic_views import BaseTemplateMixin ParameterSearchFormSet = formset_factory(ComponentParameterSearchForm, extra=1) # Create your views here. class ComponentView(LoginRequiredMixin, BaseTemplateMixin, TemplateView): template_name = 'parts/components.html' base_title = 'Components' navbar_selected = 'Components' default_page_size = 25 def get_component_query_set(self, search_string): queryset = Component.objects.all() if not search_string: return queryset search_fragments = search_string.strip().split() for search in search_fragments: queryset = queryset.filter(Q(name__icontains = search) | Q(manufacturer__name__icontains = search) | Q(package__name__icontains = search)) return queryset def get_component_queryset_from_advanced_search(self, cleaned_data): queryset = Component.objects.all() if cleaned_data['name']: queryset = queryset.filter(Q(name__icontains=cleaned_data['name'])) if cleaned_data['package']: queryset = queryset.filter(package=cleaned_data['package']) if cleaned_data['package_pin_count']: queryset = queryset.filter(package__pin_count=cleaned_data['package_pin_count']) if cleaned_data['component_type']: queryset = queryset.filter(component_type=cleaned_data['component_type']) if cleaned_data['distributor_num']: if cleaned_data['distributor']: distri = cleaned_data['distributor'] queryset = queryset.filter(Q(distributornum__distributor_part_number__icontains=cleaned_data['distributor_num']) & Q(distributornum__distributor=distri)) else: queryset = queryset.filter(Q(distributornum__distributor_part_number__icontains=cleaned_data['distributor_num'])) if cleaned_data['manufacturer']: queryset = queryset.filter(manufacturer=cleaned_data['manufacturer']) return queryset def get_context_data_int(self, advanced_search, parameter_formset : ParameterSearchFormSet, **kwargs): context = super().get_context_data(**kwargs) comp_page_num = self.request.GET.get('comp_page', default=1) if advanced_search and parameter_formset: search = None context['advanced_search_shown'] = True context['advanced_search_form'] = advanced_search context['advanced_search_param_formset'] = parameter_formset if advanced_search.is_valid(): paginator_queryset = self.get_component_queryset_from_advanced_search( advanced_search.cleaned_data) else: paginator_queryset = Component.objects.all() if parameter_formset.is_valid(): # Process parameters pass else: search = self.request.GET.get('search', default=None) paginator_queryset = self.get_component_query_set(search) comp_paginator = Paginator(paginator_queryset, self.default_page_size) context['components'] = comp_paginator.get_page(comp_page_num) context['comp_form'] = ComponentForm() context['import_comp_form'] = ImportComponentForm() context['search_string'] = search if not parameter_formset: context['advanced_search_param_formset'] = ParameterSearchFormSet() if not advanced_search: context['advanced_search_form'] = AdvancedComponentSearchForm(auto_id='adv_search_%s') return context def get_context_data(self, **kwargs): return self.get_context_data_int(advanced_search = None, parameter_formset=None, **kwargs) def handle_new_component_post(self, request, open=False, **kwargs): cform = ComponentForm(data=request.POST, files=request.FILES) new_component = None if cform.is_valid(): new_component = cform.save() context = self.get_context_data(**kwargs) if not cform.is_valid(): context['comp_form'] = cform if open and new_component: return redirect(reverse('parts-components-detail', kwargs={'uuid':new_component.id})) return self.render_to_response(context) def handle_import_components_post(self, request, open=False, **kwargs): cform = ImportComponentForm(data=request.POST, files=request.FILES) context = self.get_context_data(**kwargs) if cform.is_valid(): try: import_components_from_csv(cform.files['csv_file'].file) except Exception as ex: cform.add_error('csv_file', str(ex)) context['import_comp_form'] = cform else: context['import_comp_form'] = cform return self.render_to_response(context) def handle_advanced_search_post(self, request, **kwargs): form = AdvancedComponentSearchForm(auto_id='adv_search_%s', data=request.POST) param_formset = ParameterSearchFormSet(data=request.POST) if form.is_valid(): print('Valid') if param_formset.is_valid(): print('Formset is valid!') context = self.get_context_data_int(form, param_formset, **kwargs) return self.render_to_response(context) def post(self, request, *args, **kwargs): if 'submit-edit-component' in request.POST: return self.handle_new_component_post(request, open=False, **kwargs) elif 'submit-edit-component-open' in request.POST: return self.handle_new_component_post(request, open=True, **kwargs) elif 'submit-import-components' in request.POST: return self.handle_import_components_post(request, open=True, **kwargs) elif 'submit-advanced-search' in request.POST: return self.handle_advanced_search_post(request, **kwargs) else: return super().post(request, *args, **kwargs) class ComponentDetailView(LoginRequiredMixin, BaseTemplateMixin, DetailView): template_name = 'parts/components-detail.html' model = Component pk_url_kwarg = 'uuid' base_title = '' navbar_selected = 'Components' def get_context_data(self, **kwargs): self.base_title = 'Component / '+self.object.name context = super().get_context_data(**kwargs) context['component'] = self.object context['stocks'] = Stock.objects.filter(component=self.object) context['comp_form'] = ComponentForm(instance=self.object) context['new_distri_num_form'] = DistributorNumberCreateForm() context['new_param_form'] = ComponentParameterCreateForm() context['distri_nums'] = DistributorNum.objects.filter(component=self.object).order_by( 'distributor__name') context['parameters'] = ComponentParameter.objects.filter(component=self.object).order_by( 'parameter_type__parameter_name') return context def handle_submit_edit_component_post(self, request, **kwargs): cform = ComponentForm(instance=self.object, data=request.POST, files=request.FILES) if cform.is_valid(): cform.save() context = self.get_context_data(**kwargs) if not cform.is_valid(): context['comp_form'] = cform return self.render_to_response(context) def handle_submit_delete_post(self, request, **kwargs): delete_error = None protected_stuff = None try: self.object.delete() except ProtectedError as pe: delete_error = 'Component is protected' protected_stuff = pe.protected_objects except: delete_error = 'Cannot delete component. Unknown error' if delete_error is None: return redirect('parts-components') else: context = self.get_context_data(**kwargs) context['delete_error'] = delete_error context['protected_stuff'] = protected_stuff return self.render_to_response(context) def handle_submit_new_distri_num_post(self, request, **kwargs): form = DistributorNumberCreateForm(data=request.POST) if form.is_valid(): new_number = form.save(commit=False) new_number.component = self.object try: new_number.save() except IntegrityError as ie: form.add_error('__all__', 'Number for given distributor already exists') context = self.get_context_data(**kwargs) if not form.is_valid(): context['new_distri_num_form'] = form return self.render_to_response(context) def handle_submit_delete_distri_num_post(self, request, **kwargs): form = DistributorNumberDeleteForm(data=request.POST) if form.is_valid(): form.cleaned_data['distributor_num'].delete() context = self.get_context_data(**kwargs) return self.render_to_response(context) def handle_submit_delete_param_post(self, request, **kwargs): form = ComponentParameterDeleteForm(data=request.POST) if form.is_valid(): form.cleaned_data['param_num'].delete() context = self.get_context_data(**kwargs) return self.render_to_response(context) def handle_submit_new_param_post(self, request, **kwargs): form = ComponentParameterCreateForm(data=request.POST) if form.is_valid(): try: form.save(self.object) except IntegrityError: form.add_error('__all__', 'This parameter is already set') context = self.get_context_data(**kwargs) if not form.is_valid(): context['new_param_form'] = form return self.render_to_response(context) def post(self, request, *args, **kwargs): self.object = self.get_object() if 'submit-edit-component' in request.POST: return self.handle_submit_edit_component_post(request, **kwargs) elif 'submit-component-delete' in request.POST: return self.handle_submit_delete_post(request, **kwargs) elif 'submit-create-new-distri-num' in request.POST: return self.handle_submit_new_distri_num_post(request, **kwargs) elif 'submit-delete-distributor-num' in request.POST: return self.handle_submit_delete_distri_num_post(request, **kwargs) elif 'submit-delete-param' in request.POST: return self.handle_submit_delete_param_post(request, **kwargs) elif 'submit-create-new-param' in request.POST: return self.handle_submit_new_param_post(request, **kwargs) else: return super().post(request, *args, **kwargs)