[Code Review] Script en Python para organizar mi carpeta de descargas

Punk

¡Buenas!

He creado un pequeño script en Python para poner orden en mi caótica carpeta de descargas. La idea es que mueva los archivos a subcarpetas (Imágenes, Documentos, Vídeos, etc.) según su extensión.

Funciona, pero como estoy aprendiendo Python, me gustaría saber si hay formas de hacerlo más "pythónico" o más eficiente. ¿Alguna sugerencia?

import os
import shutil

downloads_path = "/home/user/Downloads"
dest_paths = {
    "Images": [".jpg", ".jpeg", ".png", ".gif"],
    "Documents": [".pdf", ".docx", ".txt"],
    "Videos": [".mp4", ".mov", ".mkv"]
}

for filename in os.listdir(downloads_path):
    file_path = os.path.join(downloads_path, filename)
    if os.path.isfile(file_path):
        extension = os.path.splitext(filename)[1]
        for folder, extensions in dest_paths.items():
            if extension in extensions:
                dest_folder = os.path.join(downloads_path, folder)
                if not os.path.exists(dest_folder):
                    os.makedirs(dest_folder)
                shutil.move(file_path, dest_folder)
                break

StreetArt

El script es un gran comienzo. Una mejora "pythónica" sería usar el módulo pathlib en lugar de os. Hace el manejo de rutas mucho más limpio y legible.

Por ejemplo, en lugar de os.path.join(downloads_path, filename), podrías hacer downloads_path / filename. Y para crear directorios, dest_folder.mkdir(parents=True, exist_ok=True) reemplaza tu if not os.path.exists... de forma más elegante.

Quilting

Buen punto el de pathlib. Otra optimización sería invertir tu diccionario dest_paths para tener una búsqueda más directa. En lugar de iterar por las carpetas y luego por las extensiones, puedes crear un mapa de extensión -> carpeta.

# Al principio del script
extension_map = {ext: folder for folder, exts in dest_paths.items() for ext in exts}

# Dentro del bucle
if extension in extension_map:
    folder_name = extension_map[extension]
    # ... mover el archivo

Para un diccionario tan pequeño no notarás la diferencia en velocidad, pero es un patrón más eficiente si la lista crece.

Punk

¡Qué buenos consejos! No conocía pathlib y se ve mucho más limpio. Y la idea de invertir el diccionario es genial, muy elegante.

Voy a refactorizar el script con estas dos sugerencias. ¡Muchísimas gracias por el feedback, es justo lo que necesitaba para mejorar!