270 lines
9.1 KiB
Python
270 lines
9.1 KiB
Python
from random import *
|
|
from os import system, name
|
|
|
|
def clear():
|
|
if name == 'nt':
|
|
_ = system('cls')
|
|
else:
|
|
_ = system('clear')
|
|
|
|
class Square:
|
|
def __init__(self, location):
|
|
self.location = location #where the square is as an [x,y] pair
|
|
self.x = location[0]
|
|
self.y = location[1]
|
|
self.state = "empty" #behind the scenes data
|
|
self.display = " " #what the user will see
|
|
|
|
def __repr__(self):
|
|
return f"{self}"
|
|
|
|
#The names of the squares are according to chess convention,
|
|
#but the coordinates are according to usual matrix notation.
|
|
a1 = Square([0,3])
|
|
a2 = Square([0,2])
|
|
a3 = Square([0,1])
|
|
a4 = Square([0,0])
|
|
b1 = Square([1,3])
|
|
b2 = Square([1,2])
|
|
b3 = Square([1,1])
|
|
b4 = Square([1,0])
|
|
c1 = Square([2,3])
|
|
c2 = Square([2,2])
|
|
c3 = Square([2,1])
|
|
c4 = Square([2,0])
|
|
d1 = Square([3,3])
|
|
d2 = Square([3,2])
|
|
d3 = Square([3,1])
|
|
d4 = Square([3,0])
|
|
|
|
board=[a1,a2,a3,a4,b1,b2,b3,b4,c1,c2,c3,c4,d1,d2,d3,d4]
|
|
board_strings=[]
|
|
for box in board:
|
|
board_strings.append(str(box))
|
|
|
|
setup_seed = sample(board, 4) #We pick four squares for initial amazon placement
|
|
|
|
setup_seed[0].state = "ba1" #black amazon 1
|
|
setup_seed[1].state = "ba2" #black amazon 2
|
|
setup_seed[2].state = "wa1" #white amazon 1
|
|
setup_seed[3].state = "wa2" #white amazon 2
|
|
|
|
turn = "w" #White moves first.
|
|
turn_counter = 1
|
|
|
|
def display(): #This interprets the states of the squares for display
|
|
global turn, turn_counter, board, board_strings
|
|
for box in board:
|
|
if box.state == "empty":
|
|
box.display = " "
|
|
if box.state == "ba1":
|
|
box.display = "B"
|
|
if box.state == "ba2":
|
|
box.display = "B"
|
|
if box.state == "wa1":
|
|
box.display = "W"
|
|
if box.state == "wa2":
|
|
box.display = "W"
|
|
if box.state == "fire":
|
|
box.display = "X"
|
|
|
|
def gui(): #This displays the board state to the user
|
|
global turn, turn_counter, board, board_strings
|
|
clear()
|
|
display()
|
|
print(" a b c d")
|
|
print("4 "+a4.display+" "+b4.display+" "+c4.display+" "+d4.display)
|
|
print("3 "+a3.display+" "+b3.display+" "+c3.display+" "+d3.display)
|
|
print("2 "+a2.display+" "+b2.display+" "+c2.display+" "+d2.display)
|
|
print("1 "+a1.display+" "+b1.display+" "+c1.display+" "+d1.display)
|
|
print("It's +")
|
|
|
|
def choice_translate(box):
|
|
global turn, turn_counter, board, board_strings
|
|
if box == "a1":
|
|
return a1
|
|
if box == "a2":
|
|
return a2
|
|
if box == "a3":
|
|
return a3
|
|
if box == "a4":
|
|
return a4
|
|
if box == "b1":
|
|
return b1
|
|
if box == "b2":
|
|
return b2
|
|
if box == "b3":
|
|
return b3
|
|
if box == "b4":
|
|
return b4
|
|
if box == "c1":
|
|
return c1
|
|
if box == "c2":
|
|
return c2
|
|
if box == "c3":
|
|
return c3
|
|
if box == "c4":
|
|
return c4
|
|
if box == "d1":
|
|
return d1
|
|
if box == "d2":
|
|
return d2
|
|
if box == "d3":
|
|
return d3
|
|
if box == "d4":
|
|
return d4
|
|
else:
|
|
return False
|
|
|
|
def valid_move(start, end):
|
|
global turn, turn_counter, board, board_strings
|
|
test_set=[]
|
|
if end.state != "empty":
|
|
return False
|
|
if start.location == end.location:
|
|
return False
|
|
if (start.x != end.x) and (start.y != end.y) and (abs(start.x - end.x) != abs(start.y - end.y)):
|
|
return False
|
|
if start.x == end.x:
|
|
if start.y > end.y:
|
|
for box in board:
|
|
if box.x == start.x:
|
|
for i in range(1, abs(start.y-end.y)):
|
|
if box.y == start.y-i:
|
|
test_set.append(box)
|
|
if start.y < end.y:
|
|
for box in board:
|
|
if box.x == start.x:
|
|
for i in range(1, abs(start.y-end.y)):
|
|
if box.y == start.y+i:
|
|
test_set.append(box)
|
|
if start.y == end.y:
|
|
if start.x > end.x:
|
|
for box in board:
|
|
if box.y == start.y:
|
|
for i in range(1, abs(start.x-end.x)):
|
|
if box.x == start.x-i:
|
|
test_set.append(box)
|
|
if start.x < end.x:
|
|
for box in board:
|
|
if box.y == start.y:
|
|
for i in range(1, abs(start.x-end.x)):
|
|
if box.x == start.x+i:
|
|
test_set.append(box)
|
|
if abs(start.x - end.x) == abs(start.y - end.y):
|
|
if (start.x < end.x) and (start.y < end.y):
|
|
for box in board:
|
|
for i in range(1, abs(start.x-end.x)):
|
|
if ((box.x == start.x+i) and (box.y == start.y+i)):
|
|
test_set.append(box)
|
|
if (start.x > end.x) and (start.y > end.y):
|
|
for box in board:
|
|
for i in range(1, abs(start.x-end.x)):
|
|
if ((box.x == start.x-i) and (box.y == start.y-i)):
|
|
test_set.append(box)
|
|
if (start.x > end.x) and (start.y < end.y):
|
|
for box in board:
|
|
for i in range(1, abs(start.x-end.x)):
|
|
if ((box.x == start.x-i) and (box.y == start.y+i)):
|
|
test_set.append(box)
|
|
if (start.x < end.x) and (start.y > end.y):
|
|
for box in board:
|
|
for i in range(1, abs(start.x-end.x)):
|
|
if ((box.x == start.x+i) and (box.y == start.y-i)):
|
|
test_set.append(box)
|
|
for test in test_set:
|
|
if test.state != "empty":
|
|
return False
|
|
return True
|
|
|
|
def possible_moves(amazon):
|
|
moves=[]
|
|
for box in board:
|
|
if valid_move(amazon, box):
|
|
moves.append(box)
|
|
return(moves)
|
|
|
|
def play():
|
|
global turn, turn_counter, board, board_strings
|
|
gui()
|
|
#We check which amazons can legally move
|
|
movable_amazons=[]
|
|
for box in board:
|
|
if box.state == (turn+"a1" or turn+"a2"):
|
|
if len(possible_moves(box)) != 0:
|
|
movable_amazons.append(box)
|
|
#If none can move, then the game is over!
|
|
if len(movable_amazons)==0:
|
|
if turn_counter%2 == 0:
|
|
return("White wins!")
|
|
if turn_counter%2 == 1:
|
|
return("Black wins!")
|
|
#Otherwise, the player chooses an amazon to move
|
|
if len(movable_amazons) == 2:
|
|
amazon_choice = choice_translate(input("Choose a square containing one of your amazons, either "+movable_amazons[0].self, )) #make this a list of options
|
|
while amazon_choice == False:
|
|
amazon_choice = choice_translate(input("That's not the name of any square. Try again: "))
|
|
while (amazon_choice.state != turn+"a1") and (amazon_choice.state != turn+"a2"):
|
|
amazon_choice = choice_translate(input("You don't have an amazon there. Try again: "))
|
|
gui()
|
|
possible_moves = [] #do this first so your list of choices is only legal ones
|
|
for box in board:
|
|
if valid_move(amazon_choice, box):
|
|
possible_moves.append(box)
|
|
if len(possible_moves) == 0:
|
|
print("That amazon has no moves. You must use the other one.")
|
|
if amazon_choice.state == turn+"a1":
|
|
for box in board:
|
|
if box.state == turn+"a2":
|
|
amazon_choice = box
|
|
if amazon_choice.state == turn+"a2":
|
|
for box in board:
|
|
if box.state == turn+"a1":
|
|
amazon_choice = box
|
|
#The player chooses a square to move to
|
|
move_choice = choice_translate(input("Choose the square you want to move to: "))
|
|
while move_choice == False:
|
|
move_choice = choice_translate(input("That's not the name of any square. Try again: "))
|
|
while valid_move(amazon_choice, move_choice) == False:
|
|
move_choice = choice_translate(input("That amazon can't move there. Try again: "))
|
|
if valid_move(amazon_choice, move_choice):
|
|
move_choice.state = amazon_choice.state
|
|
amazon_choice.state = "empty"
|
|
gui()
|
|
#The player chooses a square to shoot
|
|
burn_choice = choice_translate(input("Choose the square you want to set aflame: "))
|
|
while burn_choice == False:
|
|
burn_choice = choice_translate(input("That's not the name of any square. Try again: "))
|
|
while burn_choice.state != "empty":
|
|
burn_choice = choice_translate(input("That's square isn't empty. Try again: "))
|
|
while valid_move(move_choice, burn_choice) == False:
|
|
burn_choice = choice_translate(input("You can't shoot there. Try again: "))
|
|
burn_choice.state = "fire"
|
|
#Bookkeepping for turn taking
|
|
turn_counter+=1
|
|
if turn_counter%2 == 0:
|
|
turn = "b"
|
|
if turn_counter%2 == 1:
|
|
turn = "w"
|
|
#Now we check to see who won.
|
|
possible_moves = []
|
|
for amazon, box in board:
|
|
if amazon.state == (choice_translate(turn+"a1") or choice_translate(turn+"a2")):
|
|
if valid_move(amazon, box):
|
|
possible_moves.append(box)
|
|
if valid_move(amazon, box):
|
|
possible_moves.append(box)
|
|
if len(possible_moves) == 0:
|
|
if turn == "w":
|
|
return("Black wins!")
|
|
if turn == "b":
|
|
return("White wins!")
|
|
|
|
while True:
|
|
play()
|
|
|
|
|
|
#Next Steps:
|
|
#have students change the abilities of the amazons, size of the board, etc.
|
|
#Add a class to record game state!?!?!?
|