mirror of
https://github.com/jazzband/django-categories.git
synced 2026-03-16 22:30:24 +00:00
61 lines
2.3 KiB
Python
61 lines
2.3 KiB
Python
"""
|
|
These functions handle the adding of fields to other models
|
|
"""
|
|
from django.db.models import FieldDoesNotExist
|
|
import fields
|
|
from settings import FIELD_REGISTRY, MODEL_REGISTRY
|
|
|
|
|
|
def register_m2m(model, field_name='categories', extra_params={}):
|
|
return _register(model, field_name, extra_params, fields.CategoryM2MField)
|
|
|
|
|
|
def register_fk(model, field_name='category', extra_params={}):
|
|
return _register(model, field_name, extra_params, fields.CategoryFKField)
|
|
|
|
|
|
def _register(model, field_name, extra_params={}, field=fields.CategoryFKField):
|
|
app_label = model._meta.app_label
|
|
registry_name = ".".join((app_label, model.__name__, field_name)).lower()
|
|
|
|
if registry_name in FIELD_REGISTRY:
|
|
return # raise AlreadyRegistered
|
|
opts = model._meta
|
|
try:
|
|
opts.get_field(field_name)
|
|
except FieldDoesNotExist:
|
|
if app_label not in MODEL_REGISTRY:
|
|
MODEL_REGISTRY[app_label] = []
|
|
if model not in MODEL_REGISTRY[app_label]:
|
|
MODEL_REGISTRY[app_label].append(model)
|
|
FIELD_REGISTRY[registry_name] = field(**extra_params)
|
|
FIELD_REGISTRY[registry_name].contribute_to_class(model, field_name)
|
|
|
|
|
|
def _process_registry(registry, call_func):
|
|
"""
|
|
Given a dictionary, and a registration function, process the registry
|
|
"""
|
|
from django.core.exceptions import ImproperlyConfigured
|
|
from django.db.models.loading import get_model
|
|
|
|
for key, value in registry.items():
|
|
model = get_model(*key.split('.'))
|
|
if model is None:
|
|
raise ImproperlyConfigured('%s is not a model' % key)
|
|
if isinstance(value, (tuple, list)):
|
|
for item in value:
|
|
if isinstance(item, basestring):
|
|
call_func(model, item)
|
|
elif isinstance(item, dict):
|
|
field_name = item.pop('name')
|
|
call_func(model, field_name, extra_params=item)
|
|
else:
|
|
raise ImproperlyConfigured("CATEGORY_SETTINGS doesn't recognize the value of %s" % key)
|
|
elif isinstance(value, basestring):
|
|
call_func(model, value)
|
|
elif isinstance(value, dict):
|
|
field_name = value.pop('name')
|
|
call_func(model, field_name, extra_params=value)
|
|
else:
|
|
raise ImproperlyConfigured("CATEGORY_SETTINGS doesn't recognize the value of %s" % key)
|