Pass variable from view to form in Django

Question

I am building a simple task management system, where a Company can have multiple projects, and each company has employees. I want a form that allows managers to add users to projects, with the constraint that the available users belong to the company.

I am passing the variable company_pk from the view to the form, but I am not sure how to set/access the variable outside the init finction.

class AddUserForm(forms.Form):
    def __init__(self, company_pk=None, *args, **kwargs):
        """
        Intantiation service.
        This method extends the default instantiation service.
        """
        super(AddUserForm, self).__init__(*args, **kwargs)
        if company_pk:
            print("company_pk: ", company_pk)
            self._company_pk = company_pk

    user = forms.ModelChoiceField(
        queryset=User.objects.filter(company__pk=self._company_pk))

form = AddUserForm(company_pk=project_id)

As mentioned, I want to filter the users to only those belonging to a given company, however I do not know how to access the company_pk outside of init. I get the error: NameError: name 'self' is not defined

I think so; I have project.users as ManyToMany and project.company as ForeignKey — user1314147, Jul 03 '19 at 15:59
this may help you https://stackoverflow.com/questions/291945/how-do-i-filter-foreignkey-choices-in-a-django-modelform — Bob White, Jul 03 '19 at 15:59

score 2 · Accepted Answer · answered Jul 03 '19 at 16:04

2

class AddUserForm(forms.Form):
    def __init__(self, company_pk=None, *args, **kwargs):
        super(AddUserForm, self).__init__(*args, **kwargs)
        self.fields['user'].queryset = User.objects.filter(company__pk=company_pk)

    user = forms.ModelChoiceField(queryset=User.objects.all())

answered Jul 03 '19 at 16:04

Mahmood Al Rayhan

642
5
14

Thanks, this did the trick! I had to modify my views.py method a little: form = AddUserForm(project.company.pk, request.POST) – user1314147 Jul 04 '19 at 18:24

score 0 · Answer 2 · answered Jul 03 '19 at 16:07

You must use self.fields to override the user's queryset

class AddUserForm(forms.Form):
    def __init__(self, company_pk=None, *args, **kwargs):
        super(AddUserForm, self).__init__(*args, **kwargs)
        if company_pk:
            self.fields['user'].queryset = User.objects.filter(company__pk=company_pk))

for more information about it. check this out How to dynamically filter ModelChoice's queryset in a ModelForm

Pass variable from view to form in Django

2 Answers2