diff --git a/computers/forms.py b/computers/forms.py index 34e5332..014ab8d 100644 --- a/computers/forms.py +++ b/computers/forms.py @@ -1,6 +1,7 @@ import floppyforms.__future__ as forms from .models import Computer +from customers.models import Customer class ComputerCreateForm(forms.ModelForm): @@ -11,6 +12,11 @@ class ComputerCreateForm(forms.ModelForm): 'customer', ) + def __init__(self, pk=None, user=None, *args, **kwargs): + super(ComputerCreateForm, self).__init__(*args, **kwargs) + if not user.is_superuser: + self.fields['customer'].queryset = (Customer.objects.filter(id=pk)) + class ComputerUpdateForm(forms.ModelForm): class Meta: