task3_sample_solutions.py

# Skeleton Program code for the AQA COMP1 Summer 2015 examination
# this code should be used in conjunction with the Preliminary Material
# written by the AQA COMP1 Programmer Team
# developed in the Python 3.4 programming environment

from datetime import *
import pickle

BOARDDIMENSION = 8
KASHSHAPTU = False

def CreateBoard():
  Board = []
  for Count in range(BOARDDIMENSION + 1):
    Board.append([])
    for Count2 in range(BOARDDIMENSION + 1):
      Board[Count].append("  ")
  return Board

def DisplayWhoseTurnItIs(WhoseTurn):
  if WhoseTurn == "W":
    print("It is White's turn")
  else:
    print("It is Black's turn")

def GetTypeOfGame():
  #Task 1 - amended to valid input
  valid = False
  while not valid:
    TypeOfGame = input("Do you want to play the sample game (enter Y for Yes)? ")
    TypeOfGame = TypeOfGame.upper()[0]
    if TypeOfGame in ["Y","N"]:
      valid = True
    else:
      print("Please enter Y or N")
  return TypeOfGame

def DisplayWinner(WhoseTurn,surrender,moves):
  print()
  if WhoseTurn == "W":
    moves = moves // 2 + 1
  else:
    moves = moves // 2
  if WhoseTurn == "W" and not surrender:
    winner = "White"
    print("Black's Sarrum has been captured in {0} moves.  White wins!".format(moves))
  elif WhoseTurn == "W" and surrender:
    winner = "White"
    print("Black surrendered. White wins!")
  elif WhoseTurn == "B" and not surrender:
    winner = "Black"
    print("White's Sarrum has been captured in {0} moves.  Black wins!".format(moves))
  elif WhoseTurn == "B" and surrender:
    winner = "Black"
    print("White surrendered. Black wins!")
  print()
  return moves

def CheckIfGameWillBeWon(Board, FinishRank, FinishFile):
  if Board[FinishRank][FinishFile][1] == "S":
    return True
  else:
    return False

def DisplayBoard(Board):
  print()
  for RankNo in range(1, BOARDDIMENSION + 1):
    print("{0:<5}{1}".format("","-"*25))
    print("R{0:<1}".format(RankNo),end="   ")
    for FileNo in range(1, BOARDDIMENSION + 1):
      print("|" + Board[RankNo][FileNo], end="")
    print("|")
  print("{0:<5}{1}".format("","-"*25))
  print()
  print("{0:<6}{1:<3}{2:<3}{3:<3}{4:<3}{5:<3}{6:<3}{7:<3}{8:<3}".format("","F1","F2","F3","F4","F5","F6","F7","F8"))
  print()
  print()

def CheckRedumMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile, ColourOfPiece):
  CheckRedumMoveIsLegal = False
  if ColourOfPiece == "W":
    if FinishRank == StartRank - 1:
      if FinishFile == StartFile and Board[FinishRank][FinishFile] == "  ":
        CheckRedumMoveIsLegal = True
      elif abs(FinishFile - StartFile) == 1 and Board[FinishRank][FinishFile][0] == "B":
        CheckRedumMoveIsLegal = True
    #task 17 - two space opening move
    elif FinishRank == StartRank - 2 and StartRank == 7:
      if FinishFile == StartFile and Board[FinishRank][FinishFile] == "  ":
        CheckRedumMoveIsLegal = True
  elif FinishRank == StartRank + 1:
    if FinishFile == StartFile and Board[FinishRank][FinishFile] == "  ":
      CheckRedumMoveIsLegal = True
    elif abs(FinishFile - StartFile) == 1 and Board[FinishRank][FinishFile][0] == "W":
      CheckRedumMoveIsLegal = True
  #task 17 - two space opening move
  elif FinishRank == StartRank + 2 and StartRank == 2:
    if FinishFile == StartFile and Board[FinishRank][FinishFile] == "  ":
      CheckRedumMoveIsLegal = True
  return CheckRedumMoveIsLegal

def CheckSarrumMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile):
  CheckSarrumMoveIsLegal = False
  if abs(FinishFile - StartFile) <= 1 and abs(FinishRank - StartRank) <= 1:
    CheckSarrumMoveIsLegal = True
  return CheckSarrumMoveIsLegal

def CheckGisgigirMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile):
  GisgigirMoveIsLegal = False
  RankDifference = FinishRank - StartRank
  FileDifference = FinishFile - StartFile
  if RankDifference == 0:
    if FileDifference >= 1:
      GisgigirMoveIsLegal = True
      for Count in range(1, FileDifference):
        if Board[StartRank][StartFile + Count] != "  ":
          GisgigirMoveIsLegal = False
    elif FileDifference <= -1:
      GisgigirMoveIsLegal = True
      for Count in range(-1, FileDifference, -1):
        if Board[StartRank][StartFile + Count] != "  ":
          GisgigirMoveIsLegal = False
  elif FileDifference == 0:
    if RankDifference >= 1:
      GisgigirMoveIsLegal = True
      for Count in range(1, RankDifference):
        if Board[StartRank + Count][StartFile] != "  ":
          GisgigirMoveIsLegal = False
    elif RankDifference <= -1:
      GisgigirMoveIsLegal = True
      for Count in range(-1, RankDifference, -1):
        if Board[StartRank + Count][StartFile] != "  ":
          GisgigirMoveIsLegal = False
  return GisgigirMoveIsLegal

def CheckNabuMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile):
  CheckNabuMoveIsLegal = False
  FileDifference = FinishFile - StartFile
  RankDifference = FinishRank - StartRank
  if abs(FileDifference) == abs(RankDifference):
      #moving north east
      CheckNabuMoveIsLegal = True
      if RankDifference > 0 and FileDifference < 0:
          for count in range(1,RankDifference):
              if Board[StartRank+count][StartFile-count] != "  ":
                  CheckNabuMoveIsLegal = False
      #moving south east
      elif RankDifference > 0 and FileDifference > 0:
          for count in range(1,RankDifference):
              if Board[StartRank+count][StartFile+count] != "  ":
                  CheckNabuMoveIsLegal = False
      #moving north west
      elif RankDifference < 0 and FileDifference < 0:
          for count in range(1,RankDifference):
              if Board[StartRank-count][StartFile-count] != "  ":
                  CheckNabuMoveIsLegal = False
      #moving south west
      elif RankDifference < 0 and FileDifference > 0:
          for count in range(1,RankDifference):
              if Board[StartRank-count][StartFile+count] != "  ":
                  CheckNabuMoveIsLegal = False
  return CheckNabuMoveIsLegal

def CheckMarzazPaniMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile):
  CheckMarzazPaniMoveIsLegal = False
  #task 15 - add diagonal moves
  if (abs(FinishFile - StartFile) == 1 and abs(FinishRank - StartRank) == 0) or (abs(FinishFile - StartFile) == 0 and abs(FinishRank - StartRank) ==1) or (abs(FinishFile - StartFile) == 1 and abs(FinishRank - StartRank) == 1):
    CheckMarzazPaniMoveIsLegal = True
  return CheckMarzazPaniMoveIsLegal

def CheckEtluMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile):
  CheckEtluMoveIsLegal = False
  #task 17 - make knight
  if (abs(FinishFile - StartFile) == 2 and abs(FinishRank - StartRank) == 1) or (abs(FinishFile - StartFile) == 1 and abs(FinishRank - StartRank) == 2):
    CheckEtluMoveIsLegal = True
  return CheckEtluMoveIsLegal

def CheckKashshaptuMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile):
  CheckKashshaptuMoveIsLegal = False
  if CheckEtluMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile):
    CheckKashshaptuMoveIsLegal = True
  elif CheckMarzazPaniMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile):
    CheckKashshaptuMoveIsLegal = True
  elif CheckNabuMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile):
    CheckKashshaptuMoveIsLegal = True
  elif CheckGisgigirMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile):
    CheckKashshaptuMoveIsLegal = True
  elif CheckSarrumMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile):
    CheckKashshaptuMoveIsLegal = True
  elif CheckRedumMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile):
    CheckKashshaptuMoveIsLegal = True
  return CheckKashshaptuMoveIsLegal


def CheckMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile, WhoseTurn):
  MoveIsLegal = True
  if (FinishFile == StartFile) and (FinishRank == StartRank):
    MoveIsLegal = False
  #Task 2 - prevent moves outside the board area
  elif StartRank in [0,BOARDDIMENSION+1]:
    MoveIsLegal = False
  elif FinishRank in [0,BOARDDIMENSION+1]:
    MoveIsLegal = False
  elif StartFile in [0,BOARDDIMENSION+1]:
    MoveIsLegal = False
  elif FinishFile in [0,BOARDDIMENSION+1]:
    MoveIsLegal = False
  else:
    PieceType = Board[StartRank][StartFile][1]
    PieceColour = Board[StartRank][StartFile][0]
    if WhoseTurn == "W":
      if PieceColour != "W":
        MoveIsLegal = False
      if Board[FinishRank][FinishFile][0] == "W":
        MoveIsLegal = False
    else:
      if PieceColour != "B":
        MoveIsLegal = False
      if Board[FinishRank][FinishFile][0] == "B":
        MoveIsLegal = False
    if MoveIsLegal == True:
      if PieceType == "R":
        MoveIsLegal = CheckRedumMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile, PieceColour)
      elif PieceType == "S":
        MoveIsLegal = CheckSarrumMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile)
      elif PieceType == "M":
        MoveIsLegal = CheckMarzazPaniMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile)
      elif PieceType == "G":
        MoveIsLegal = CheckGisgigirMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile)
      elif PieceType == "N":
        MoveIsLegal = CheckNabuMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile)
      elif PieceType == "E":
        MoveIsLegal = CheckEtluMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile)
      elif PieceType == "K":
        MoveIsLegal = CheckKashshaptuMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile)
  return MoveIsLegal

def InitialiseBoard(Board,SampleGame):
  if SampleGame == "Y":
    InitialiseSampleBoard(Board)
  else:
    InitialiseNewBoard(Board)


def InitialiseNewBoard(Board):
  for RankNo in range(1, BOARDDIMENSION + 1):
    for FileNo in range(1, BOARDDIMENSION + 1):
      if RankNo == 2:
        Board[RankNo][FileNo] = "BR"
      elif RankNo == 7:
        Board[RankNo][FileNo] = "WR"
      elif RankNo == 1 or RankNo == 8:
        if RankNo == 1:
          Board[RankNo][FileNo] = "B"
        if RankNo == 8:
          Board[RankNo][FileNo] = "W"
        if FileNo == 1 or FileNo == 8:
          Board[RankNo][FileNo] = Board[RankNo][FileNo] + "G"
        elif FileNo == 2 or FileNo == 7:
          Board[RankNo][FileNo] = Board[RankNo][FileNo] + "E"
        elif FileNo == 3 or FileNo == 6:
          Board[RankNo][FileNo] = Board[RankNo][FileNo] + "N"
        elif (FileNo == 4 and RankNo == 8) or (FileNo == 5 and RankNo == 1):
          Board[RankNo][FileNo] = Board[RankNo][FileNo] + "M"
        elif (FileNo == 5 and RankNo == 8) or (FileNo == 4 and RankNo == 1):
          Board[RankNo][FileNo] = Board[RankNo][FileNo] + "S"
      else:
        Board[RankNo][FileNo] = "  "

def InitialiseSampleBoard(Board):
  for RankNo in range(1, BOARDDIMENSION + 1):
    for FileNo in range(1, BOARDDIMENSION + 1):
      Board[RankNo][FileNo] = "  "
  Board[1][2] = "BG"
  Board[1][4] = "BS"
  Board[1][8] = "WG"
  Board[2][1] = "WR"
  Board[3][1] = "WS"
  Board[3][2] = "BE"
  Board[3][8] = "BE"
  Board[6][8] = "BR"


def GetMove(StartSquare, FinishSquare,board_state):
  #task 3 - validate moves
  valid = False
  surrender = False
  while not valid:
    try:
      StartSquare = int(input("Enter coordinates of square containing piece to move (file first) or type '-1' for menu: "))
      if StartSquare == -1:
        valid,surrender = present_in_game_menu(board_state)
      else:
        if StartSquare // 10 > 0:
          valid = True
        else:
          print("Please provide both FILE and RANK for this move")
    except ValueError:
      print("Please provide both FILE and RANK for this move")
  if StartSquare != -1:
    valid = False
    while not valid:
      try:
        FinishSquare = int(input("Enter coordinates of square to move piece to (file first) or type '-1' for menu: "))
        if FinishSquare == -1:
          valid,surrender = present_in_game_menu()
        else:
          if FinishSquare // 10 > 0:
            valid = True
          else:
            print("Please provide both FILE and RANK for this move")
      except ValueError:
        print("Please provide both FILE and RANK for this move")
  else:
    FinishSquare = -1
  return StartSquare, FinishSquare,surrender

#task 4 - move confirmation
def ConfirmMove(StartSquare, FinishSquare):
  StartRank = StartSquare % 10
  StartFile = StartSquare // 10
  FinishRank = FinishSquare % 10
  FinishFile = FinishSquare // 10
  print()
  print("Move from Rank {0}, File {1} to Rank {2}, File {3}?".format(StartRank,StartFile,FinishRank, FinishFile))
  valid = False
  while not valid:
    confirm = input("Confirm move (Yes/No): ")
    confirm = confirm.upper()[0]
    if confirm in ["Y","N"]:
      valid = True
  if confirm == "Y":
    print("Move confirmed")
  else:
    print("Move cancelled")
    valid = False
  return valid

