向 Django 用户表单添加验证

Question

我想在 Django/Mezzanine 中自定义用户注册表单以只允许某些电子邮件地址，所以我尝试如下进行猴子补丁：

# Monkey-patch Mezzanine's user email address check to allow only
# email addresses at @example.com.
from django.forms import ValidationError
from django.utils.translation import ugettext
from mezzanine.accounts.forms import ProfileForm
from copy import deepcopy
original_clean_email = deepcopy(ProfileForm.clean_email)
def clean_email(self):
    email = self.cleaned_data.get("email")
    if not email.endswith('@example.com'):
        raise ValidationError(
            ugettext("Please enter a valid example.com email address"))
    return original_clean_email(self)
ProfileForm.clean_email = clean_email

此代码添加在我的models.py 之一的顶部。

但是，当我运行服务器时，我会感到害怕

django.core.exceptions.AppRegistryNotReady: Models aren't loaded yet.

如果我添加

import django
django.setup()

然后python manage.py runserver 一直挂到我^C。

我应该怎么做才能添加这个功能？

Answer 1

为您的一个应用程序创建一个文件myapp/apps.py（我在这里使用myapp），并定义一个应用程序配置类，在ready() 方法中执行monkeypatching。

from django.apps import AppConfig

class MyAppConfig(AppConfig):
    name = 'myapp'

    def ready(self):
        # do the imports and define clean_email here
        ProfileForm.clean_email = clean_email

然后在您的INSTALLED_APPS 设置中使用'myapp.apps.MyAppConfig' 而不是'myapp'。

INSTALLED_APPS = [
    ...
    'myapp.apps.MyAppConfig',
    ...
]

您可能需要将 Mezzanine 放在应用配置之上才能使其正常工作。