[Fix #126689795] Fix share

app/app/urls.py

@@ -24,6 +24,7 @@
     url(r'^$', views.index),
     url('', include('social.apps.django_app.urls', namespace='social')),
     url(r'^logout/$', views.logout_view),
+    url(r'^download/$', views.download),
     url(r'^api/v1/', include('photos.api.urls', namespace="api")),
 ]
 

app/photos/api/image_edit.py

@@ -1,3 +1,4 @@
+from django.http import HttpResponse
 from PIL import (
     Image, ImageFilter,
     ImageOps, ImageStat,
@@ -165,3 +166,9 @@ def save(self):
         if not os.path.exists(edit_path):
             os.makedirs(edit_path)
         self.output.save(path, format=self.image_format)
+
+    def download(self, title):
+        response = HttpResponse(content_type="image/jpg")
+        self.output.save(response, 'JPEG')
+        response['Content-Disposition'] = 'attachment; filename="{}.jpg"'.format(title)
+        return response

app/photos/views.py

@@ -2,6 +2,9 @@
 from django.http import HttpResponse
 from django.shortcuts import render
 from django.shortcuts import redirect
+import urllib
+from photos.models import Photo
+from photos.api.image_edit import *
 
 # Create your views here.
 
@@ -18,3 +21,12 @@ def index(request):
 def logout_view(request):
     logout(request)
     return redirect('/')
+
+
+def download(request):
+    id = request.GET.get('image')
+    photo = Photo.objects.get(id=id)
+    if photo:
+        image = ImageEdit(photo.image.path.replace('main','edited'))
+        return image.download(photo.title)
+    return redirect('/')