wumpus.py

#Hunt the Wumpus
#From a vintage BASIC game program
#by CREATIVE COMPUTING MORRISTOWN, NEW JERSEY
#Rewritten in Python by Gordon Reeder
# Python 3.4
# ** To do **
# - Make connections within cave random. So that no two caves are the same.

import random
import sys

def show_instructions():
    print ("""
        WELCOME TO 'HUNT THE WUMPUS'
        THE WUMPUS LIVES IN A CAVE OF 8 ROOMS. EACH ROOM
        HAS 3 TUNNELS LEADING TO OTHER ROOMS. (LOOK AT A
        DODECAHEDRON TO SEE HOW THIS WORKS-IF YOU DON'T KNOW
        WHAT A DODECHADRON IS, ASK SOMEONE, or Google it.)
        
    HAZARDS:
        BOTTOMLESS PITS: TWO ROOMS HAVE BOTTOMLESS PITS IN THEM
        IF YOU GO THERE, YOU FALL INTO THE PIT (& LOSE!)

    WUMPUS:
        IF YOU ENTER A ROOM WHERE THE WUMPUS LIVES YOU LOSE

    YOU:
        EACH TURN YOU MAY MOVE OR SHOOT AN ARROW
        MOVING: YOU CAN GO ONE ROOM (THRU ONE TUNNEL)
        ARROWS: YOU HAVE 5 ARROWS. YOU LOSE WHEN YOU RUN
        OUT. YOU AIM BY TELLING
        THE COMPUTER THE ROOM YOU WANT THE ARROW TO GO TO.
        IF THE ARROW HITS THE WUMPUS, YOU WIN.

    WARNINGS:
        WHEN YOU ARE ONE ROOM AWAY FROM WUMPUS OR A HAZARD,
        THE COMPUTER SAYS:
        WUMPUS:   'I SMELL A WUMPUS'
        PIT   :   'I FEEL A DRAFT'
        """)


class Room:
    """Defines a room. 
    A room has a name (or number),
    a list of other rooms that it connects to.
    and a description. 
    How these rooms are built into something larger 
    (cave, dungeon, skyscraper) is up to you.
    """

    def __init__(self, **kwargs):
        self.number = 0
        self.name =''
        self.connects_to = [] #These are NOT objects
        self.description = ""

        for key, value in kwargs.items():
            setattr(self, key, value)

    def __str__(self):
        return str(self.number)

    def remove_connect(self, arg_connect):
        if arg_connect in self.connects_to:
            self.connects_to.remove(arg_connects)

    def add_connect(self, arg_connect):
        if arg_connect not in self.connects_to:
            self.connects_to.append(arg_connect)

    def is_valid_connect(self, arg_connect):
        return arg_connect in self.connects_to

    def get_number_of_connects(self):
        return len(self.connects_to)

    def get_connects(self):
        return self.connects_to

    def describe(self):
        if len(self.description) > 0:
            print(self.description)
        else:
            print("You are in room {}.\nPassages lead to {}".format(self.number, self.connects_to))
        


class Thing:
    """Defines the things that are in the cave.
    That is the Wumpus, Player, pits and bats.
    """

    def __init__(self, **kwargs):
        self.location = 0 # this is a room object
        
        for key, value in kwargs.items():
            setattr(self, key, value)

    def move(self, a_new_location):
        if a_new_location.number in self.location.connects_to or a_new_location == self.location:
            self.location = a_new_location
            return True
        else:
            return False

    def validate_move(self, a_new_location):
        return a_new_location.number in self.location.connects_to or a_new_location == self.location
                
    def get_location(self):
        return self.location.number

    def is_hit(self, a_room):
        return self.location == a_room


def create_things(a_cave):

    Things = []
    Samples = random.sample(a_cave, 4)
    for room in Samples:
        Things.append(Thing(location = room))

    return Things


