#!/usr/bin/python3
# Dice simulator. Project for http://redd.it/2aaeou
# author: wukaem
import random
DICE_SEGMENTS = {
0: ' .-------.',
1: ' | |',
2: ' | O |',
3: ' | O |',
4: ' | O |',
5: ' | O O |',
6: ' | O O O |',
7: " '-------'",
}
SIDE_REPR = (
(0, 1, 2, 1, 7),
(0, 3, 1, 4, 7),
(0, 4, 2, 3, 7),
(0, 5, 1, 5, 7),
(0, 5, 2, 5, 7),
(0, 6, 1, 6, 7),
)
def dice_img_lst(side):
img_lst = []
for segment_num in SIDE_REPR[side-1]:
img_lst.append(DICE_SEGMENTS[segment_num])
return img_lst
def roll_dice(num):
return [random.randint(1, 6) for _ in range(num)]
def dice_image(dice_lst):
dice_lst_img = []
for side in dice_lst:
dice_lst_img.append(dice_img_lst(side))
rows = []
for i in range(5):
rows.append(''.join(segment[i] for segment in dice_lst_img))
return '\n'.join(rows)
def ask_question1():
print('\nHow many dice do you want to roll? (1-5)')
while True:
ans = input('> ')
try:
ans = int(ans)
except ValueError:
print('\nGive me a number. Try again.')
continue
if ans in range(1, 6):
return ans
print('\nRoll at least 1 dice and no more than 5. Try again.')
def ask_question2(num):
question = ("Type 'q' to quit, 'c' to change number of dice,\n"
"or hit <Enter> to roll {} dice again.".format(num))
print(question)
new_num = 0
while not new_num:
ans = input('> ')
if ans == '':
new_num = num
elif ans.lower()[0] not in ('q', 'c'):
print('Wrong answer.\n')
print(question)
continue
elif ans.lower()[0] == 'q':
new_num = -1
else:
new_num = ask_question1()
return new_num
def main():
print('\n Welcome to DICE SIMULATOR')
no_dice = ask_question1()
while no_dice != -1:
print('\n Rolling {} dice:\n'.format(no_dice))
dice_list = roll_dice(no_dice)
print(dice_image(dice_list))
print('\n Total: {}\n'.format(sum(dice_list)))
no_dice = ask_question2(no_dice)
print('Goodbye!')
if __name__ == "__main__": main()
