django-categories/categories/fields.py

25 lines
776 B
Python
Raw Normal View History

from django.db.models import ForeignKey, ManyToManyField
2012-07-12 23:24:55 +00:00
class CategoryM2MField(ManyToManyField):
def __init__(self, **kwargs):
from .models import Category
if 'to' in kwargs:
kwargs.pop('to')
super(CategoryM2MField, self).__init__(to=Category, **kwargs)
2010-02-12 17:09:58 +00:00
class CategoryFKField(ForeignKey):
def __init__(self, **kwargs):
from .models import Category
if 'to' in kwargs:
kwargs.pop('to')
super(CategoryFKField, self).__init__(to=Category, **kwargs)
2011-05-08 00:33:00 +00:00
try:
from south.modelsinspector import add_introspection_rules
add_introspection_rules([], ["^categories\.fields\.CategoryFKField"])
add_introspection_rules([], ["^categories\.fields\.CategoryM2MField"])
except ImportError:
pass