def MakeMove(Board, StartRank, StartFile, FinishRank, FinishFile, WhoseTurn):
  #Task 6 - promotion message
  if WhoseTurn == "W" and FinishRank == 1 and Board[StartRank][StartFile][1] == "R":
    if KASHSHAPTU:
      Board[FinishRank][FinishFile] = "WK"
    else:
      Board[FinishRank][FinishFile] = "WM"
    OriginalColour, OriginalPiece = GetPieceName(Board[StartRank][StartFile])
    PromotedColour, PromotedPiece = GetPieceName(Board[FinishRank][FinishFile])
    print()
    print("{0} {1} promoted to {2}.".format(OriginalColour, OriginalPiece, PromotedPiece))
    print()
    Board[StartRank][StartFile] = "  "
  elif WhoseTurn == "B" and FinishRank == 8 and Board[StartRank][StartFile][1] == "R":
    if KASHSHAPTU:
      Board[FinishRank][FinishFile] = "BK"
    else:
      Board[FinishRank][FinishFile] = "BM"
    OriginalColour, OriginalPiece = GetPieceName(Board[StartRank][StartFile])
    PromotedColour, PromotedPiece = GetPieceName(Board[FinishRank][FinishFile])
    print()
    print("{0} {1} promoted to {2}.".format(OriginalColour, OriginalPiece, PromotedPiece))
    print()
    Board[StartRank][StartFile] = "  "
  else:
    #Task 5 - indicate piece has been taken
    if Board[FinishRank][FinishFile] != "  ":
      print()
      TakeColour, TakePiece = GetPieceName(Board[StartRank][StartFile])
      TakenColour, TakenPiece = GetPieceName(Board[FinishRank][FinishFile])
      print("{0} {1} takes {2} {3}.".format(TakeColour, TakePiece, TakenColour, TakenPiece))
      print()
    Board[FinishRank][FinishFile] = Board[StartRank][StartFile]
    Board[StartRank][StartFile] = "  "

#Task 7 - Check whether Sarrum is in check
def CheckValidBoardPosition(Rank, File):
  valid = False
  if 1 <= Rank <= BOARDDIMENSION and 1 <= File <= BOARDDIMENSION:
    valid = True
  return valid

#Task 7 - Check whether Sarrum is in check
def CheckWithRedum(Board,FinishRank, FinishFile,WhoseTurn):
  #Redum can take diagonally FORWARD one space to LEFT and RIGHT
  if WhoseTurn == "W":
    Sarrum = "BS"
  elif WhoseTurn == "B":
    Sarrum = "WS"
  Check = False
  if WhoseTurn == "W":
    #diagonal left foward
    if CheckValidBoardPosition(FinishRank-1, FinishFile-1):
      if Board[FinishRank-1][FinishFile-1] == Sarrum:
        Check = True
    #diagonal right foward
    if CheckValidBoardPosition(FinishRank-1,FinishFile+1) and not Check:
      if Board[FinishRank-1][FinishFile+1] == Sarrum:
        Check = True
  elif WhoseTurn == "B":
    #diagonal left foward
    if CheckValidBoardPosition(FinishRank+1,FinishFile-1):
      if Board[FinishRank+1][FinishFile-1] == Sarrum:
        Check = True
    #diagonal right foward
    if CheckValidBoardPosition(FinishRank+1,FinishFile+1) and not Check:
      if Board[FinishRank+1][FinishFile+1] == Sarrum:
        Check = True
  return Check

#Task 7 - Check whether Sarrum is in check
def CheckWithNabu(Board,FinishRank, FinishFile,WhoseTurn):
  #Nabu can take diagonally one space in any direction
  if WhoseTurn == "W":
    Sarrum = "BS"
  elif WhoseTurn == "B":
    Sarrum = "WS"
  Check = False
  #diagonal left foward
  if CheckValidBoardPosition(FinishRank-1,FinishFile-1):
    if Board[FinishRank-1][FinishFile-1] == Sarrum:
      Check = True
  #diagonal right foward
  if CheckValidBoardPosition(FinishRank-1,FinishFile+1) and not Check:
    if Board[FinishRank-1][FinishFile+1] == Sarrum:
      Check = True
  #diagonal left back
  if CheckValidBoardPosition(FinishRank+1, FinishFile-1) and not Check:
    if Board[FinishRank+1][FinishFile-1] == Sarrum:
      Check = True
  #diagonal right back
  if CheckValidBoardPosition(FinishRank+1,FinishFile+1) and not Check:
    if Board[FinishRank+1][FinishFile+1] == Sarrum:
      Check = True
  return Check

