Добавление проверки в Django формы User

Question

Я хотел бы настроить форму регистрации пользователя в Django/Мезонин разрешить только определенные адреса электронной почты, поэтому я попытался обезьяньего пластыря следующим образом:

# 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.

Однако, когда я runserver я получаю страшный

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

Если добавить

import django 
django.setup() 

тогда python manage.py runserver просто зависает, пока я ^C.

Что я должен делать, чтобы добавить эту функцию?

Answer 1

Создайте файл myapp/apps.py для одного из приложений (я имею использование myapp здесь), и определить класс конфигурации приложения, который делает monkeypatching в методе ready().

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 

Затем используйте 'myapp.apps.MyAppConfig' вместо 'myapp' в настройках INSTALLED_APPS.

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

Возможно, вам понадобится поставить Mezzanine над конфигурацией приложения, чтобы он работал.