combo/combo/apps/gallery/models.py

67 lines
2.4 KiB
Python

# combo - content management system
# Copyright (C) 2015 Entr'ouvert
#
# This program is free software: you can redistribute it and/or modify it
# under the terms of the GNU Affero General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import json
from django import template
from django.db import models
from django.core import serializers
from django.forms import models as model_forms
from django.utils.translation import ugettext_lazy as _
from combo.data.models import CellBase
from combo.data.library import register_cell_class
@register_cell_class
class GalleryCell(CellBase):
title = models.CharField(_('Title'), max_length=50, blank=True, null=True)
template_name = 'combo/gallerycell.html'
manager_form_template = 'combo/gallery_manager.html'
class Meta:
verbose_name = _('Gallery')
def get_additional_label(self):
if self.title:
return self.title
return ''
def export_subobjects(self):
return {'images': [x.get_as_serialized_object() for x in self.image_set.all()]}
def import_subobjects(self, cell_json):
for image in cell_json['images']:
image['fields']['gallery_id'] = self.id
for image in serializers.deserialize('json', json.dumps(cell_json['images'])):
image.save()
class Image(models.Model):
gallery = models.ForeignKey(GalleryCell, verbose_name=_('Gallery'))
image = models.ImageField(_('Image'),
upload_to='uploads/gallery/%Y/%m/')
order = models.PositiveIntegerField()
title = models.CharField(_('Title'), max_length=50, blank=True)
class Meta:
ordering = ['order']
def get_as_serialized_object(self):
serialized_image = json.loads(serializers.serialize('json', [self], use_natural_foreign_keys=True, use_natural_primary_keys=True))[0]
del serialized_image['fields']['gallery']
del serialized_image['pk']
return serialized_image