def create_cave():
    # First create a list of all the rooms.
    # The cave will have 8 rooms
    # Create the connections between rooms
    for number in range(8):
        Cave.append(Room(number = number +1))


    # Then stich them together.
    for idx, room in enumerate(Cave):

        #connect to room to the right
        if idx == 3:
            room.add_connect(Cave[0].number)
            #print("for room: " + str(Cave[3].number) +  " adding connection to " + str(Cave[0].number))
        elif idx == 7:
            room.add_connect(Cave[4].number)
            #print("for room: " + str(Cave[7].number) +  " adding connection to " + str(Cave[4].number))
        else:    
            #print("for room: " + str(Cave[idx].number) +  " adding connection to " + str(Cave[idx +1].number))
            room.add_connect(Cave[idx +1].number)

        #connect to the room to the left
        if idx == 0:
            room.add_connect(Cave[3].number)
            #print("for room: " + str(Cave[0].number) +  " adding connection to " + str(Cave[3].number))

        elif idx == 4:
            room.add_connect(Cave[7].number)
            #print("for room: " + str(Cave[4].number) +  " adding connection to " + str(Cave[7].number))

        else:
            room.add_connect(Cave[idx -1].number)
            #print("for room: " + str(Cave[idx].number) +  " adding connection to " + str(Cave[idx -1].number))

        #connect to the room in the other ring
        if idx < 4:
            room.add_connect(Cave[idx +4].number) #I connect to it.
            Cave[idx +4].add_connect(room.number) #It connects to me
            #print("for room: " + str(Cave[idx].number) +  " adding connection to " + str(Cave[idx+4].number))
            #print("for room: " + str(Cave[idx+4].number) +  " adding connection to " + str(room.number))




Cave = []
create_cave()

from KB import KB

KB_HELPER=KB()
# Make player, wumpus, pits and put into cave.

Wumpus, Player, Pit1, Pit2 = create_things(Cave)

Arrows = 2

# Now play the game

print("""\n   Welcome to the cave, Great White Hunter.
    You are hunting the Wumpus.
    On any turn you can move or shoot.
    Commands are entered in the form of ACTION LOCATION
    IE: 'SHOOT  5' or 'MOVE 8'
    type 'HELP' for instructions.
    'QUIT' to end the game.
    """)


while True:

    Player.location.describe()
    #Check each <Player.location.connects_to> for hazards.
    breeze=False
    smell=False
    KB_HELPER.location_info(Player.location.number)
    for room in Player.location.connects_to:
        if Wumpus.location.number == room:
            #print("Wumpus location is " + str(room))
            print("I smell a Wumpus!")
            smell=True
        if Pit1.location.number == room or Pit2.location.number == room:
            #print("Pit location is " + str(room))
            print("I feel a draft!")
            breeze=True

    ##################################################
    # ADDED THIS CODE TO PROVIDE INFORMATION TO THE KB
    #################################################
    if smell:
            KB_HELPER.smell_info(Player.location.number, smell_wumpus=True)
    else:
            KB_HELPER.smell_info(Player.location.number, smell_wumpus=False)
            
    if breeze:
            KB_HELPER.breeze_info(Player.location.number, feel_breeze=True)
    else:
            KB_HELPER.breeze_info(Player.location.number, feel_breeze=False)
   


    
    raw_command = input("\n> ")
    command_list = raw_command.split(' ')
    command = command_list[0].upper()
    if len(command_list) > 1:
        try:
            move = Cave[int(command_list[1]) -1]
        except:
            print("\n **What??")
            continue
    else:
        move = Player.location

    if command == 'HELP' or command == 'H':
        show_instructions()
        continue

    elif command == 'QUIT' or command == 'Q':
        print("\nOK, Bye.")
        sys.exit()

    elif command == 'ASK_KB' or command == 'K':
        KB_HELPER.make_suggestion(Player.location.connects_to)
    elif command == 'MOVE' or command == 'M':
        if not Player.move(move):
            print("\n **You can't get there from here")
            continue

    elif command == 'SHOOT' or command == 'S':
        if Player.validate_move(move):
            print("\n-Twang-") 
            if Wumpus.location == move:
                print("\n Good Shooting!! You hit the Wumpus. \n Wumpi will have their revenge.\n")
                sys.exit()
        else:
            print("\n** Stop trying to shoot through walls.")

        Arrows -= 1
        if Arrows == 0:
            print("\n You are out of arrows\n Better luck next time\n")
            sys.exit()
    
    else:
        print("\n **What?")
        continue

        
    # By now the player has moved. See what happened.
    # Handle problems with pits and wumpus.
 

    if Player.location == Wumpus.location:
        print("TROMP TROMP - WUMPUS GOT YOU!\n")
        sys.exit()    

    elif Player.location == Pit1.location or Player.location == Pit2.location:
        print("YYYIIIIEEEE . . . FELL INTO A PIT!\n")
        sys.exit()

    else: # Keep playing
        pass