import random
import drawzero as dz
dz.screen=(1000,1000)
dz.fill("white")
city=[]
print("Если вам нужен черно-белый город, введите 0. Если же вам нужен цветной горд, введите 1")
color_choice=int(input())
if color_choice==0:
    colors=['black']
elif color_choice==1:
    colors = ['red', 'green', 'blue', 'black', 'yellow', 'orange', 'brown']
else:
    print("Такого варианта ввода нет")
#генерация схематичного изображения города
for i in range(10):
    city.append([])
    for j in range(10):
        city[i].append(0)
xroad=random.randint(0,9)*100
yroad=random.randint(0,9)*100
#Создание дороги
for i in range(0, 1000, 75):
    dz.filled_polygon('black',yroad+50, i, yroad+60, i,yroad+60,i+30,yroad+50,i+30)
for i in range(0, 1000, 75):
    dz.filled_polygon('black',i,xroad+50 , i+30, xroad+50,i+30,xroad+60,i,xroad+60)
for i in range(0,10):
    city[xroad//100][i]=1
for i in range(0,10):
    city[i][yroad//100]=1
#создание рандомной крыши
def roof(x,y):
    r=random.randint(0,2)
    roof_color = random.choice(colors)
    if r==0:
        dz.filled_polygon(roof_color, x,y,x+50,y-30,x+100,y)
    elif r==1:
        dz.filled_polygon(roof_color, x, y, x + 32, y - 30, x + 68, y-30,x+100,y)
    elif r==2:
        dz.filled_polygon(roof_color,x,y,x+16,y-15,x+42,y-30,x+58,y-30,x+84,y-15,x+100,y)
#создание рандомного окна
def window(x,y):
    window_color=random.choice(colors)
    r = random.randint(0, 2)
    if r==0:
        dz.rect(window_color, (x, y-50), 50, 50)
        dz.line(window_color, (x+25, y-50), (x+25, y))
        dz.line(window_color, (x, y - 25), (x + 50, y - 25))
    elif r==1:
        dz.polygon(window_color,[(x, y), (x+25, y-50), (x+50, y)])
        dz.line(window_color,(x+12.5, y - 25), (x + 37.5, y - 25))
        dz.line(window_color, (x+25, y-50), (x + 25, y))
    elif r==2:
        dz.polygon(window_color, [(x, y-50), (x + 25, y), (x + 50, y-50)])
        dz.line(window_color, (x + 12.5, y - 25), (x + 37.5, y - 25))
        dz.line(window_color, (x + 25, y - 50), (x + 25, y))
#создание 1 этажного дома
def Dom1(a,b):
    x=a*100
    y=(b+1)*100
    roof(x, y-70)
    window(x + 25, y - 10)
    main_color=random.choice(colors)
    dz.polygon(main_color,x+3,y,x+97,y,x+97,y-70,x+3,y-70)
    city[b][a]=1
#создание 2 этажного дома
def Dom2(a,b):
    x = a * 100
    y = (b+1)* 100
    roof(x,y-170)
    window(x+25,y-20)
    window(x + 25, y - 105)
    main_color = random.choice(colors)
    dz.line(main_color,x,y-85,x+100,y-85)
    dz.polygon(main_color, x+3, y, x + 97, y, x + 97, y - 170, x+3, y - 170)
    city[b][a] = 1
    city[b-1][a]=1
#создание 3 этажного дома
def Dom3(a, b):
    x = a * 100
    y = (b + 1) * 100
    roof(x, y - 270)
    window(x + 25, y - 20)
    window(x + 25, y - 110)
    window(x+25,y-200)
    main_color = random.choice(colors)
    dz.line(main_color, x, y - 90, x + 100, y - 90)
    dz.line(main_color, x, y - 180, x + 100, y - 180)
    dz.polygon(main_color, x+3, y, x + 97, y, x + 97, y - 270, x+3, y - 270)
    city[b][a] = 1
    city[b - 1][a] = 1
    city[b - 2][a] = 1
#генерация рандомного города
for i in range(9,-1,-1):
    for j in range(9,-1,-1):
        if city[i][j]==0 and i>1:
            if city[i-1][j] ==0 and city[i-2][j]==0:
                floor=random.randint(0,2)
                if floor==0:
                    Dom1(j,i)
                elif floor==1:
                    Dom2(j,i)
                else:
                    Dom3(j,i)
            elif city[i-1][j] ==0 and city[i-2][j]==1:
                floor = random.randint(0, 1)
                if floor==0:
                    Dom1(j,i)
                elif floor==1:
                    Dom2(j,i)
            else:
                Dom1(j, i)
        elif city[i][j]==0 and i==1:
            if city[i-1][j] ==0:
                floor = random.randint(0, 1)
                if floor==0:
                    Dom1(j,i)
                elif floor==1:
                    Dom2(j,i)
            else:
                Dom1(j, i)
        elif city[i][j]==0 and i==0:
            Dom1(j, i)
dz.run()  # Run