Django 管理表单 - 如何动态更改选择选项? [英] Django admin form - how to change select options dynamically?

问题描述

我有 2 个模型:

class City(models.Model):
    name = models.CharField(max_length=50)
    slug = models.SlugField(max_length=50)


class CityNews(models.Model):
    title = models.CharField(max_length=100)
    slug = models.SlugField(max_length=100)
    add_date = models.DateTimeField(auto_now=False, auto_now_add=True, editable=False)
    content = models.TextField()
    city = models.ForeignKey(City)

我的每个用户都与 1 个城市建立了联系.我希望他只向他所连接的城市添加新闻.但是超级管理员必须有可能向每个城市添加新闻.如何更改 CityNews 中的城市"字段，使其仅显示用户连接的城市?我可以编写自定义 ModelForm，但如何检查 user_city 并更改其查询集?

My each user has been connected with 1 city. I want him to add news only to the city he id connected with. But superadmin must have possibility to add news to each city. How can I change 'city' field in CityNews, that they show only the city that user is connected with? I can write custom ModelForm but how can I check user_city there and change its queryset?

推荐答案

一个明显的方法是利用 formfield_for_foreignkey() 方法.

One obvious way of doing this is to utilize the formfield_for_foreignkey() method on the ModelAdmin.

所以如果你的 models.py 看起来像这样:

So if your models.py looks like this:

from django.db import models
from django.contrib.auth.models import User

class City(models.Model):
    name = models.CharField(max_length=50)

    def __unicode__(self):
        return self.name

class CityNews(models.Model):
    added_by = models.ForeignKey(User)
    city = models.ForeignKey(City)
    title = models.CharField(max_length=100)
    content = models.TextField()

class UserExtra(models.Model):
    user = models.ForeignKey(User)
    city = models.ForeignKey(City)

那么您的 admin.py 可能如下所示:

Then your admin.py could look like this:

from django.contrib import admin
from formtesting.models import City, CityNews, UserExtra
from django.forms.models import ModelChoiceField
from django.contrib.auth.models import User

class CityAdmin(admin.ModelAdmin):
    pass

admin.site.register(City, CityAdmin)

class CityNewsAdmin(admin.ModelAdmin):
    def formfield_for_foreignkey(self, db_field, request, **kwargs):
        if db_field.name == "city":
            if request.user.is_superuser:
                queryset = City.objects.all()
            else:
                queryset = City.objects.filter(userextra__user=request.user)
            return ModelChoiceField(queryset, initial=request.user)
        elif db_field.name == "added_by":
            if request.user.is_superuser:
                queryset = User.objects.all()
            else:
                queryset = User.objects.filter(id=request.user.id)
            return ModelChoiceField(queryset, initial=request.user)
        else:
            return super(CityNewsAdmin, self).formfield_for_foreignkey(db_field, 
                                                              request, **kwargs)

admin.site.register(CityNews, CityNewsAdmin)

class UserExtraAdmin(admin.ModelAdmin):
    pass

admin.site.register(UserExtra, UserExtraAdmin)

这篇关于Django 管理表单 - 如何动态更改选择选项?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！