new edited code for mastermind

Author: vartika-arora

mastermind_code.py

@@ -0,0 +1,54 @@
+import random
+def mastermind():
+    colours = ["R", "G", "B", "Y", "W", "O"] #R-RED,G-GREEN,B-BLUE,Y-YELLOW,W-WHITE,O-ORANGE
+    attempts = 0
+    game = True
+    s=1
+    colour_code = random.sample(colours,4)
+    print (colour_code)
+    #player's guessing starts
+    print ("START GUESSING")	
+    while game:
+            correct_colour = ""
+            guessed_colour = ""
+            player_guess=""
+            player_guess=raw_input().upper()
+            attempts+= 1
+            if len(player_guess) != len(colour_code):
+                    print ("The secret code needs to have exactly four colors")
+                    continue
+            for i in range(4):
+                    if player_guess[i] not in colours:
+                            print ("invalid input")
+                            continue
+            if correct_colour != "bbbb":
+                    for i in range(4):
+                            if player_guess[i] == colour_code[i]:
+                                    correct_colour += "b"#black peg for correct colour and position
+                            if  player_guess[i] != colour_code[i] and player_guess[i] in colour_code:
+                                    guessed_colour += "w"# white peg for correct colour and incorrect position
+                    print (correct_colour +  guessed_colour + "\n")		
+            # matching player's input with original colour code                  
+            if correct_colour == "bbbb":
+                    if attempts == 1:
+                            print ("You guessed at the first attempt!")
+                    else:
+                            print ("Well done... You needed " + str(attempts) + " attempts to guess.")
+                    game = False
+            #player gets maximum of 5 attempts to guess the code                   
+            if attempts >= 1 and attempts <6 and correct_colour != "bbbb":
+                    print ("Next attempt: ")
+            elif attempts >= 6:
+                    print ("You didn't guess! The secret color code was: " + str(colour_code))	
+
+            #To play or not to play
+            while game == False:
+                    finish_game = raw_input("Do you want to play again (Y/N)?").upper()	
+                    attempts = 0
+                    if finish_game =="N":
+                            print ("Thanks for the game")
+                            break
+                    elif finish_game == "Y":
+                            game = True
+                            print ("Let's play again.Guess the secret code: ")
+mastermind()