Python. Crear una barra de progreso en la terminal

Cuando trabajamos con Python pintando resultados en la terminal, puede quedar un poco feo dejar el cursor esperando mientras procesamos ciertas acciones que van a tardar un tiempo en completarse. Esto es así aquí y en la China. El cursor parpadeando pone nervioso a todo el mundo.

Pero… ¿cuál podría ser la solución? Cierta librería, llamada tqdm, disponible para Python 3.x.

¿Qué es tqdm?

Es una librería de Python que nos va a permitir pintar barras de progreso directamente en la terminal, iterando en bucles for / while y de la que nos vamos a desentender prácticamente desde el primer minuto.

El resultado final sería tal que así

Una simple barra de progreso, en el color por defecto de la terminal, que se irá rellenando progresivamente según iteremos dentro de nuestro bucle.

¿Y cómo se hace eso?

Lo primero, será instalar tqdm. Para ello nos iremos a la terminal y escribiremos el siguiente comando:

pip install tqdm

Por supuesto, si en nuestro ordenador tenemos Python2 y Python3, deberemos reemplazar el comando pip por pip3.

Hecho esto, veamos un par de ejemplos, harto sencillos ambos dos. En el primer script, vamos a realizar una iteración hasta llegar a un número en concreto, por ejemplo el 100:

from tqdm import tqdm
from time import sleep

tareas = 100

for i in tqdm(range(tareas)):
    sleep(0.2)

Básicamente lo que estamos haciendo es importar nuestra librería, definir un valor hasta el que iterar y envolver el objeto sobre el que iteraremos con el método tqdm. Rápido y sencillo. La función sleep simplemente “duerme” la aplicación 0.2 segundos cada vez que entra en el for, para poder ver el progreso de nuestra barra tranquilamente. Por supuesto, no es necesario mantenerla y está ahí como mero ejemplo del código que podemos insertar.

Y, en caso de que lo que deseemos sea un lista, lo haremos del siguiente modo:

from tqdm import tqdm
from time import sleep

tareas = ["tarea1", "tarea2", "tarea3", "tarea4", "tarea5"]

for i in tqdm(tareas):
    sleep(0.2)

¿Cuál es la única diferencia? Que ya no usamos el range, sino que lo recorremos como si fuese una lista (¡que lo es!) normal pythonesca de toda la vida.

Y hecho esto, ya podemos trabajar felices de la vida sabiendo que nuestros resultados aparecerán por pantalla de una forma un poquito más visual que antes.

¡Salud y coding!