from django.db import models
from taggit.managers import TaggableManager
from photologue.models import Gallery


class GalleryExtended(models.Model):
    # Extend Photologue Gallery model.
    gallery = models.OneToOneField(
        Gallery,
        related_name='extended',
        on_delete='cascade',
    )

    # Add tags
    tags = TaggableManager(blank=True)

    class Meta:
        verbose_name = u'Extra fields'
        verbose_name_plural = u'Extra fields'

    def __str__(self):
        return self.gallery.title