qrurls/models.py
changeset 32 82eae853e240
parent 30 d0f56303d8de
child 33 b714115da724
--- a/qrurls/models.py	Wed Aug 14 23:59:57 2013 +0300
+++ b/qrurls/models.py	Thu Aug 15 00:17:01 2013 +0300
@@ -72,13 +72,19 @@
         verbose_name_plural = u"URL Images"
         ordering = ['uploaded']
 
+    def get_absolute_url (self) :
+        return self.image.url
+
     def __unicode__ (self) :
         return self.image.name
 
 class URLItem(models.Model):
     shorturl = models.ForeignKey(URL)
     published = models.DateTimeField()
-    url = models.URLField()
+
+    # either-or
+    url = models.URLField(blank=True) # populated from image
+    image = models.ForeignKey(URLImage, null=True, blank=True)
     
     class Meta:
         verbose_name = u"URL Item"
@@ -120,3 +126,10 @@
                 published_state=self.published_state(),
                 url=self.get_absolute_url(),
         )
+
+    def save (self, *args, **kwargs) :
+        if self.image :
+            self.url = self.image.get_absolute_url()
+
+        super(URLItem, self).save(*args, **kwargs)
+