115 lines
3.2 KiB
Python
115 lines
3.2 KiB
Python
import floppyforms.__future__ as forms
|
|
|
|
from core import utils
|
|
|
|
from customers.models import Customer
|
|
from customers.models import Location
|
|
|
|
from devices.models import Device
|
|
from devices.models import DeviceInNet
|
|
from devices.models import Warranty
|
|
|
|
from users.models import User
|
|
|
|
|
|
class DeviceCreateForm(forms.ModelForm):
|
|
class Meta:
|
|
model = Device
|
|
fields = (
|
|
'name',
|
|
'customer',
|
|
)
|
|
|
|
def __init__(self, user=None, *args, **kwargs):
|
|
"""
|
|
If the user is not a superuser it's always assigned to a customer which
|
|
we can use to assign to the field.
|
|
"""
|
|
super(DeviceCreateForm, self).__init__(*args, **kwargs)
|
|
self.fields['customer'].queryset = (
|
|
utils.objects_for_allowed_customers(
|
|
Customer, user=user))
|
|
|
|
|
|
class DeviceUpdateForm(forms.ModelForm):
|
|
"""
|
|
Basic form class to use crispies HTML5 forms.
|
|
"""
|
|
|
|
def __init__(self, request, *args, **kwargs):
|
|
super(DeviceUpdateForm, self).__init__(*args, **kwargs)
|
|
customers = utils.objects_for_allowed_customers(Customer,
|
|
user=request.user)
|
|
locations = utils.objects_for_allowed_customers(Location,
|
|
user=request.user)
|
|
users = utils.objects_for_allowed_customers(User,
|
|
user=request.user)
|
|
self.fields['customer'].queryset = customers
|
|
self.fields['location'].queryset = locations
|
|
self.fields['user'].queryset = users
|
|
|
|
class Meta:
|
|
model = Device
|
|
fields = '__all__'
|
|
exclude = ('net',)
|
|
|
|
|
|
class WarrantyCreateForm(forms.ModelForm):
|
|
"""
|
|
Basic form class to use crispies HTML5 forms.
|
|
"""
|
|
|
|
def clean(self):
|
|
cleaned_data = super().clean()
|
|
valid_from = cleaned_data.get("valid_from")
|
|
valid_until = cleaned_data.get("valid_until")
|
|
|
|
if valid_from and valid_until:
|
|
if valid_from > valid_until:
|
|
raise forms.ValidationError(
|
|
"Valid from date must be before valid until date")
|
|
return cleaned_data
|
|
|
|
class Meta:
|
|
model = Warranty
|
|
fields = '__all__'
|
|
|
|
|
|
class WarrantyUpdateForm(forms.ModelForm):
|
|
"""
|
|
Basic form class to use crispies HTML5 forms.
|
|
"""
|
|
|
|
def clean(self):
|
|
cleaned_data = super().clean()
|
|
valid_from = cleaned_data.get("valid_from")
|
|
valid_until = cleaned_data.get("valid_until")
|
|
|
|
if valid_from and valid_until:
|
|
if valid_from > valid_until:
|
|
raise forms.ValidationError(
|
|
"Valid from date must be before valid until date")
|
|
return cleaned_data
|
|
|
|
class Meta:
|
|
model = Warranty
|
|
fields = '__all__'
|
|
|
|
|
|
class DeviceInNetCreateForm(forms.ModelForm):
|
|
"""
|
|
Basic form class to use crispies HTML5 forms.
|
|
"""
|
|
class Meta:
|
|
model = DeviceInNet
|
|
fields = '__all__'
|
|
|
|
|
|
class DeviceInNetUpdateForm(forms.ModelForm):
|
|
"""
|
|
Basic form class to use crispies HTML5 forms.
|
|
"""
|
|
class Meta:
|
|
model = DeviceInNet
|
|
fields = '__all__'
|