Question

我正在为公司构建一个通知系统，管理员用户可以在其中创建项目并向其中添加用户。 Project模型具有9个属性，但是我只想在创建Project时显示3或4个字段，但是在更新现有Project时全部显示它们。

此更改仅需要反映在Django管理站点上，因此我用自己的ProjectForm扩展了ProjectAdmin，在其中扩展了 init 方法以检查它是否为新实例，并且如果是这样，请删除某些字段。

# models.py
class Project(models.Model):
    project_number = models.IntegerField()
    name = models.CharField(max_length=100)
    permit = models.CharField(max_length=100, blank=True, default='')
    is_active = models.BooleanField(default=True)
    users = models.ManyToManyField(CustomUser, blank=True, related_name='project_users')
    # add a default

    levels = models.ManyToManyField('Level', blank=True, related_name='project_levels')

    total_contract_hours = models.IntegerField(default=0, blank=True, verbose_name='Total Design Hours')
    hours_used = models.IntegerField(default=0, blank=True, verbose_name='Total Design Hours Used')

    notes = models.ManyToManyField('notes.ProjectNote', related_name='core_project_notes', blank=True)

    history = HistoricalRecords()

    def __str__(self):
        ret_str = "{} {}".format(self.project_number, self.name)
        if self.permit:
            ret_str += "  |  Permit: {}".format(self.permit)
        return ret_str



# admin.py
class ProjectForm(forms.ModelForm):

    def __init__(self, *args, **kwargs):
        super(ProjectForm, self).__init__(*args, **kwargs)
        attrs = {'class': 'form-control', 'required': True}
        if self.instance and self.instance.pk is None:
            # creating project

            exclude = ['is_active', 'users', 'levels', 'hours_used', 'notes']
            for field in exclude:
                try:
                    del self.fields[field]
                except ValueError:
                    print('{} does not exist'.format(field))

        for field in self.fields.values():
            field.widget.attrs = attrs

    class Meta:
        model = Project
        fields = ['project_number', 'name', 'total_contract_hours']


class ProjectAdmin(admin.ModelAdmin):
    form = ProjectForm
    fields = ['project_number', 'name', 'permit', 'is_active', 'users', 'levels', 'total_contract_hours', 'hours_used', 'notes']

正如我所说，我只希望创建时具有基本的Project字段，但是在更新现有Project时显示所有归因于的字段。通过这些更改，我现在得到一个KeyError：

KeyError：“在'ProjectForm'中找不到键'is_active'。选择包括：名称，许可证，项目编号，总合同小时数。”

但是，当我打印可用字段时，它将返回一个OrderedDict，其中所有模型属性都作为键。我究竟做错了什么？谢谢！

Answer 1

我知道了，该字段必须在Meta中列出，然后您才将其设置为隐藏字段。

ProjectForm（forms.ModelForm）类：

def __init__(self, *args, **kwargs):
    super(ProjectForm, self).__init__(*args, **kwargs)
    print("Adding project")

    if not self.instance or self.instance.pk is None:
        for name, field in self.fields.items():
            if name in ['design_manager', ]:
                field.widget = forms.HiddenInput()

class Meta:
    model = Project
    fields = ['project_number', 'name', 'design_manager', 'total_contract_hours']

类ProjectAdmin（admin.ModelAdmin）：表格= ProjectForm

def save_model(self, request, obj, form, change):
    obj.design_manager = request.user
    super().save_model(request, obj, form, change)

仅在创建时排除Django模型的字段

1 个答案: