Имитация игры в кости Python

Я немного застрял на домашнем задании, имитирующем игру в кости. Вопрос состоит в том, чтобы создать функцию, которая возвращает случайное целочисленное значение от 1 до 6, и создать основную программу, которая спрашивает пользователя, сколько кубиков бросить (должно быть ограничено 5 кубиками), и вызывает функцию для печати сгенерированного ценности. Пока у меня есть это:

import random

def dice(number_of_dice):
    for i in range(0,number_of_dice):
        return random.randint(1,6)


number_of_dice = input("How many dice would you like to roll? ")
while number_of_dice >5:
    number_of_dice = input("You may only have a limit of 5 dice, enter a number under 5. " )
print dice(number_of_dice) 

При запуске программы она возвращает только одно случайное целочисленное значение независимо от значения & quot; number_of_dice & quot; вход есть. Что именно здесь не так?

Любая помощь приветствуется, спасибо.

Ответы на вопрос(3)

Ваш ответ на вопрос