从 Django 中的 ENUM 中选择

Question

下面是我的模型，

class Employee(models.Model):
    name = models.CharField(max_length=100)
    sex = models.PositiveIntegerField(choices=choices(Type))

枚举类是

class Type(Enum):
    Male = 0
    Female = 1

现在来到我们的主要模型

class Tournament(models.Model):
    player_name = models.ForeignKey(Employee)

我的管理员是

@admin.register(Tournament, site=admin.site)
class TournamentAdmin(admin.ModelAdmin):
    list_display = ('player_name')

要求是我只想在我的管理页面中的字段 player_name 中获取男性玩家。

如何从 ENUM 类中过滤以在 Django 管理字段中仅获取男性玩家姓名。 请帮我。谢谢！

Answer 1

我已经通过在管理员中覆盖 formfield_for_foreignkey 方法来做到这一点。如果您有更多需要应用过滤器的字段，只需添加更多 if 块并从相关模型中过滤（通过导入模型）。

如果存在 ForeignKey 关系，则需要在子类到父类的字段之间使用 __ 从子类到父类

@admin.register(Tournament, site=admin.site)
class TournamentAdmin(admin.ModelAdmin):
    list_display = ('player_name')

    def formfield_for_foreignkey(self,db_field,request,**kwargs):
        if db_field.name == 'player_name':
            kwargs['queryset'] = Employee.objects.filter(type=Type.Male.value)
        return super().formfield_for_foreignkey(db_field,request,**kwargs)

Answer 2

使用参考 Django admin.SimpleListFilter 。你可以试试这个：

from django.contrib import admin

class GenderListFilter(admin.SimpleListFilter):
    # Human-readable title which will be displayed in the
    # right admin sidebar just above the filter options.
    title = _('gender')

    # Parameter for the filter that will be used in the URL query.
    parameter_name = 'gender'

    def lookups(self, request, model_admin):
        """
        Returns a list of tuples. The first element in each
        tuple is the coded value for the option that will
        appear in the URL query. The second element is the
        human-readable name for the option that will appear
        in the right sidebar.
        """
       return (
        (0, _('Male')),
        (1, _('Female')),
        )

def queryset(self, request, queryset):
    """
    Returns the filtered queryset based on the value
    provided in the query string and retrievable via
    `self.value()`.
    """
    return queryset.filter(player_name__sex==self.value())

class TournamentAdmin(admin.ModelAdmin):
    list_filter = (GenderListFilter,)

希望对你有帮助。