and some more fixes

[redakcja.git] / apps / cover / models.py

diff --git a/apps/cover/models.py b/apps/cover/models.py
index 87a70a8..6900e78 100644 (file)
--- a/apps/cover/models.py
+++ b/apps/cover/models.py
@@ -4,7 +4,6 @@
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 import re
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 import re

-from urllib2 import urlopen

 from urlparse import urljoin
 from django.conf import settings
 from django.core.files.base import ContentFile
 from urlparse import urljoin
 from django.conf import settings
 from django.core.files.base import ContentFile


@@ -12,6 +11,8 @@ from django.db import models
 from django.db.models.signals import post_save
 from django.dispatch import receiver
 from django.utils.translation import ugettext_lazy as _
 from django.db.models.signals import post_save
 from django.dispatch import receiver
 from django.utils.translation import ugettext_lazy as _

+from django.contrib.sites.models import Site
+from cover.utils import URLOpener

 
 
 class Image(models.Model):
 
 
 class Image(models.Model):


@@ -34,11 +35,14 @@ class Image(models.Model):
     def get_absolute_url(self):
         return ('cover_image', [self.id])
 
     def get_absolute_url(self):
         return ('cover_image', [self.id])
 

+    def get_full_url(self):
+        return "http://%s%s" % (Site.objects.get_current().domain, self.get_absolute_url())
+

 
 @receiver(post_save, sender=Image)
 def download_image(sender, instance, **kwargs):
     if instance.pk and not instance.file:
 
 @receiver(post_save, sender=Image)
 def download_image(sender, instance, **kwargs):
     if instance.pk and not instance.file:

-        t = urlopen(instance.download_url).read()
+        t = URLOpener().open(instance.download_url).read()

         instance.file.save("%d.jpg" % instance.pk, ContentFile(t))
         
         
\ No newline at end of file
         instance.file.save("%d.jpg" % instance.pk, ContentFile(t))
         
         
\ No newline at end of file