Game.py

from Human import Human
from Computer import Computer
import socket

FORMAT = 'utf-8'


class Game:
    def __init__(self):
        self.player0 = Human()
        self.player1 = None

    def single_player(self):
        """
        single player game against the computer, with option for easy/hard mode
        :return:
        """
        mode = input('Choose game mode (easy/hard): ').lower()
        while mode not in ['easy', 'hard']:
            mode = input('please enter a valid input: ')
        self.player1 = Computer(mode)

        self.player0.place_battleships()
        self.player1.place_battleships()
        while True:
            result = self.player0.turn(self.player1)
            if result == 'win':
                print('player 0 has won!')
                return

            result = self.player1.turn(self.player0)
            if result == 'win':
                print('player 1 has won!')
                return

    def multi_player(self):
        """
        multiplayer game, the second player can join through running SecondPlayer.py
        :return:
        """
        self.player0.output('waiting for another player...')
        server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        server.bind((socket.gethostname(), 1234))
        server.listen(1)
        conn, addr = server.accept()
        self.player1 = Human(conn)
        self.player1.output('waiting for opponent to place battleships...')
        self.player0.place_battleships()
        self.player0.output('waiting for opponent to place battleships...')
        self.player1.place_battleships()
        while True:
            self.player1.output('Opponent\'s turn:')
            result = self.player0.turn(self.player1)
            if result == 'win':
                self.player0.output('player 0 has won!')
                self.player1.output('player 0 has won!')
                return

            self.player0.output('Opponent\'s turn:')
            result = self.player1.turn(self.player0)
            if result == 'win':
                self.player0.output('player 1 has won!')
                self.player1.output('player 1 has won!')
                return