#Task 7 - Check whether Sarrum is in check
def CheckWithMarzazPani(Board,FinishRank, FinishFile,WhoseTurn):
  #Marzaz Pani can take horizontally and vertically one space in any direction
  if WhoseTurn == "W":
    Sarrum = "BS"
  elif WhoseTurn == "B":
    Sarrum = "WS"
  Check = False
  #foward one vertical
  if CheckValidBoardPosition(FinishRank-1,FinishFile):
    if Board[FinishRank-1][FinishFile] == Sarrum:
      Check = True
  #back one vertical
  if CheckValidBoardPosition(FinishRank+1, FinishFile) and not Check:
    if Board[FinishRank+1][FinishFile] == Sarrum:
      Check = True
  #left one horizontal
  if CheckValidBoardPosition(FinishRank,FinishFile-1) and not Check:
    if Board[FinishRank][FinishFile-1] == Sarrum:
      Check = True
  #right one horizontal
  if CheckValidBoardPosition(FinishRank, FinishFile) and not Check:
    if Board[FinishRank][FinishFile+1] == Sarrum:
      Check = True
  return Check

#Task 7 - Check whether Sarrum is in check
def CheckWithEtlu(Board,FinishRank, FinishFile,WhoseTurn):
  #Etlu can take horizontally and vertically two spaces in any direction
  if WhoseTurn == "W":
    Sarrum = "BS"
  elif WhoseTurn == "B":
    Sarrum = "WS"
  Check = False
  #foward two vertical
  if CheckValidBoardPosition(FinishRank-2,FinishFile):
    if Board[FinishRank-2][FinishFile] == Sarrum:
      Check = True
  #back two vertical
  if CheckValidBoardPosition(FinishRank-2, FinishFile) and not Check:
    if Board[FinishRank-2][FinishFile] == Sarrum:
      Check = True
  #left two horizontal
  if CheckValidBoardPosition(FinishRank, FinishFile-2) and not Check:
    if Board[FinishRank][FinishFile-2] == Sarrum:
      Check = True
  #right two horizontal
  if CheckValidBoardPosition(FinishRank,FinishFile+2) and not Check:
    if Board[FinishRank][FinishFile+2] == Sarrum:
      Check = True
  return Check

#Task 7 - Check whether Sarrum is in check
def CheckWithGisgigir(Board,FinishRank, FinishFile,WhoseTurn):
  #Gisgigir can take horizontally and vertically any number of spaces in any direction
  if WhoseTurn == "W":
    Sarrum = "BS"
  elif WhoseTurn == "B":
    Sarrum = "WS"
  Check = False
  #check forward
  blocked = False
  for rank in range(FinishRank+1,BOARDDIMENSION+1):
    if CheckValidBoardPosition(rank,FinishFile):
      if Board[rank][FinishFile] == Sarrum and not blocked:
        Check = True
      elif Board[rank][FinishFile] != "  ":
        blocked = True
  #check backwards
  if not Check:
    blocked = False
    for rank in range(1,FinishRank):
      if CheckValidBoardPosition(rank,FinishFile):
        if Board[rank][FinishFile] == Sarrum and not blocked:
          Check = True
        elif Board[rank][FinishFile] != "  ":
          blocked = True
  #check right
  if not Check:
    blocked = False
    for file in range(FinishFile+1, BOARDDIMENSION+1):
      if CheckValidBoardPosition(FinishRank,file):
        if Board[FinishRank][file] == Sarrum and not blocked:
          Check = True
        elif Board[FinishRank][file] != "  ":
          blocked = True
  #check left
  if not Check:
    blocked = False
    for file in range(FinishFile-1, 1,-1):
      if CheckValidBoardPosition(FinishRank,file):
        if Board[FinishRank][file] == Sarrum and not blocked:
          Check = True
        elif Board[FinishRank][file] != "  ":
          blocked = True
  return Check

def CheckWithKashshaptu(Board,FinishRank, FinishFile,WhoseTurn):
  Check = False
  if CheckWithGisgigir(Board,FinishRank, FinishFile,WhoseTurn):
    Check = True
  elif CheckWithEtlu(Board,FinishRank, FinishFile,WhoseTurn):
    Check = True
  elif CheckWithMarzazPani(Board,FinishRank, FinishFile,WhoseTurn):
    Check = True
  elif CheckWithNabu(Board,FinishRank, FinishFile,WhoseTurn):
    Check = True
  elif CheckWithRedum(Board,FinishRank, FinishFile,WhoseTurn):
    Check = True
  return Check

