如何限制 django 数据库中的条目数？使用默认数据库（sqlite3）

Question

如何指定我的 django 数据库 (sqlite3) 中的条目数。例如我的模型是：

from django.db import models

class About(models.Model):
    firstname = models.CharField(max_length=500)
    lastname = models.CharField(max_length=500, blank=True)
    address = models.TextField()
    details = models.TextField()

    class Meta:
        verbose_name_plural = 'About'

    def __str__(self):
        return self.firstname

现在我想在我的数据库中存储只有 2 个成员的数据。从 django 管理面板在数据库中添加两个用户的数据后，如果我尝试存储第三个成员的数据，那么它应该显示超出数据库限制，您无法存储此数据，请先删除以前的数据。 是否可以从 django models.py 中限制条目。

Answer 1

您可以创建一个自定义 ModelAdmin 并将您的逻辑放在那里，因此，如果您为您的 About 模型所在的应用程序编辑您的 admin.py 如下：

from django.contrib import admin
from django.http import HttpResponseRedirect
from django.shortcuts import reverse
from django.contrib import messages

from .models import About

class AboutAdmin(admin.ModelAdmin):
    def add_view(self, request, form_url='', extra_context=None):
        if self.model.objects.count() >= 2:
            self.message_user(request, 'Only two entries can exist at once - please remove others first', messages.ERROR)
            return HttpResponseRedirect(reverse(f'admin:{self.model._meta.app_label}_about_changelist'))
        return super().add_view(request, form_url, extra_context)

admin.site.register(About, AboutAdmin)

这应该可以解决问题...

Answer 2

写一个用户定义的异常

class DbLimitException(BaseException):
      pass

覆盖About类的save方法

def save(self,*args,**kwargs):
    total_records = About.objects.count()
    if total_records >= 2:
       raise DbLimitException({"message": "Db limit reached please delete to add more data"})
    else:
       super().save(*args,**kwargs)

Answer 3

你可以做的一件事是，

view.py

about_count = About.objects.count()
if about_count == 2:
    return render(request, template_name="template.html", context={"error": "You cannot add more entries to the table, you need to delete the previous ones"}

模板.html

{% if error %}
   {{error}}
{% endif %}