A Python Program to Print colors in shapes you like

def rectangle():
color = raw_input("Enter a color you like : ")
color1 = (color + " ")*10
color2 = color + (" "*(len(color1)-(2*(len(color))+1))) + color
print color1
print color2
print color2
print color1
def square():
color = raw_input("Enter a color you like : rectangle or square ")
color1 = (color + " ")*5
color2 = color + (" "*(len(color1)-(2*(len(color))+1))) + color
print color1
print color2
print color2
print color2
print color2
print color2
print color1
__diagram__ = raw_input("Enter the dia ")
if __diagram__=="rectangle":
rectangle()
elif __diagram__=="square":
square()