Update P75_TicTacToe.py

Programs/P75_TicTacToe.py

@@ -2,74 +2,87 @@
 
 # A simple example of tic tac toe game
 
-# For storing user choices
-choices = []
-
-# For initializing the board with numbers
-for i in range(0, 9):
-    choices.append(str(i))
-
-firstPlayer = True
-winner = False
-iterations = 0      # To terminate the loop
-
-# For drawing board on to the terminal
-def printBoard():
-    print('\n=============')
-    print('| ' + choices[0] + ' | ' + choices[1] + ' | ' + choices[2] + ' |')
-    print('=============')
-    print('| ' + choices[3] + ' | ' + choices[4] + ' | ' + choices[5] + ' |')
-    print('=============')
-    print('| ' + choices[6] + ' | ' + choices[7] + ' | ' + choices[8] + ' |')
-    print('=============\n')
-
-# Play the game while the winner is not decided or the game is drawn
-while not winner and iterations < 9:
-    printBoard()
-
-    iterations += 1
-
-    if firstPlayer == True:
-        print('Player 1: ', end = '')
-    else:
-        print('Player 2: ', end = '')
-
-    try:
-        playerInput = int(input())
-    except:
-        print('Please enter a valid number from the board')
-        continue
-
-    # Check if userInput already has 'X' or 'O'
-    if choices[playerInput] == 'X' or choices[playerInput] == 'O':
-        print('Illegal move, try again!')
-        continue
-
-    if firstPlayer:
-        choices[playerInput] = 'X'
-    else:
-        choices[playerInput] = 'O'
-
-    firstPlayer = not firstPlayer
+def game():
+    # For storing user choices
+    choices = []
+
+    # For initializing the board with numbers
+    for i in range(0, 9):
+        choices.append(str(i))
+
+    firstPlayer = True
+    winner = False
+    iterations = 0      # To terminate the loop
+
+    # For drawing board on to the terminal
+    def printBoard():
+        print('\n=============')
+        print('| ' + choices[0] + ' | ' + choices[1] + ' | ' + choices[2] + ' |')
+        print('=============')
+        print('| ' + choices[3] + ' | ' + choices[4] + ' | ' + choices[5] + ' |')
+        print('=============')
+        print('| ' + choices[6] + ' | ' + choices[7] + ' | ' + choices[8] + ' |')
+        print('=============\n')
+
+    # Play the game while the winner is not decided or the game is drawn
+    while not winner and iterations < 9:
+        printBoard()
 
-    # Winning conditions
-    for index in range(0, 3):
-        # For [0,1,2], [3,4,5], [6,7,8]
-        if (choices[index * 3] == choices[((index * 3) + 1)] and choices[index * 3] == choices[((index * 3) + 2)]):
+        iterations += 1
+
+        if firstPlayer == True:
+            print('Player 1: ', end = '')
+        else:
+            print('Player 2: ', end = '')
+
+        try:
+            playerInput = int(input())
+        except:
+            print('Please enter a valid number from the board')
+            continue
+
+        # Check if userInput already has 'X' or 'O'
+        if choices[playerInput] == 'X' or choices[playerInput] == 'O':
+            print('Illegal move, try again!')
+            continue
+
+        if firstPlayer:
+            choices[playerInput] = 'X'
+        else:
+            choices[playerInput] = 'O'
+
+        firstPlayer = not firstPlayer
+
+        # Winning conditions
+        for index in range(0, 3):
+            # For [0,1,2], [3,4,5], [6,7,8]
+            if (choices[index * 3] == choices[((index * 3) + 1)] and choices[index * 3] == choices[((index * 3) + 2)]):
+                winner = True
+                printBoard()
+
+            # For [0,3,6], [1,4,7], [2,5,8]
+            if(choices[index] == choices[index + 3] and choices[index + 3] == choices[index + 6]):
+                winner = True
+                printBoard()
+
+        if((choices[0] == choices[4] and choices[4] == choices[8]) or
+          (choices[2] == choices[4] and choices[4] == choices[6])):
             winner = True
             printBoard()
 
-        # For [0,3,6], [1,4,7], [2,5,8]
-        if(choices[index] == choices[index + 3] and choices[index + 3] == choices[index + 6]):
-            winner = True
-            printBoard()
+    if winner:
+        print('Player ' + str(int(firstPlayer + 1)) + ' wins!')
+    else:
+        print('Game drawn')
+    restart()
+
+#Ask the user to play again. If so, restart the game().
+def restart():
+    restart = input("Play again? (y/n): ")
+
+    if(restart == 'y'):
+        game()
 
-    if((choices[0] == choices[4] and choices[4] == choices[8]) or
-      (choices[2] == choices[4] and choices[4] == choices[6])):
-        winner = True
-        printBoard()
 
-if winner:
-    print('Player ' + str(int(firstPlayer + 1)) + ' wins!')
-else:
-    print('Game drawn')
+if __name__ == '__main__':
+    game()