Generar pin de 4 dígitos en python

Tenía que generar un pequeño número para una aplicación, aquí el snippet por si algún día lo necesitas, con longitud personalizada.

from random import randint

def generate_pin(length=4):
  upper_limit = 10 ** length
  pin = randint(0, upper_limit-1)
  return "{:0{length}d}".format(pin, length=length)