Add: Image Compression

requirements.txt

@@ -29,6 +29,7 @@ nodeenv==1.3.3
 oauthlib==3.0.1
 pre-commit==1.16.1
 pyasn1==0.4.5
+Pillow==7.0.0
 pyasn1-modules==0.2.5
 PyJWT==1.7.1
 python3-openid==3.1.0

uploader/serializers.py

@@ -12,3 +12,4 @@ class FileSerializer(serializers.Serializer):
 class GooglePhotosUploadInputSerializer(serializers.Serializer):
     fileList = FileSerializer(many=True)
     token = fields.CharField()
+    compress = fields.BooleanField()

uploader/templates/upload.html

@@ -192,7 +192,7 @@ <h5 class="card-title text-center">
   var oauthToken;
   var fileStagingTable;
 
-  
+
   function TDate() 
   {
     var dc = document.getElementById(`wpDateCreated`).value;
@@ -282,6 +282,7 @@ <h5 class="card-title text-center">
       fileStagingTable = new Tabulator("#picked-files-table", {
         data: pickedFiles,
         layout: "fitColumns",
+
         responsiveLayout:"collapse",
         columns: [
           {
@@ -292,6 +293,8 @@ <h5 class="card-title text-center">
                 cell.getValue()
               )}'>`;
             },
+
+
             width: 80,
             responsive:0,
           },
@@ -329,14 +332,17 @@ <h5 class="card-title text-center">
             responsive:1, 
           },            
           {
+
             formatter: "buttonCross",
             width: 40,
             align: "center",
             headerSort: false,
             cellClick: function(e, cell) {
               cell.getRow().delete();
+
             },
             responsive:0,
+
           }
         ]
       });
@@ -351,7 +357,8 @@ <h5 class="card-title text-center">
       .removeClass("fa-caret-right")
       .addClass("fa-spinner fa-spin");
     var fileData = {
-      fileList: fileStagingTable.getData(),
+      fileList: fileStagingTable.getData(),    
+      compress: $("#imageCompress").is(":checked"),
       token: oauthToken
     };
 
@@ -369,6 +376,7 @@ <h5 class="card-title text-center">
       fileData["fileList"][x]["license"] = sel.value;
     }
 
+
     function getCookie(name) {
       var cookieValue = null;
       if (document.cookie && document.cookie !== "") {

uploader/utils.py

@@ -0,0 +1,10 @@
+from PIL import Image
+
+
+def resize_image(img):
+
+    basewidth = 512
+    wpercent = basewidth / float(img.size[0])
+    hsize = int((float(img.size[1]) * float(wpercent)))
+    img = img.resize((basewidth, hsize), Image.ANTIALIAS)
+    return img

uploader/views.py

@@ -11,6 +11,8 @@
 from uploader.serializers import GooglePhotosUploadInputSerializer
 from uploader.wiki_uploader import WikiUploader
 
+from uploader.utils import resize_image
+from PIL import Image
 from uploader.models import FileUpload
 
 
@@ -55,6 +57,7 @@ def post(self, request, format=None, *args, **kwargs):
         drive_service = self.get_google_drive_service(
             access_token=validated_data.get("token", None)
         )
+        compress = validated_data.get("compress", False)
         social_auth = self.request.user.social_auth.get(
             provider="mediawiki"
         ).extra_data["access_token"]
@@ -80,10 +83,21 @@ def post(self, request, format=None, *args, **kwargs):
             while done is False:
                 download_status, done = downloader.next_chunk()
 
-            uploaded, image_info = wiki_uploader.upload_file(
+            image = Image.open(fh)
+
+            if compress:
+                image = resize_image(image)
+
+            fh_resized = io.BytesIO()
+            image.save(
+                fh_resized,
+                Image.registered_extensions()["." + file["name"].split(".")[-1]],
+            )
 
+            uploaded, image_info = wiki_uploader.upload_file(
+                file_stream=fh_resized,
+                description=file["description"],
                 file_name=file["name"],
-                file_stream=fh,
                 date_created=file["date_created"],
                 description=file["description"],
                 license=file["license"],