minesweeper.py

# Importing packages
import random
import os

# Printing the Minesweeper Layout
def print_mines_layout():

	global mine_values
	global n

	print()
	print("\t\t\tMINESWEEPER\n")

	st = "   "
	for i in range(n):
		st = st + "     " + str(i + 1)
	print(st)	

	for r in range(n):
		st = "     "
		if r == 0:
			for c in range(n):
				st = st + "______"	
			print(st)

		st = "     "
		for c in range(n):
			st = st + "|     "
		print(st + "|")
		
		st = "  " + str(r + 1) + "  "
		for c in range(n):
			st = st + "|  " + str(mine_values[r][c]) + "  "
		print(st + "|")	

		st = "     "
		for c in range(n):
			st = st + "|_____"
		print(st + '|')

	print()
 
# Function for setting up Mines
def set_mines():

	global numbers
	global mines_no
	global n

	# Track of number of mines already set up
	count = 0
	while count < mines_no:

		# Random number from all possible grid positions 
		val = random.randint(0, n*n-1)

		# Generating row and column from the number
		r = val // n
		c = val % n

		# Place the mine, if it doesn't already have one
		if numbers[r][c] != -1:
			count = count + 1
			numbers[r][c] = -1

# Function for setting up the other grid values
def set_values():

	global numbers
	global n

	# Loop for counting each cell value
	for r in range(n):
		for c in range(n):

			# Skip, if it contains a mine
			if numbers[r][c] == -1:
				continue

			# Check up	
			if r > 0 and numbers[r-1][c] == -1:
				numbers[r][c] = numbers[r][c] + 1
			# Check down	
			if r < n-1  and numbers[r+1][c] == -1:
				numbers[r][c] = numbers[r][c] + 1
			# Check left
			if c > 0 and numbers[r][c-1] == -1:
				numbers[r][c] = numbers[r][c] + 1
			# Check right
			if c < n-1 and numbers[r][c+1] == -1:
				numbers[r][c] = numbers[r][c] + 1
			# Check top-left	
			if r > 0 and c > 0 and numbers[r-1][c-1] == -1:
				numbers[r][c] = numbers[r][c] + 1
			# Check top-right
			if r > 0 and c < n-1 and numbers[r-1][c+1] == -1:
				numbers[r][c] = numbers[r][c] + 1
			# Check below-left	
			if r < n-1 and c > 0 and numbers[r+1][c-1] == -1:
				numbers[r][c] = numbers[r][c] + 1
			# Check below-right
			if r < n-1 and c < n-1 and numbers[r+1][c+1] == -1:
				numbers[r][c] = numbers[r][c] + 1

# Recursive function to display all zero-valued neighbours	
def neighbours(r, c):
	
	global mine_values
	global numbers
	global vis

	# If the cell already not visited
	if [r,c] not in vis:

		# Mark the cell visited
		vis.append([r,c])

		# If the cell is zero-valued
		if numbers[r][c] == 0:

			# Display it to the user
			mine_values[r][c] = numbers[r][c]

			# Recursive calls for the neighboring cells
			if r > 0:
				neighbours(r-1, c)
			if r < n-1:
				neighbours(r+1, c)
			if c > 0:
				neighbours(r, c-1)
			if c < n-1:
				neighbours(r, c+1)	
			if r > 0 and c > 0:
				neighbours(r-1, c-1)
			if r > 0 and c < n-1:
				neighbours(r-1, c+1)
			if r < n-1 and c > 0:
				neighbours(r+1, c-1)
			if r < n-1 and c < n-1:
				neighbours(r+1, c+1)	

		# If the cell is not zero-valued 			
		if numbers[r][c] != 0:
				mine_values[r][c] = numbers[r][c]

# Function for clearing the terminal
def clear():
	os.system("clear")		

# Function to display the instructions
def instructions():
	print("Instructions:")
	print("1. Enter row and column number to select a cell, Example \"2 3\"")
	print("2. In order to flag a mine, enter F after row and column numbers, Example \"2 3 F\"")

