Use singleton registry to import register_fk and register_m2m since they are members on Registry class.

This commit is contained in:
Orestes Sanchez 2015-08-12 16:53:47 +02:00
parent e40b240e24
commit c7344f6b60
2 changed files with 4 additions and 5 deletions

View file

@ -23,10 +23,9 @@ default_app_config = 'categories.apps.CategoriesConfig'
def register():
from categories import settings
from categories.registration import (_process_registry, register_fk,
register_m2m)
_process_registry(settings.FK_REGISTRY, register_fk)
_process_registry(settings.M2M_REGISTRY, register_m2m)
from categories.registration import (_process_registry, registry)
_process_registry(settings.FK_REGISTRY, registry.register_fk)
_process_registry(settings.M2M_REGISTRY, registry.register_m2m)
try:
register()
except Exception as e:

View file

@ -51,7 +51,7 @@ class Categorym2mTest(TestCase):
M2M_REGISTRY = {
'flatpages.flatpage': 'categories'
}
_process_registry(M2M_REGISTRY, register_m2m)
_process_registry(M2M_REGISTRY, registry.register_m2m)
from django.contrib.flatpages.models import FlatPage
self.assertTrue('category' in FlatPage()._meta.get_all_field_names())