#Task 7 - Check whether Sarrum is in check
def CheckSarrumInCheck(Board,FinishRank, FinishFile,WhoseTurn):
  PieceCode = Board[FinishRank][FinishFile]
  Colour, Piece = GetPieceName(PieceCode)
  print(Colour,Piece)
  Check = False
  if Piece == "Marzaz Pani":
    Check = CheckWithMarzazPani(Board,FinishRank, FinishFile,WhoseTurn)
  elif Piece == "Nabu":
    Check = CheckWithNabu(Board,FinishRank, FinishFile,WhoseTurn)
  elif Piece == "Etlu":
    Check = CheckWithEtlu(Board,FinishRank, FinishFile,WhoseTurn)
  elif Piece == "Gisgigir":
    Check = CheckWithGisgigir(Board,FinishRank, FinishFile,WhoseTurn)
  elif Piece == "Redum":
    Check = CheckWithRedum(Board,FinishRank, FinishFile,WhoseTurn)
  elif Piece == "Kashshaptu":
    Check = CheckWithKashshaptu(Board,FinishRank, FinishFile,WhoseTurn)
  return Check,PieceCode,FinishRank,FinishFile

#Task 7 - Check whether Sarrum is in check
def CheckMessage(PieceCode,FinishRank,FinishFile):
  Colour, Piece = GetPieceName(PieceCode)
  if Colour == "White":
    Sarrum = "Black"
  elif Colour == "Black":
    Sarrum = "White"
  print("{0} {1} in Rank {2}, File {3} has {4} Sarrum in Check.".format(Colour,Piece,FinishRank,FinishFile,Sarrum))


#Task 5 - new function (indicate piece has been taken)
def GetPieceName(PieceCode):
  Colour = PieceCode[0]
  Piece = PieceCode[1]
  if Colour == "W":
    Colour = "White"
  elif Colour == "B":
    Colour = "Black"
  if Piece == "S":
    Piece = "Sarrum"
  elif Piece == "M":
    Piece = "Marzaz Pani"
  elif Piece == "N":
    Piece = "Nabu"
  elif Piece == "E":
    Piece = "Etlu"
  elif Piece == "G":
    Piece = "Gisgigir"
  elif Piece == "R":
    Piece = "Redum"
  elif Piece == "K":
    Piece = "Kashshaptu"
  return Colour, Piece


def display_menu():
  print("Main Menu")
  print()
  print("1. Start new game")
  print("2. Load existing game")
  print("3. Play sample game")
  print("4. View high scores")
  print("5. Settings")
  print("6. Quit program")
  print()

def get_menu_selection():
  valid = False
  while not valid:
    try:
      selection = int(input("Please select an option: "))
      if selection >= 1 and selection <=6:
        valid = True
      else:
        print("Please select an option in range 1 to 6")
    except ValueError:
      print("Please input a valid value")
  return selection

def make_selection(selection,high_scores):
  exit_game = False
  if selection == 1:
    restore, board_state = restore_temp_state()
    if restore:
      play_game("N",high_scores,board_state)
    else:
      play_game("N",high_scores,None) #play game without sample
  elif selection == 2:
    board_state = load_board_state("board_state.dat")
    play_game("N",high_scores,board_state)
  elif selection == 3:
    restore, board_state = restore_temp_state()
    if restore:
      play_game("N",high_scores,board_state)
    else:
      play_game("Y",high_scores,None) #play game without sample
  elif selection == 4:
    sort_high_scores(high_scores)
    display_high_scores(high_scores)
  elif selection == 5:
    present_settings()
  elif selection == 6:
    exit_game = True
    save_high_scores(high_scores)
    print()
    print("Thank you for playing!")
    print()
  return exit_game

