pick_six.py

import random

MAX_NUMBER = 10


def user_picks():
    user_numbers = set()

    # Add six numbers to the set.
    while len(user_numbers) < 6:
        new_number = input("Enter another number (1-"+str(MAX_NUMBER)+"): ")
        try:
            new_number = int(new_number)
            if new_number < 1 or new_number > MAX_NUMBER:
                print("Number out of range.  Try again.")
            else:
                user_numbers.add(new_number)
        except ValueError:
            print("Be sure to enter a integer!")

    return user_numbers


def comp_picks():
    winning_numbers = set()

    # Add six numbers to the set.
    while len(winning_numbers) < 6:
        winning_numbers.add(random.randint(1, MAX_NUMBER))

    return winning_numbers


# Simulate a pick 6 lotto
def main():
    # Get user's picks
    users_numbers = user_picks()

    # Get computer's picks
    computers_numbers = comp_picks()

    # See how many in common
    number_in_common = len(users_numbers & computers_numbers)

    print("Your numbers: ", users_numbers)
    print("Computer numbers: ", computers_numbers)
    print("You have", number_in_common, "winning numbers!")


main()