Question

I can’t find a single example of anyone doing this apart from this example, which doesn’t help me other than to know where the code needs to sit.

How to prepopulate UserProfile fields in the Django admin?

so this is my code

class QuoteMaterial(models.Model):
    name = models.CharField(_('name'), max_length=255)
    content = models.TextField(_('content'),
                           help_text=_('A static priced item used when doing a job. Selectable when creating a quote. '))
    price = models.DecimalField(_('price'), max_digits=6, help_text="not sure if this is before or after VAT yet", decimal_places=2, default="0.00")

    def get_companies():
        return CompanyProfile.objects.filter(user=request.user)

    company = models.ForeignKey(CompanyProfile, default=get_companies) 

If its not obvious, im trying in the admin section to populate a dropdown with the available companies that belong to the user that is logged in.

my problem is that i dont know how to pass the request object to “get_companies”. anyone know of any examples.

Answer 1

You will have to do this overriding in your admin class that extends the ModelAdmin, not in your class that extends models.Model. Specifically, you need to override formfield_for_foreignkey.

From the docs:

class MyModelAdmin(admin.ModelAdmin):
    def formfield_for_foreignkey(self, db_field, request, **kwargs):
        if db_field.name == "car":
            kwargs["queryset"] = Car.objects.filter(owner=request.user)
    return super(MyModelAdmin, self).formfield_for_foreignkey(db_field, request, **kwargs)

For your case, it would seem like:

if db_field.name == "company":
    kwargs['queryset'] = request.user.company_set.all()