diff --git a/categories/fields.py b/categories/fields.py index 21ba41a..f2cdb00 100644 --- a/categories/fields.py +++ b/categories/fields.py @@ -1,12 +1,18 @@ from django.db.models import ForeignKey, ManyToManyField +from categories.models import Category + + + class CategoryM2MField(ManyToManyField): def __init__(self, **kwargs): - from categories.models import Category + if 'to' in kwargs: + kwargs.pop('to') super(CategoryM2MField, self).__init__(to=Category, **kwargs) class CategoryFKField(ForeignKey): def __init__(self, **kwargs): - from categories.models import Category - super(CategoryFKField, self).__init__(to=Category, **kwargs) \ No newline at end of file + if 'to' in kwargs: + kwargs.pop('to') + super(CategoryFKField, self).__init__(to=Category, **kwargs)