diff --git a/categories/fields.py b/categories/fields.py new file mode 100644 index 0000000..fde87ca --- /dev/null +++ b/categories/fields.py @@ -0,0 +1,11 @@ +from django.db.models import ForeignKey, ManyToManyField +from categories.models import Category + +class CategoryM2MField(ManyToManyField): + def __init__(self, **kwargs): + super(CategoryM2MField, self).__init__(to=Category, **kwargs) + + +class CategoryFKField(ForeignKey): + def __init__(self, **kwargs): + super(CategoryFKField, self).__init__(to=Category, **kwargs) \ No newline at end of file