def parseBoard(
self, string
): # because JSONs are strings, we must parse the board and create a new board
temp = Board()
string = string.replace(',', '')
string = string.replace('[', '')
string = string.replace(']', '')
string = string.replace('\'', '')
for i in range(1, len(string)):
string = string[:i] + string[i + 1:]
s = 0
for i in range(0, 10):
for j in range(0, 10):
temp.board[i][j] = string[s]
s += 1
return temp
def testOverlapCheck(self):
testObject = Board()
model = ModelBattleship()
model.placeShip(testObject, 5, [0, 0], 'v')
actualValue = model.overlapCheck(testObject, 5, [0, 0], 'h')
expectedValue = False
assert (actualValue == expectedValue)
def testBoatLocation(self):
testObject = Board()
model = ModelBattleship()
model.placeShip(testObject, 1, [0, 2], 'v')
#view = ViewBattleship()
#view.displayBoard(testObject)
expectedValue = 'O'
actualValue = testObject.board[0][2]
assert (actualValue == expectedValue)
def testGameBoardInit(self):
testObject = Board()
# test the size of the board
expectedValue = 10
actualValue = testObject.size
assert (actualValue == expectedValue)
# test that the board is empty
expectedValue = ' '
actualValue = testObject.board[0][0]
assert (actualValue == expectedValue)
def testCheckShot(self):
testObject = Board()
model = ModelBattleship()
model.placeShip(testObject, 5, [0, 0], 'v')
model.placeShip(testObject, 4, [0, 1], 'v')
model.placeShip(testObject, 3, [0, 2], 'v')
model.placeShip(testObject, 2, [0, 3], 'v')
model.placeShip(testObject, 1, [0, 4], 'v')
model.placeShot(testObject, [0, 0])
actualValue = model.checkShot(testObject, [0, 0])
expectedValue = False
assert (actualValue == expectedValue)
def testMaskBoard(self):
testObject = Board()
model = ModelBattleship()
model.placeShip(testObject, 5, [0, 0], 'v')
model.placeShip(testObject, 4, [0, 1], 'v')
model.placeShip(testObject, 3, [0, 2], 'v')
model.placeShip(testObject, 2, [0, 3], 'v')
model.placeShip(testObject, 1, [0, 4], 'v')
model.placeShot(testObject, [0, 0])
maskedBoard = model.maskBoard(testObject)
actualValue = testObject
expectedValue = maskedBoard
assert (actualValue != expectedValue)
def testWin(self):
testObject = Board()
model = ModelBattleship()
model.placeShip(testObject, 2, [0, 0], 'v')
model.placeShot(testObject, [0, 0])
expectedValue = False
actualValue = model.checkWin(testObject)
assert (actualValue == expectedValue)
model.placeShot(testObject, [1, 0])
expectedValue = True
actualValue = model.checkWin(testObject)
assert (actualValue == expectedValue)
def testFireShot(self):
testObject = Board()
model = ModelBattleship()
model.placeShip(testObject, 3, [5, 2], 'v')
model.placeShot(testObject, [5, 2])
expectedValue = 'X'
actualValue = testObject.board[5][2]
assert (actualValue == expectedValue)
model.placeShot(testObject, [9, 9])
expectedValue = '*'
actualValue = testObject.board[9][9]
assert (actualValue == expectedValue)
def testScore(self):
testObject = Board()
model = ModelBattleship()
model.placeShip(testObject, 3, [5, 2], 'v')
model.placeShot(testObject, [5, 2])
expectedValue = 100
actualValue = testObject.score
assert (actualValue == expectedValue)
model.placeShot(testObject, [9, 9])
expectedValue = 90
actualValue = testObject.score
assert (actualValue == expectedValue)
def testOutofBounds(self):
testObject = Board()
model = ModelBattleship()
class Client:
# client data
shutdown = False
host = socket.gethostbyname(socket.gethostname())
port = 5000
server = (host, port)
buffer = 1024
player_board = Board()
player_shot = False
enemy_board = Board()
enemy_shot = False
view = ViewBattleship()
model = ModelBattleship()
username = view.get_username()
def __init__(self):
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
try:
sock.connect((self.host, self.port))
except socket.error:
self.view.display("Error connecting to the server.")
os._exit(1)
sock.sendto(str.encode(self.username), self.server)
data, server = sock.recvfrom(self.buffer)
data = data.decode('utf-8')
self.view.display(data)
# start here
data, server = sock.recvfrom(self.buffer)
data = data.decode('utf-8')
self.view.display(data)
self.view.displayBoard(self.player_board)
self.inputShips()
send_board = '{"username":"******", "action":"board", "data":{"board":"' + str(
self.player_board.board) + '"}}'
sock.sendto(str.encode(send_board), server)
data, server = sock.recvfrom(self.buffer)
data = data.decode('utf-8')
if data == 'Game begin':
while 1:
send = '{"username":"******", "action":"win request", "data":{"win_c":""}}'
sock.sendto(str.encode(send), server)
win_c, server = sock.recvfrom(self.buffer)
win_c = json.loads(win_c.decode('utf-8'))
if win_c['data']['win_c'] == 'true' and win_c[
'username'] == self.username:
self.view.display("Congrats you won!!")
break
if win_c['data']['win_c'] == 'true' and win_c[
'username'] != self.username:
self.view.display("Oh no you lost!!")
break
shot = self.view.getShot()
out_shot = '{"username":"******", "action":"outgoing shot", "data":{"coordinate":"' + str(
shot[0]) + ' ' + str(
shot[1]) + '", "result":"awaiting response"}}'
if self.model.checkShot(self.enemy_board, shot):
sock.sendto(str.encode(out_shot), self.server)
self.getShot(sock)
else:
self.view.display(
"Shot already placed in this location. Try again: \n")
self.view.displayScore(self.enemy_board)
def inputShips(
self): # copied from the Controller to please the Python gods.
self.view.display(
"Time to place your ships! Please input coordinates of the head. e.g. A6 or E2 "
)
for i in range(0, 5):
while 1:
loc = self.view.getShips(i)
if self.model.boundaryCheck(
5 - i, [int(loc[0][0]), int(loc[0][1])],
loc[1]) and self.model.overlapCheck(
self.player_board, 5 - i,
[int(loc[0][0]), int(loc[0][1])], loc[1]):
self.model.placeShip(
self.player_board, 5 - i,
[int(loc[0][0]), int(loc[0][1])], loc[1])
self.view.displayBoard(self.player_board)
break
else:
self.view.display("Invalid ship location. Try again. \n")
def getShot(self,
sock): # copied from the Controller to please the Python gods.
while True:
if self.enemy_shot and self.player_shot:
self.view.display("Your board: ")
self.view.displayBoard(self.player_board)
self.view.display("Enemy Board: ")
self.view.displayBoard(self.enemy_board)
self.enemy_shot = False
self.player_shot = False
break
data, server = sock.recvfrom(self.buffer)
data = json.loads(data.decode('utf-8'))
if data['action'] == 'server shot' and data[
'username'] != self.username:
result = (data['data']['coordinate'], data['data']['result'])
coord = (int(result[0][1]), int(result[0][4]))
if result[1] == 'hit':
self.enemy_board.board[coord[0]][coord[1]] = 'X'
self.enemy_board.score = self.enemy_board.score + 100
elif result[1] == 'miss':
self.enemy_board.board[coord[0]][coord[1]] = '*'
self.enemy_board.score = self.enemy_board.score - 10
self.player_shot = True
if data['action'] == 'server shot' and data[
'username'] == self.username:
result = (data['data']['coordinate'], data['data']['result'])
coord = (int(result[0][1]), int(result[0][4]))
if result[1] == 'hit':
self.player_board.board[coord[0]][coord[1]] = 'X'
elif result[1] == 'miss':
self.player_board.board[coord[0]][coord[1]] = '*'
self.enemy_shot = True
class ControllerBattleship:
playerBoard = Board()
enemyBoard = Board()
view = ViewBattleship()
model = ModelBattleship()
def inputShips(self):
print(
"Time to place your ships! Please input coordinates of the head. e.g. A6 or E2 "
)
for i in range(0, 5):
while 1:
loc = self.view.getShips(i)
if self.model.boundaryCheck(
5 - i, [int(loc[0][0]), int(loc[0][1])],
loc[1]) and self.model.overlapCheck(
self.playerBoard, 5 - i,
[int(loc[0][0]), int(loc[0][1])], loc[1]):
self.model.placeShip(
self.playerBoard, 5 - i,
[int(loc[0][0]), int(loc[0][1])], loc[1])
self.view.displayBoard(self.playerBoard)
break
else:
self.view.display("Invalid ship location. Try again. \n")
def getShot(self):
while True:
shot = self.view.getShot()
if self.model.checkShot(self.enemyBoard, shot):
self.model.placeShot(self.enemyBoard, shot)
break
else:
self.view.display(
"Shot already placed in this location. Try again: \n")
self.view.displayScore(self.enemyBoard)
def makeAIBoard(self):
for i in range(0, 5):
while 1:
start = [random.randint(0, 9), random.randint(0, 9)]
orient = random.choice(['v', 'h'])
loc = (start, orient)
if self.model.boundaryCheck(
i + 1, [int(loc[0][0]), int(loc[0][1])],
loc[1]) and self.model.overlapCheck(
self.enemyBoard, i + 1,
[int(loc[0][0]), int(loc[0][1])], loc[1]):
try:
self.enemyBoard.board[start[0]][start[1]] = str(i + 1)
if orient == 'v':
for j in range(0, i + 1):
self.enemyBoard.board[start[0] +
j][start[1]] = 'O'
break
elif orient == 'h':
for j in range(0, i + 1):
self.enemyBoard.board[start[0]][start[1] +
j] = 'O'
break
except:
pass
break
'''
break
'''
self.model.placeShip(self.enemyBoard, 5, [0, 0], 'v')
self.model.placeShip(self.enemyBoard, 4, [0, 1], 'v')
self.model.placeShip(self.enemyBoard, 3, [0, 2], 'v')
self.model.placeShip(self.enemyBoard, 2, [0, 3], 'v')
self.model.placeShip(self.enemyBoard, 1, [0, 4], 'v')
'''
if __name__ == '__main__':
flag = True
while flag:
controller = ControllerBattleship()
controller.playerBoard = Board()
controller.enemyBoard = Board()
view = ViewBattleship()
model = ModelBattleship()
userInput = view.displayMenu()
if userInput[0] == '1':
controller.makeAIBoard()
view.displayBoard(controller.playerBoard)
controller.inputShips()
if userInput[1] == '1':
while 1:
view.display("Enemy Board: ")
view.displayBoard(model.maskBoard(controller.enemyBoard))
controller.getShot()
if model.checkWin(controller.enemyBoard):
view.display("Congrats you won!!")