Primera versión del compresor web

Author: mikexarau

.DS_Store

@@ -0,0 +1,73 @@
+from flask import Flask, request, send_from_directory, render_template
+from werkzeug.utils import secure_filename
+from PIL import Image
+from moviepy.editor import VideoFileClip
+import os
+from comprimirpdf import compress
+
+app = Flask(__name__)
+
+UPLOAD_FOLDER = 'uploads'
+DOWNLOAD_FOLDER = 'downloads'
+
+if not os.path.exists(UPLOAD_FOLDER):
+    os.makedirs(UPLOAD_FOLDER)
+
+if not os.path.exists(DOWNLOAD_FOLDER):
+    os.makedirs(DOWNLOAD_FOLDER)
+
+def compress_file(file_path, scale, quality):
+    if file_path:
+        file_name, file_extension = os.path.splitext(os.path.basename(file_path))
+        save_path = os.path.join(DOWNLOAD_FOLDER, f"{file_name}-compressed{file_extension}")
+                
+        if file_extension.lower() in [".jpg", ".jpeg", ".png"]:
+            image = Image.open(file_path)
+            width, height = image.size
+            new_width = int(width * scale / 100)
+            new_height = int(height * scale / 100)
+            new_image = image.resize((new_width, new_height))
+            new_image.save(save_path, optimize=True, quality=quality)
+        elif file_extension.lower() in [".mp4", ".avi", ".mov"]:
+            video = VideoFileClip(file_path)
+            new_width = int(video.w * scale / 100)
+            new_height = int(video.h * scale / 100)
+            new_video = video.resize(width=new_width, height=new_height)
+            if file_extension.lower() == ".mov":
+                save_path = f"{file_name}-compressed.mp4"
+            new_video.write_videofile(save_path)
+        elif file_extension.lower() == ".pdf":
+            compress_quality = quality
+            compress_quality //= 25
+            compress(file_path, save_path, compress_quality)
+
+        return save_path
+
+@app.route('/', methods=['GET', 'POST'])
+def home():
+    if request.method == 'POST':
+        file = request.files['file']
+        filename = secure_filename(file.filename)
+        filepath = os.path.join(UPLOAD_FOLDER, filename)
+        file.save(filepath)
+        
+        scale = request.form.get('scale', type=int)
+        if scale is None:
+            scale = 50
+    
+        quality = request.form.get('quality', type=int)
+        if quality is None:
+            quality = 85
+
+        output_file = compress_file(filepath, scale, quality)
+        return os.path.basename(output_file)
+
+    return render_template('index.html')
+
+@app.route('/downloads/<filename>', methods=['GET'])
+def download_file(filename):
+    return send_from_directory(DOWNLOAD_FOLDER, filename, as_attachment=True)
+
+if __name__ == '__main__':
+    app.run(debug=True)
+

__pycache__/comprimirpdf.cpython-311.pyc

@@ -0,0 +1,60 @@
+#!/usr/bin/env python3
+"""
+Script de Python que utiliza ghoscript para comprimir ficheros PDF.
+Niveles de compresión:
+    0: defecto
+    1: prepress
+    2: printer
+    3: ebook
+    4: screen
+Dependencias: Ghostscript.
+En MacOSX aplica la siguiente línea `brew install ghostscript`.
+"""
+import argparse
+import subprocess
+import os.path
+import sys
+import shutil
+ 
+def compress(input_file_path, output_file_path, power=0):
+    """Función para comprimir PDF via Ghostscript """
+    quality = {
+        0: '/default',
+        1: '/prepress',
+        2: '/printer',
+        3: '/ebook',
+        4: '/screen'
+    }
+ 
+    # Comprovamos si existe el fichero
+    if not os.path.isfile(input_file_path):
+        print("Error: invalid path for input PDF file")
+        sys.exit(1)
+ 
+    #Comprobamos si es un pdf
+    if input_file_path.split('.')[-1].lower() != 'pdf':
+        print("Error: input file is not a PDF")
+        sys.exit(1)
+ 
+    gs = get_ghostscript_path()
+    print("Compress PDF...")
+    initial_size = os.path.getsize(input_file_path)
+    subprocess.call([gs, '-sDEVICE=pdfwrite', '-dCompatibilityLevel=1.4',
+                    '-dPDFSETTINGS={}'.format(quality[power]),
+                    '-dNOPAUSE', '-dQUIET', '-dBATCH',
+                    '-sOutputFile={}'.format(output_file_path),
+                     input_file_path]
+    )
+    final_size = os.path.getsize(output_file_path)
+    ratio = 1 - (final_size / initial_size)
+    print("Compression by {0:.0%}.".format(ratio))
+    print("Final file size is {0:.1f}MB".format(final_size / 1000000))
+    print("Done.")
+ 
+ 
+def get_ghostscript_path():
+    gs_names = ['gs', 'gswin32', 'gswin64']
+    for name in gs_names:
+        if shutil.which(name):
+            return shutil.which(name)
+    raise FileNotFoundError(f'No GhostScript executable was found on path ({"/".join(gs_names)})')