# Function to check for completion of the game
def check_over():
	global mine_values
	global n
	global mines_no

	# Count of all numbered values
	count = 0

	# Loop for checking each cell in the grid
	for r in range(n):
		for c in range(n):

			# If cell not empty or flagged
			if mine_values[r][c] != ' ' and mine_values[r][c] != 'F':
				count = count + 1
	
	# Count comparison 			
	if count == n * n - mines_no:
		return True
	else:
		return False
					
def show_mines():
	global mine_values
	global numbers
	global n

	for r in range(n):
		for c in range(n):
			if numbers[r][c] == -1:
				mine_values[r][c] = 'M'


if __name__ == "__main__":

	# Size of grid
	n = 8
	# Number of mines
	mines_no = 8

	# The actual values of the grid
	numbers = [[0 for y in range(n)] for x in range(n)] 
	# The apparent values of the grid
	mine_values = [[' ' for y in range(n)] for x in range(n)]
	# The positions that have been flagged
	flags = []

	# Set the mines
	set_mines()

	# Set the values
	set_values()

	# Display the instructions
	instructions()

	# Variable for maintaining Game Loop
	over = False
		
	# The GAME LOOP	
	while not over:
		print_mines_layout()

		# Input from the user
		inp = input("Enter row number followed by space and column number = ").split()
		
		# Standard input
		if len(inp) == 2:

			# Try block to handle errant input
			try: 
				val = list(map(int, inp))
			except ValueError:
				clear()
				print("Wrong input!")
				instructions()
				continue

		# Flag input
		elif len(inp) == 3:
			if inp[2] != 'F' and inp[2] != 'f':
				clear()
				print("Wrong Input!")
				instructions()
				continue

			# Try block to handle errant input	
			try:
				val = list(map(int, inp[:2]))
			except ValueError:
				clear()
				print("Wrong input!")
				instructions()
				continue

			# Sanity checks	
			if val[0] > n or val[0] < 1 or val[1] > n or val[1] < 1:
				clear()
				print("Wrong input!")
				instructions()
				continue

			# Get row and column numbers
			r = val[0]-1
			c = val[1]-1	

			# If cell already been flagged
			if [r, c] in flags:
				clear()
				print("Flag already set")
				continue

			# If cell already been displayed
			if mine_values[r][c] != ' ':
				clear()
				print("Value already known")
				continue

			# Check the number for flags 	
			if len(flags) < mines_no:
				clear()
				print("Flag set")

				# Adding flag to the list
				flags.append([r, c])
				
				# Set the flag for display
				mine_values[r][c] = 'F'
				continue
			else:
				clear()
				print("Flags finished")
				continue	 

		else: 
			clear()
			print("Wrong input!")	
			instructions()
			continue
			

		# Sanity checks
		if val[0] > n or val[0] < 1 or val[1] > n or val[1] < 1:
			clear()
			print("Wrong Input!")
			instructions()
			continue
			
		# Get row and column number
		r = val[0]-1
		c = val[1]-1

		# Unflag the cell if already flagged
		if [r, c] in flags:
			flags.remove([r, c])

		# If landing on a mine --- GAME OVER	
		if numbers[r][c] == -1:
			mine_values[r][c] = 'M'
			show_mines()
			print_mines_layout()
			print("Landed on a mine. GAME OVER!!!!!")
			over = True
			continue

		# If landing on a cell with 0 mines in neighboring cells
		elif numbers[r][c] == 0:
			vis = []
			mine_values[r][c] = '0'
			neighbours(r, c)

		# If selecting a cell with atleast 1 mine in neighboring cells	
		else:	
			mine_values[r][c] = numbers[r][c]

		# Check for game completion	
		if(check_over()):
			show_mines()
			print_mines_layout()
			print("Congratulations!!! YOU WIN")
			over = True
			continue
		clear()