def play_game(SampleGame,high_scores,board_state):
  Board = CreateBoard() #0th index not used
  StartSquare = 0
  FinishSquare = 0
  PlayAgain = "Y"
  while PlayAgain == "Y":
    no_of_moves_made = 0
    WhoseTurn = "W"
    GameOver = False
    if board_state == None:
      InitialiseBoard(Board, SampleGame)
      Check = False
      PieceCode = None
      CheckRank = None
      CheckFile = None
    else:
      Board = board_state[0]
      WhoseTurn = board_state[1]
      Check = board_state[2]
      PieceCode = board_state[3]
      CheckRank = board_state[4]
      CheckFile = board_state[5]
      no_of_moves_made = board_state[6]
    while not(GameOver):
      DisplayBoard(Board)
      DisplayWhoseTurnItIs(WhoseTurn)
      MoveIsLegal = False
      board_state = [Board,WhoseTurn,Check,PieceCode,CheckRank,CheckFile,no_of_moves_made]
      while not(MoveIsLegal):
        StartSquare, FinishSquare,surrender = GetMove(StartSquare, FinishSquare,board_state)
        if StartSquare != -1 and FinishSquare != -1:
          Confirm = ConfirmMove(StartSquare, FinishSquare)
          if Confirm:
            StartRank = StartSquare % 10
            StartFile = StartSquare // 10
            FinishRank = FinishSquare % 10
            FinishFile = FinishSquare // 10
            MoveIsLegal = CheckMoveIsLegal(Board, StartRank, StartFile, FinishRank, FinishFile, WhoseTurn)
            if Check:
              MoveSquareContents = Board[FinishRank][FinishFile]
            print()
            if not(MoveIsLegal):
              print("That is not a legal move - please try again")
            if MoveIsLegal:
              GameOver = CheckIfGameWillBeWon(Board, FinishRank, FinishFile)
              MakeMove(Board, StartRank, StartFile, FinishRank, FinishFile, WhoseTurn)
              no_of_moves_made += 1 #task 23
            if MoveIsLegal and Check:
              Colour, Piece = GetPieceName(PieceCode)
              if WhoseTurn == "W":
                CheckTurn = "B"
              else:
                CheckTurn = "W"
              #check to see if check piece has been taken
              CheckPiece = Board[CheckRank][CheckFile]
              print("Check piece: {0}".format(CheckPiece))
              print("Piece code: {0}".format(PieceCode))
              if CheckPiece == PieceCode:
                print("hello")
                Check,PieceCode,CheckRank,CheckFile = CheckSarrumInCheck(Board,CheckRank, CheckFile,CheckTurn)
              else:
                Check = False
              #if still in check then move must be prevented
              if Check:
                print("Sarrum must be moved out of check from {0} {1} in Rank {2}, File {3}".format(Colour, Piece, CheckRank, CheckFile))
                MoveIsLegal = False
                #rewind move
                Board[StartRank][StartFile] = Board[FinishRank][FinishFile]
                Board[FinishRank][FinishFile] = MoveSquareContents
                print("Move cancelled")
            print()
        else:
          MoveIsLegal = True
          PlayAgain = "N"
          GameOver = True
          if surrender:
            winner = WhoseTurn
            if WhoseTurn == "W":
              winning_no_moves = DisplayWinner("B",surrender,no_of_moves_made)
            else:
              winning_no_moves = DisplayWinner("W",surrender,no_of_moves_made)
      if PlayAgain == "Y":
        #GameOver = CheckIfGameWillBeWon(Board, FinishRank, FinishFile)
        #MakeMove(Board, StartRank, StartFile, FinishRank, FinishFile, WhoseTurn)
        Check,PieceCode,CheckRank,CheckFile = CheckSarrumInCheck(Board,FinishRank, FinishFile,WhoseTurn)
        print("Sarrum in Check = {0} by {1} in {2},{3}".format(Check,PieceCode,CheckRank,CheckFile))
        if Check:
          CheckMessage(PieceCode,CheckRank,CheckFile)
        if GameOver:
          winner = WhoseTurn
          winning_no_moves = DisplayWinner(WhoseTurn,False,no_of_moves_made)
        if WhoseTurn == "W":
          WhoseTurn = "B"
        else:
          WhoseTurn = "W"
    if PlayAgain == "Y":
      present_high_score_option(winner,winning_no_moves,high_scores)
      PlayAgain = input("Do you want to play again (enter Y for Yes)? ")
      if ord(PlayAgain) >= 97 and ord(PlayAgain) <= 122:
        PlayAgain = chr(ord(PlayAgain) - 32)

def display_in_game_menu():
  print()
  print("Options")
  print()
  print("1. Save Game")
  print("2. Quit to Menu")
  print("3. Return to Game")
  print("4. Surrender")
  print()

def get_game_option():
  valid = False
  while not valid:
    try:
      choice = int(input("Please select an option: "))
      if choice >= 1 and choice <= 4:
        valid = True
      else:
        print("Please enter a valid choice")
    except ValueError:
      print("Please enter a valid choice")
    print()
  return choice

