"""Utils app method.""" import random def generate_code(digits=6, string_output=True): """Generate random int.""" max_value = 10 ** digits - 1 min_value = 10 ** (digits - 1) value = random.randint(min_value, max_value) return str(value) if string_output else value