337 lines
13 KiB
Python
337 lines
13 KiB
Python
from django import forms
|
|
from django.contrib.auth import get_user_model
|
|
from django.core.exceptions import ValidationError
|
|
from django.forms import widgets
|
|
from parts import models as parts_models
|
|
from shimatta_modules.EngineeringNumberConverter import EngineeringNumberConverter
|
|
import uuid
|
|
from django.urls import reverse
|
|
|
|
|
|
from crispy_forms.helper import FormHelper
|
|
from crispy_forms.layout import Layout, Row, Column
|
|
|
|
class AutoCompleteWidget(widgets.Input):
|
|
template_name = 'widgets/autocomplete-foreign-key.html'
|
|
|
|
def __init__(self, api_search_url, image_field_name, foreign_model, name_field_name, prepend=None, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.image_field_name = image_field_name
|
|
self.foreign_model = foreign_model
|
|
self.api_search_url = api_search_url
|
|
self.name_field_name = name_field_name
|
|
self.prepend = prepend
|
|
|
|
def get_context(self, name, value, attrs):
|
|
context = super().get_context(name, value, attrs)
|
|
if value is not None:
|
|
try:
|
|
try:
|
|
my_id = uuid.UUID(value)
|
|
except:
|
|
my_id = int(value)
|
|
instance = self.foreign_model.objects.get(id=my_id)
|
|
except Exception as ex:
|
|
instance = None
|
|
else:
|
|
instance = None
|
|
|
|
image = None
|
|
if instance is not None and self.image_field_name is not None:
|
|
image = getattr(instance, self.image_field_name)
|
|
|
|
display_name = None
|
|
if instance:
|
|
display_name = getattr(instance, self.name_field_name)
|
|
|
|
context['custom'] = {
|
|
'search_url': reverse(self.api_search_url),
|
|
'image_field_name': self.image_field_name,
|
|
'current_instance': instance,
|
|
'image': image,
|
|
'name_field_name': self.name_field_name,
|
|
'prepend': self.prepend,
|
|
'name': display_name,
|
|
}
|
|
return context
|
|
|
|
class AutocompleteForeingKeyField(forms.UUIDField):
|
|
def __init__(self,
|
|
foreign_model=None,
|
|
api_search_url=None,
|
|
image_field_name='image',
|
|
name_field_name='name',
|
|
prepend=None,
|
|
**kwargs):
|
|
super().__init__(**kwargs)
|
|
self.widget = AutoCompleteWidget(api_search_url,
|
|
image_field_name,
|
|
foreign_model,
|
|
name_field_name,
|
|
prepend)
|
|
self.foreign_model = foreign_model
|
|
|
|
|
|
def clean(self, value):
|
|
try:
|
|
pre_cleaned_uuid = super().clean(value)
|
|
except Exception as ex:
|
|
try:
|
|
pre_cleaned_uuid = int(value)
|
|
except:
|
|
raise ex
|
|
if pre_cleaned_uuid is None and not self.required:
|
|
return None
|
|
try:
|
|
obj = self.foreign_model.objects.get(id=pre_cleaned_uuid)
|
|
except self.foreign_model.DoesNotExist:
|
|
raise ValidationError('Given element does not exist')
|
|
return obj
|
|
|
|
class MyTestForm(forms.Form):
|
|
pass
|
|
|
|
class ChangeStorageForm(forms.Form):
|
|
storage_name = forms.CharField(label="storage_name", initial='')
|
|
responsible = AutocompleteForeingKeyField(api_search_url='user-list',
|
|
image_field_name=None,
|
|
name_field_name='username',
|
|
foreign_model=get_user_model(),
|
|
prepend='@')
|
|
|
|
is_template = forms.BooleanField(label='is_template', required=False)
|
|
|
|
class AddSubStorageForm(ChangeStorageForm):
|
|
template = AutocompleteForeingKeyField(api_search_url='storage-template-list',
|
|
image_field_name=None,
|
|
foreign_model=parts_models.Storage,
|
|
required=False)
|
|
|
|
class DeleteStockForm(forms.Form):
|
|
stock_uuid = forms.UUIDField()
|
|
|
|
class EditWatermarkForm(forms.Form):
|
|
stock_uuid = forms.UUIDField()
|
|
watermark_active = forms.BooleanField(required=False) #If it is false, the webbrowser won't send it at all. Therefore we have to set it to required=False
|
|
watermark = forms.IntegerField(min_value=0)
|
|
|
|
def clean(self):
|
|
cleaned_data = super().clean()
|
|
id = cleaned_data.get("stock_uuid")
|
|
|
|
if not id:
|
|
raise ValidationError("No stock UUID given")
|
|
|
|
stock = None
|
|
try:
|
|
stock = parts_models.Stock.objects.get(id=id)
|
|
except:
|
|
raise ValidationError("Stock with uuid %s does not exist" % (id))
|
|
cleaned_data['stock'] = stock
|
|
|
|
return cleaned_data
|
|
|
|
def save(self):
|
|
stock = self.cleaned_data['stock']
|
|
active = self.cleaned_data['watermark_active']
|
|
watermark = self.cleaned_data['watermark']
|
|
|
|
if not active:
|
|
watermark = -1
|
|
|
|
stock.watermark = watermark
|
|
stock.save()
|
|
|
|
class EditStockAmountForm(forms.Form):
|
|
stock_uuid = forms.UUIDField()
|
|
amount = forms.IntegerField(min_value=0)
|
|
|
|
def clean(self):
|
|
cleaned_data = super().clean()
|
|
id = cleaned_data.get("stock_uuid")
|
|
|
|
if not id:
|
|
raise ValidationError("No stock UUID given")
|
|
|
|
stock = None
|
|
try:
|
|
stock = parts_models.Stock.objects.get(id=id)
|
|
except:
|
|
raise ValidationError("Stock with uuid %s does not exist" % (id))
|
|
cleaned_data['stock'] = stock
|
|
|
|
return cleaned_data
|
|
|
|
def save(self, increase: bool):
|
|
stock = self.cleaned_data['stock']
|
|
amount = self.cleaned_data['amount']
|
|
|
|
if not increase:
|
|
amount = -amount
|
|
|
|
return stock.atomic_increment(amount)
|
|
|
|
class AddStockForm(forms.Form):
|
|
watermark_active = forms.BooleanField(required=False)
|
|
watermark = forms.IntegerField(min_value=0, required=True, initial=0)
|
|
amount = forms.IntegerField(min_value=0, required=True, initial=1)
|
|
component_uuid = forms.UUIDField(required=True)
|
|
lot = forms.CharField(max_length=255, required=False)
|
|
|
|
def clean(self):
|
|
cleaned_data = super().clean()
|
|
|
|
id = cleaned_data.get('component_uuid')
|
|
if not id:
|
|
raise ValidationError('No valid component selected!')
|
|
component = None
|
|
try:
|
|
component = parts_models.Component.objects.get(id=id)
|
|
except:
|
|
raise ValidationError("Invalid component selected!")
|
|
cleaned_data['component'] = component
|
|
|
|
return cleaned_data
|
|
|
|
def save(self, storage):
|
|
component = self.cleaned_data.get('component')
|
|
amount = self.cleaned_data.get('amount')
|
|
watermark = -1
|
|
|
|
if self.cleaned_data.get('watermark_active'):
|
|
watermark = self.cleaned_data.get('watermark')
|
|
|
|
new_stock = parts_models.Stock.objects.create(storage=storage, component=component, watermark=watermark, amount=amount, lot=self.cleaned_data['lot'])
|
|
new_stock.save()
|
|
|
|
class ComponentForm(forms.ModelForm):
|
|
manufacturer = AutocompleteForeingKeyField(api_search_url='manufacturer-list', foreign_model=parts_models.Manufacturer, required=False)
|
|
component_type = AutocompleteForeingKeyField(api_search_url='componenttype-list', foreign_model=parts_models.ComponentType, name_field_name='class_name', required=False, image_field_name=None)
|
|
package = AutocompleteForeingKeyField(api_search_url='package-list', foreign_model=parts_models.Package, required=False)
|
|
pref_distri = AutocompleteForeingKeyField(api_search_url='distributor-list', foreign_model=parts_models.Distributor, required=False)
|
|
class Meta:
|
|
model = parts_models.Component
|
|
fields = '__all__'
|
|
|
|
class PackageForm(forms.ModelForm):
|
|
class Meta:
|
|
model = parts_models.Package
|
|
fields = '__all__'
|
|
|
|
class DistributorForm(forms.ModelForm):
|
|
class Meta:
|
|
model = parts_models.Distributor
|
|
fields = '__all__'
|
|
class ManufacturerForm(forms.ModelForm):
|
|
class Meta:
|
|
model = parts_models.Manufacturer
|
|
fields = '__all__'
|
|
|
|
class DistributorNumberCreateForm(forms.ModelForm):
|
|
distributor = AutocompleteForeingKeyField(api_search_url='distributor-list', foreign_model=parts_models.Distributor, required=True)
|
|
|
|
class Meta:
|
|
model = parts_models.DistributorNum
|
|
fields = ['distributor', 'distributor_part_number']
|
|
|
|
class DistributorNumberDeleteForm(forms.Form):
|
|
distributor_num = forms.UUIDField(required=True)
|
|
|
|
def clean_distributor_num(self):
|
|
my_uuid = self.cleaned_data['distributor_num']
|
|
try:
|
|
distributor_number = parts_models.DistributorNum.objects.get(id=my_uuid)
|
|
except:
|
|
raise ValidationError('distributor number invalid')
|
|
return distributor_number
|
|
|
|
class ComponentParameterDeleteForm(forms.Form):
|
|
param_num = forms.UUIDField(required=True)
|
|
|
|
def clean_param_num(self):
|
|
my_uuid = self.cleaned_data['param_num']
|
|
try:
|
|
param = parts_models.ComponentParameter.objects.get(id=my_uuid)
|
|
except:
|
|
raise ValidationError('Parameter Number Invalid')
|
|
return param
|
|
|
|
class AdvancedComponentSearchForm(forms.Form):
|
|
name = forms.CharField(max_length=255, label='Component Name', required=False)
|
|
package = AutocompleteForeingKeyField(required=False, api_search_url='package-list', foreign_model=parts_models.Package)
|
|
package_pin_count = forms.IntegerField(min_value=0, label='Pin Count', required=False)
|
|
distributor_num = forms.CharField(max_length=100, label='Distributor Part Number', required=False)
|
|
distributor = AutocompleteForeingKeyField(required=False, api_search_url='distributor-list', foreign_model=parts_models.Distributor)
|
|
component_type = AutocompleteForeingKeyField(required=False, api_search_url='componenttype-list', foreign_model=parts_models.ComponentType, image_field_name=None, name_field_name='class_name')
|
|
manufacturer = AutocompleteForeingKeyField(required=False, api_search_url='manufacturer-list', foreign_model=parts_models.Manufacturer)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.helper = FormHelper()
|
|
self.helper.form_tag = False
|
|
self.helper.layout = Layout(
|
|
Row(
|
|
Column('name'),
|
|
),
|
|
Row(
|
|
Column('package'),
|
|
Column('package_pin_count'),
|
|
),
|
|
Row(
|
|
Column('component_type'),
|
|
Column('manufacturer'),
|
|
),
|
|
Row(
|
|
Column('distributor'),
|
|
Column('distributor_num'),
|
|
),
|
|
)
|
|
|
|
class ComponentParameterSearchForm(forms.Form):
|
|
parameter = AutocompleteForeingKeyField(required=True, foreign_model=parts_models.ComponentParameterType, api_search_url='componentparametertype-list', image_field_name=None, name_field_name='parameter_name')
|
|
value = forms.CharField(max_length=100, required=False)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.helper = FormHelper()
|
|
self.helper.form_tag = False
|
|
self.helper.layout = Layout(
|
|
Row(
|
|
Column('parameter'),
|
|
Column('value')
|
|
)
|
|
)
|
|
class ComponentParameterCreateForm(forms.Form):
|
|
parameter_type = AutocompleteForeingKeyField(required=True, foreign_model=parts_models.ComponentParameterType, api_search_url='componentparametertype-list', image_field_name=None, name_field_name='descriptive_name')
|
|
value = forms.CharField(required=True, max_length=256)
|
|
|
|
def clean(self):
|
|
data = super().clean()
|
|
ptype = data.get('parameter_type')
|
|
value = data.get('value')
|
|
|
|
if not ptype:
|
|
raise ValidationError('No valid parameter type selected')
|
|
|
|
if not value:
|
|
raise ValidationError('No valid parameter value')
|
|
|
|
if ptype.parameter_type == 'E' or ptype.parameter_type == 'I' or ptype.parameter_type == 'N':
|
|
try:
|
|
number = EngineeringNumberConverter.engineering_to_number(value)
|
|
except:
|
|
raise ValidationError('Cannot convert value to number')
|
|
data['number_value'] = number
|
|
else:
|
|
pass
|
|
|
|
def save(self, component):
|
|
param_type = self.cleaned_data['parameter_type']
|
|
if param_type.parameter_type == 'F':
|
|
text_value = self.cleaned_data['value']
|
|
value = 0
|
|
else:
|
|
text_value = ''
|
|
value = self.cleaned_data['number_value']
|
|
parts_models.ComponentParameter.objects.create(parameter_type=param_type, component=component, value=value, text_value=text_value)
|