Do some random coding for Distribotr numbers and Component parameters. This is probably not the most ideal solution. Will change this later
This commit is contained in:
@@ -11,12 +11,12 @@ from django.views import View
|
||||
import django.forms as forms
|
||||
from django.views.generic import TemplateView, DetailView
|
||||
from django.contrib.auth.mixins import LoginRequiredMixin, PermissionRequiredMixin
|
||||
from .models import Storage, Stock, Component, Distributor, Manufacturer, Package, ComponentParameter, ComponentParameterType
|
||||
from .models import Storage, Stock, Component, Distributor, Manufacturer, Package, ComponentParameter, ComponentParameterType, DistributorNum
|
||||
from .qr_parser import QrCodeValidator
|
||||
from django.core.paginator import Paginator
|
||||
from django.core.exceptions import ValidationError
|
||||
from django.db import IntegrityError
|
||||
from .forms import MyTestForm, AddSubStorageForm, DeleteStockForm, EditWatermarkForm, EditStockAmountForm, AddStockForm, EditComponentForm, EditComponentParameterForm
|
||||
from .forms import MyTestForm, AddSubStorageForm, DeleteStockForm, EditWatermarkForm, EditStockAmountForm, AddStockForm, EditComponentForm, EditComponentParameterForm, DistributorNumberForm, DistributorNumberFormSet
|
||||
from django.db.models import Q
|
||||
from django.db.models.functions import Lower
|
||||
import uuid
|
||||
@@ -391,16 +391,18 @@ class ComponentDetailView(LoginRequiredMixin, BaseTemplateMixin, DetailView):
|
||||
base_title = ''
|
||||
navbar_selected = 'Components'
|
||||
|
||||
|
||||
|
||||
def prepare_initial_param_formset_data(self):
|
||||
parameters = ComponentParameter.objects.filter(component=self.object)
|
||||
initdata = []
|
||||
for param in parameters:
|
||||
param_type = param.parameter_type
|
||||
if param_type.freetext_parameter:
|
||||
param_type = param.parameter_type.parameter_type
|
||||
if param_type == 'F':
|
||||
value = param.text_value
|
||||
else:
|
||||
value = param.value
|
||||
initdata.append({'parameter_type': param_type, 'value': value})
|
||||
initdata.append({'parameter_type': param.parameter_type, 'value': value})
|
||||
return initdata
|
||||
|
||||
def get_context_data(self, **kwargs):
|
||||
@@ -409,11 +411,14 @@ class ComponentDetailView(LoginRequiredMixin, BaseTemplateMixin, DetailView):
|
||||
context['component'] = self.object
|
||||
context['edit_form'] = EditComponentForm(instance=self.object)
|
||||
|
||||
ParameterFormset = forms.formset_factory(EditComponentParameterForm, extra=0, max_num=100)
|
||||
ParameterFormset = forms.formset_factory(EditComponentParameterForm, extra=1, max_num=100, can_delete=True)
|
||||
context['param_formset'] = ParameterFormset(
|
||||
initial=self.prepare_initial_param_formset_data())
|
||||
|
||||
context['stocks'] = Stock.objects.filter(component=self.object)
|
||||
|
||||
DistriNumFormSet = forms.modelformset_factory(DistributorNum, form=DistributorNumberForm, extra=2)
|
||||
context['distri_num_formset'] = DistriNumFormSet(queryset=DistributorNum.objects.filter(component=self.object), auto_id='id_fs_distri_no_%s')
|
||||
return context
|
||||
|
||||
def handle_submit_edit_post(self, request, **kwargs):
|
||||
@@ -438,18 +443,76 @@ class ComponentDetailView(LoginRequiredMixin, BaseTemplateMixin, DetailView):
|
||||
return self.render_to_response(context)
|
||||
|
||||
def handle_submit_edit_params_post(self, request, **kwargs):
|
||||
ParameterFormset = forms.formset_factory(EditComponentParameterForm, extra=1, max_num=100)
|
||||
fs = ParameterFormset(initial=self.prepare_initial_param_formset_data(), data=request.POST)
|
||||
errors_set = False
|
||||
|
||||
|
||||
if fs.is_valid():
|
||||
# Go through all the parameter forms:
|
||||
for form in fs:
|
||||
if not form.has_changed():
|
||||
continue
|
||||
|
||||
# Form has changed. Process the new value
|
||||
try:
|
||||
parameter_to_change = ComponentParameter.objects.get(component=self.object,
|
||||
parameter_type__parameter_name=form.initial['parameter_type'])
|
||||
parameter_to_change.parameter_type = form.cleaned_data['parameter_type_object']
|
||||
parameter_to_change.value = 0
|
||||
parameter_to_change.text_value = None
|
||||
except:
|
||||
try:
|
||||
parameter_to_change = ComponentParameter.objects.create(component=self.object,
|
||||
parameter_type=form.cleaned_data['parameter_type_object'])
|
||||
except:
|
||||
form.add_error('parameter_type', 'Parameter could not be saved. Unique?')
|
||||
errors_set = True
|
||||
break
|
||||
|
||||
parameter_to_change.value = 0
|
||||
parameter_to_change.text_value = None
|
||||
|
||||
if parameter_to_change.parameter_type.parameter_type == 'F':
|
||||
parameter_to_change.text_value = form.cleaned_data['processed_value']
|
||||
parameter_to_change.value = 0
|
||||
else:
|
||||
parameter_to_change.value = form.cleaned_data['processed_value']
|
||||
|
||||
try:
|
||||
parameter_to_change.save()
|
||||
except:
|
||||
form.add_error('parameter_type', 'Parameter could not be saved. Unique?')
|
||||
errors_set = True
|
||||
|
||||
context = self.get_context_data()
|
||||
if not fs.is_valid() or errors_set:
|
||||
context['param_formset'] = fs
|
||||
return self.render_to_response(context)
|
||||
|
||||
def handle_submit_edit_distri_nums_post(self, request, **kwargs):
|
||||
DistriNumFormSet = forms.modelformset_factory(DistributorNum, form=DistributorNumberForm, extra=2, formset=DistributorNumberFormSet)
|
||||
fs = DistriNumFormSet(queryset=DistributorNum.objects.filter(component=self.object), data=request.POST, auto_id='id_fs_distri_no_%s')
|
||||
|
||||
if fs.is_valid():
|
||||
print('Valid')
|
||||
fs.save(self.object)
|
||||
else:
|
||||
print('Invalid')
|
||||
|
||||
context = self.get_context_data()
|
||||
if not fs.is_valid():
|
||||
context['distri_num_formset'] = fs
|
||||
return self.render_to_response(context)
|
||||
|
||||
|
||||
def post(self, request, *args, **kwargs):
|
||||
self.object = self.get_object()
|
||||
if 'submit-edit-comp' in request.POST:
|
||||
return self.handle_submit_edit_post(request, **kwargs)
|
||||
elif 'submit-edit-params' in request.POST:
|
||||
return self.handle_submit_edit_params_post(request, **kwargs)
|
||||
elif 'submit-edit-distri-nums' in request.POST:
|
||||
return self.handle_submit_edit_distri_nums_post(request, **kwargs)
|
||||
|
||||
return super().post(request, *args, **kwargs)
|
||||
|
||||
|
Reference in New Issue
Block a user