extended photologue photo to add an owner field

photologue_custom/models.py

@@ -1,6 +1,8 @@
 from django.db import models
+from django.conf import settings
 from taggit.managers import TaggableManager
-from photologue.models import Gallery
+from photologue.models import Gallery, Photo
+from django.utils.translation import gettext_lazy as _
 
 
 class GalleryExtended(models.Model):
@@ -15,8 +17,31 @@ class GalleryExtended(models.Model):
     tags = TaggableManager(blank=True)
 
     class Meta:
-        verbose_name = u'Extra fields'
-        verbose_name_plural = u'Extra fields'
+        verbose_name = 'Extra fields'
+        verbose_name_plural = 'Extra fields'
 
     def __str__(self):
         return self.gallery.title
+
+
+class PhotoExtended(models.Model):
+    # Extend Photologue Photo model.
+    photo = models.OneToOneField(
+        Photo,
+        related_name='extented',
+        on_delete='cascade'
+    )
+
+    # Add a owner field to PhotoExtended
+    owner = models.ForeignKey(
+        settings.AUTH_USER_MODEL,
+        on_delete=models.CASCADE,
+        verbose_name=_("owner"),
+    )
+
+    class Meta:
+        verbose_name = 'Extra fields'
+        verbose_name_plural = 'Extra fields'
+
+    def __str__(self):
+        return str(self.photo)