diff --git a/main.py b/main.py index 55b86cd..251dca3 100644 --- a/main.py +++ b/main.py @@ -21,18 +21,18 @@ class TicTacToe: if self.board.is_win(self.player1.token): self.board.print() if len(self.board.empty_fields) == 0: - print("It's a draw.") + print("It's a draw.\n") else: - print(self.player1.name + " has won.") + print(self.player1.name + " has won.\n") break self.board.make_move(self.player2.token, self.player2.choose_random_field(self.board.empty_fields)) if self.board.is_win(self.player2.token): self.board.print() if len(self.board.empty_fields) == 0: - print("It's a draw.") + print("It's a draw.\n") else: - print(self.player2.name + " has won.") + print(self.player2.name + " has won.\n") break class Board: @@ -83,11 +83,16 @@ number_of_rounds = None try: opts, args = getopt.getopt(sys.argv[1:], "n:", ["numberofrounds="]) except getopt.GetoptError: + print("main.py -n ") + sys.exit(2) + +if(len(sys.argv) != 1): + for opt, arg in opts: + if opt in ("-n", "--numberofrounds"): + number_of_rounds = arg +else: print("main.py -n ") sys.exit() -for opt, arg in opts: - if opt in ("-n", "--numberofrounds"): - number_of_rounds = arg game = TicTacToe() for x in range(0, int(number_of_rounds)):