Not the end tools

not_the_end_simple_token_picker.py

import random


def simulate(num_tokens_to_pick, black_tokens, white_tokens):
    picked_tokens = []
    for _ in range(min(num_tokens_to_pick, black_tokens)):
        picked_tokens.append("black")
        black_tokens -= 1
        num_tokens_to_pick -= 1

    for _ in range(min(num_tokens_to_pick, white_tokens)):
        picked_tokens.append("white")
        white_tokens -= 1
        num_tokens_to_pick -= 1

    random.shuffle(picked_tokens)
    return picked_tokens


black_tokens = int(input("How black token you put in that bag?"))
white_tokens = int(input("How white token you put in that bag?"))
num_tokens_to_pick = int(input("How many time you want to pick? "))

print(simulate(num_tokens_to_pick, black_tokens, white_tokens))