import copy, random
def print_board(s,board):
# WARNING: This function was crafted with a lot of attention. Please be aware that any
# modifications to this function will result in a poor output of the board
# layout. You have been warn.
#find out if you are printing the computer or user board
player = "Computer"
if s == "u":
player = "User"
print "The " + player + "'s board look like this: \n"
#print the horizontal numbers
print " ",
for i in range(10):
print " " + str(i+1) + " ",
print "\n"
for i in range(10):
#print the vertical line number
if i != 9:
print str(i+1) + " ",
else:
print str(i+1) + " ",
#print the board values, and cell dividers
for j in range(10):
if board[i][j] == -1:
print ' ',
elif s == "u":
print board[i][j],
elif s == "c":
if board[i][j] == "*" or board[i][j] == "$":
print board[i][j],
else:
print " ",
if j != 9:
print " | ",
print
#print a horizontal line
if i != 9:
print " ----------------------------------------------------------"
else:
print
def user_place_ships(board,ships):
for ship in ships.keys():
#get coordinates from user and vlidate the postion
valid = False
while(not valid):
print_board("u",board)
print "Placing a/an " + ship
x,y = get_coor()
ori = v_or_h()
valid = validate(board,ships[ship],x,y,ori)
if not valid:
print "Cannot place a ship there.\nPlease take a look at the board and try again."
raw_input("Hit ENTER to continue")
#place the ship
board = place_ship(board,ships[ship],ship[0],ori,x,y)
print_board("u",board)
raw_input("Done placing user ships. Hit ENTER to continue")
return board
def computer_place_ships(board,ships):
for ship in ships.keys():
#genreate random coordinates and vlidate the postion
valid = False
while(not valid):
x = random.randint(1,10)-1
y = random.randint(1,10)-1
o = random.randint(0,1)
if o == 0:
ori = "v"
else:
ori = "h"
valid = validate(board,ships[ship],x,y,ori)
#place the ship
print "Computer placing a/an " + ship
board = place_ship(board,ships[ship],ship[0],ori,x,y)
return board
def place_ship(board,ship,s,ori,x,y):
#place ship based on orientation
if ori == "v":
for i in range(ship):
board[x+i][y] = s
elif ori == "h":
for i in range(ship):
board[x][y+i] = s
return board
def validate(board,ship,x,y,ori):
#validate the ship can be placed at given coordinates
if ori == "v" and x+ship > 10:
return False
elif ori == "h" and y+ship > 10:
return False
else:
if ori == "v":
for i in range(ship):
if board[x+i][y] != -1:
return False
elif ori == "h":
for i in range(ship):
if board[x][y+i] != -1:
return False
return True
def v_or_h():
#get ship orientation from user
while(True):
user_input = raw_input("vertical or horizontal (v,h) ? ")
if user_input == "v" or user_input == "h":
return user_input
else:
print "Invalid input. Please only enter v or h"
def get_coor():
while (True):
user_input = raw_input("Please enter coordinates (row,col) ? ")
try:
#see that user entered 2 values seprated by comma
coor = user_input.split(",")
if len(coor) != 2:
raise Exception("Invalid entry, too few/many coordinates.");
#check that 2 values are integers
coor[0] = int(coor[0])-1
coor[1] = int(coor[1])-1
#check that values of integers are between 1 and 10 for both coordinates
if coor[0] > 9 or coor[0] < 0 or coor[1] > 9 or coor[1] < 0:
raise Exception("Invalid entry. Please use values between 1 to 10 only.")
#if everything is ok, return coordinates
return coor
except ValueError:
print "Invalid entry. Please enter only numeric values for coordinates"
except Exception as e:
print e
def make_move(board,x,y):
#make a move on the board and return the result, hit, miss or try again for repeat hit
if board[x][y] == -1:
return "miss"
elif board[x][y] == '*' or board[x][y] == '$':
return "try again"
else:
return "hit"
def user_move(board):
#get coordinates from the user and try to make move
#if move is a hit, check ship sunk and win condition
while(True):
x,y = get_coor()
res = make_move(board,x,y)
if res == "hit":
print "Hit at " + str(x+1) + "," + str(y+1)
check_sink(board,x,y)
board[x][y] = '$'
if check_win(board):
return "WIN"
elif res == "miss":
print "Sorry, " + str(x+1) + "," + str(y+1) + " is a miss."
board[x][y] = "*"
elif res == "try again":
print "Sorry, that coordinate was already hit. Please try again"
if res != "try again":
return board
def computer_move(board):
#generate user coordinates from the user and try to make move
#if move is a hit, check ship sunk and win condition
while(True):
x = random.randint(1,10)-1
y = random.randint(1,10)-1
res = make_move(board,x,y)
if res == "hit":
print "Hit at " + str(x+1) + "," + str(y+1)
check_sink(board,x,y)
board[x][y] = '$'
if check_win(board):
return "WIN"
elif res == "miss":
print "Sorry, " + str(x+1) + "," + str(y+1) + " is a miss."
board[x][y] = "*"
if res != "try again":
return board
def check_sink(board,x,y):
#figure out what ship was hit
if board[x][y] == "A":
ship = "Aircraft Carrier"
elif board[x][y] == "B":
ship = "Battleship"
elif board[x][y] == "S":
ship = "Submarine"
elif board[x][y] == "D":
ship = "Destroyer"
elif board[x][y] == "P":
ship = "Patrol Boat"
#mark cell as hit and check if sunk
board[-1][ship] -= 1
if board[-1][ship] == 0:
print ship + " Sunk"
def check_win(board):
#simple for loop to check all cells in 2d board
#if any cell contains a char that is not a hit or a miss return false
for i in range(10):
for j in range(10):
if board[i][j] != -1 and board[i][j] != '*' and board[i][j] != '$':
return False
return True
def main():
#types of ships
ships = {"Aircraft Carrier":5,
"Battleship":4,
"Submarine":3,
"Destroyer":3,
"Patrol Boat":2}
#setup blank 10x10 board
board = []
for i in range(10):
board_row = []
for j in range(10):
board_row.append(-1)
board.append(board_row)
#setup user and computer boards
user_board = copy.deepcopy(board)
comp_board = copy.deepcopy(board)
#add ships as last element in the array
user_board.append(copy.deepcopy(ships))
comp_board.append(copy.deepcopy(ships))
#ship placement
user_board = user_place_ships(user_board,ships)
comp_board = computer_place_ships(comp_board,ships)
#game main loop
while(1):
#user move
print_board("c",comp_board)
comp_board = user_move(comp_board)
#check if user won
if comp_board == "WIN":
print "User WON! :)"
quit()
#display current computer board
print_board("c",comp_board)
raw_input("To end user turn hit ENTER")
#computer move
user_board = computer_move(user_board)
#check if computer move
if user_board == "WIN":
print "Computer WON! :("
quit()
#display user board
print_board("u",user_board)
raw_input("To end computer turn hit ENTER")
if __name__=="__main__":
main()
