diff --git a/categories/admin.py b/categories/admin.py index ea9623f..94fa4c3 100644 --- a/categories/admin.py +++ b/categories/admin.py @@ -81,7 +81,7 @@ class CategoryAdmin(TreeEditor, admin.ModelAdmin): prepopulated_fields = {'slug': ('name',)} fieldsets = ( (None, { - 'fields': ('parent', 'name') + 'fields': ('parent', 'name', 'thumbnail') }), ('Meta Data', { 'fields': ('alternate_title', 'description', 'meta_keywords', 'meta_extra'), @@ -117,4 +117,4 @@ for model,modeladmin in admin.site._registry.items(): 'fieldsets': fieldsets + (('Categories',{ 'fields': fields }),) - })) \ No newline at end of file + })) diff --git a/categories/migration/add_thumbnail.sql b/categories/migration/add_thumbnail.sql new file mode 100644 index 0000000..a5aaa85 --- /dev/null +++ b/categories/migration/add_thumbnail.sql @@ -0,0 +1,3 @@ +BEGIN; +ALTER TABLE "categories_category" ADD COLUMN "thumbnail" varchar(100); +COMMIT; diff --git a/categories/models.py b/categories/models.py index 0dd975e..a1475c4 100644 --- a/categories/models.py +++ b/categories/models.py @@ -9,7 +9,7 @@ from django.utils.translation import ugettext as _ from mptt.models import MPTTModel -from settings import RELATION_MODELS, RELATIONS +from settings import RELATION_MODELS, RELATIONS, THUMBNAIL_UPLOAD_PATH class Category(MPTTModel): parent = models.ForeignKey('self', @@ -19,6 +19,7 @@ class Category(MPTTModel): help_text="Leave this blank for an Category Tree", verbose_name='Parent') name = models.CharField(max_length=100) + thumbnail = models.ImageField(upload_to=THUMBNAIL_UPLOAD_PATH, null=True, blank=True) order = models.IntegerField(blank=True, null=True) slug = models.SlugField() alternate_title = models.CharField( diff --git a/categories/settings.py b/categories/settings.py index 47cb55b..dbda8e6 100644 --- a/categories/settings.py +++ b/categories/settings.py @@ -8,3 +8,6 @@ from django.db.models import Q DEFAULT_RELATION_MODELS = [] RELATION_MODELS = getattr(settings, 'CATEGORIES_RELATION_MODELS', DEFAULT_RELATION_MODELS) or [] RELATIONS = [Q(app_label=al, model=m) for al, m in [x.split('.') for x in RELATION_MODELS]] + +# For assigning a thumbnail to a category +THUMBNAIL_UPLOAD_PATH = getattr(settings, 'CATEGORIES_THUMBNAIL_UPLOAD_PATH', 'uploads/categories/thumbnails')