def present_in_game_menu(board_state):
  display_in_game_menu()
  choice = get_game_option()
  valid = False
  surrender = False
  if choice == 1:
    print("Game saved...")
    save_board_state(board_state,False)
  elif choice == 2:
    print("Exiting to menu...")
    save_board_state(board_state,True)
    valid = True
  elif choice == 3:
    print("Returning to game...")
  elif choice == 4:
    surrender = True
    print("Surrendering...")
    valid = True
  return valid, surrender

def display_settings():
  print()
  print("1. Use Kashshaptu Piece")
  print("9. Return to Main Menu")
  print()

def get_setting_option():
  valid = False
  while not valid:
    try:
      choice = int(input("Please select setting to change: "))
      if choice in [1,9]:
        valid = True
      else:
        print("Please enter an appropriate value")
    except ValueError:
      print()
      print("Please enter an appropriate value")
      print()
  return choice

def present_settings():
  display_settings()
  choice = get_setting_option()
  if choice == 1:
    change_kashshaptu_setting()
  elif choice == 9:
    print()
    print("Returning to main menu...")
    print()

def change_kashshaptu_setting():
  global KASHSHAPTU
  valid = False
  while not valid:
    choice = input("Do you wish to use the Kashshaptu piece (Y/N)? ")
    choice = choice.lower()[0]
    if choice == "y":
      KASHSHAPTU = True
      print("Kashshaptu Activated")
      valid = True
    elif choice == "n":
      KASHSHAPTU = False
      print("Kashshaptu Deactivated")
      valid = True
    else:
      print("Please provide an appropriate response")
  print()

def present_high_score_option(winner, moves,high_scores):
  add_score = high_score_choice(winner)
  if add_score:
    name = get_high_score_name()
    high_scores.append([name,winner,moves,datetime.now()])

def high_score_choice(winner):
  valid = False
  add = False
  print()
  print("Well done {0}!".format(winner))
  while not valid:
    choice = input("Do you want to add your name to the high score list (Y/N): ")
    choice = choice.lower()[0]
    if choice == "y":
      add = True
      valid = True
    elif choice == "n":
      valid = True
    else:
      print("Please enter an appropriate value - y or n")
  print()
  return add

def get_high_score_name():
  valid = False
  print()
  while not valid:
    name = input("Please enter your name for the high score table: ")
    if 0 < len(name) <= 10:
      valid = True
    else:
      print("Please enter something for your name (up to 10 characters).")
  print()
  return name

def display_high_scores(scores):
  print()
  print("High Scores")
  print()
  if len(scores) > 0:
    print("-"*46)
    print("|{0:<10}|{1:<6}|{2:<15}|{3:<10}|".format("Name","Colour","Number of Moves","Date"))
    print("-"*46)
    for score in scores:
      date = datetime.strftime(score[3],"%d/%m/%Y")
      print("|{0:<10}|{1:<6}|{2:<15}|{3}|".format(score[0],score[1],score[2],date))
      print("-"*46)
  else:
    print("There are no scores")
  print()

def save_high_scores(scores):
  with open("high_scores.dat",mode="wb") as score_file:
    pickle.dump(scores,score_file)

def load_high_scores():
  with open("high_scores.dat",mode="rb") as score_file:
    scores = pickle.load(score_file)
  return scores

def sort_high_scores(scores):
  swaps = True
  length = len(scores)
  while swaps:
    swaps = False
    length = length - 1
    for index in range(length):
      if scores[index][2] > scores [index+1][2]:
        temp = scores[index+1]
        scores[index+1] = scores[index]
        scores[index] = temp
        swaps = True

def save_board_state(board_state,temp):
  if temp:
    file_name = "temp_board_state.dat"
  else:
    file_name = "board_state.dat"
  with open(file_name, mode="wb") as board_file:
    pickle.dump(board_state,board_file)

def load_board_state(file_name):
  with open(file_name, mode="rb") as board_file:
    board_state = pickle.load(board_file)
  return board_state

def restore_temp_state():
  try:
    board_state = load_board_state("temp_board_state.dat")
    print("A previous game was found...")
    valid = False
    while not valid:
      choice = input("Do you wish to continue (Y or N): ")
      choice = choice.lower()[0]
      if choice == "y":
        restore = True
        valid = True
      elif choice == "n":
        restore = False
        valid = True
      else:
        print("Please enter a valid option")
        print()
  except FileNotFoundError:
    restore = False
    board_state = None
  return restore, board_state



if __name__ == "__main__":
  exit_game = False
  try:
    high_scores = load_high_scores()
  except FileNotFoundError:
    high_scores = []
  while not exit_game:
    display_menu()
    selection = get_menu_selection()
    exit_game = make_selection(selection,high_scores)