import turtle
from random import *

wn = turtle.Screen()
wn.title("Pong en classe de Première")
wn.bgcolor("black")
wn.setup(width=800, height=600)
wn.tracer(0)

# Function
def paddle_a_up():
    y = paddle_a.ycor()
    y += 20
    paddle_a.sety(y)

def paddle_a_down():
    y = paddle_a.ycor()
    y -= 20
    paddle_a.sety(y)

def paddle_b_up():
    y = paddle_b.ycor()
    y += 20
    paddle_b.sety(y)

def paddle_b_down():
    y = paddle_b.ycor()
    y -= 20
    paddle_b.sety(y)

# Keyboard binding
wn.listen()
wn.onkeypress(paddle_a_up, "z")
wn.onkeypress(paddle_a_down, "s")
wn.onkeypress(paddle_b_up, "Up")
wn.onkeypress(paddle_b_down, "Down")

# Joueur A
paddle_a = turtle.Turtle()
paddle_a.speed(1000)
paddle_a.shape("square")
paddle_a.color("white")
paddle_a.shapesize(stretch_wid=5, stretch_len=1)
paddle_a.penup()
paddle_a.goto(-350, 0)

# Joueur B
paddle_b = turtle.Turtle()
paddle_b.speed(0)
paddle_b.shape("square")
paddle_b.color("white")
paddle_b.shapesize(stretch_wid=5, stretch_len=1)
paddle_b.penup()
paddle_b.goto(350, 0)

# Balle
colours = ["cyan", "purple", "white", "blue"]
shape = ["square", "square", "turtle", "classic"]
for i in range(0, 5):
    y = randint(-150, 150)
    G = [-0.15, 0.15]
    ball = turtle.Turtle()
    ball.speed(0)
    ball.shape(shape[randint(0, 3)])
    ball.color(colours[randint(0, 3)])
    ball.penup()
    ball.goto(0, y)
    ball.dx = G[randint(0, 1)]
    ball.dy = G[randint(0, 1)]

# Pen
pen = turtle.Turtle()
pen.speed(0)
pen.color("white")
pen.penup()
pen.hideturtle()
pen.goto(0, 260)
pen.write("Joueur A : 0     Joueur B : 0", align="center", font=("Courier", 18, "normal"))

# Score
score_a = 0
score_b = 0

# Main game loop
while(True):
    wn.update()

    # Move the ball
    ball.setx(ball.xcor() + ball.dx)
    ball.sety(ball.ycor() + ball.dy)

    # Border checking
    if(ball.ycor() > 290):
        ball.sety(290)
        ball.dy *= -1

    if(ball.ycor() < -290):
        ball.sety(-290)
        ball.dy *= -1

    if(ball.xcor() > 390):
        ball.dx *= -1
        score_a += 1
        pen.clear()
        pen.write("Joueur A : " + str(score_a) + "     Joueur B : " + str(score_b), align="center", font=("Courier", 18, "normal"))

    if(ball.xcor() < -390):
        ball.dx *= -1
        score_b += 1
        pen.clear()
        pen.write("Joueur A : " + str(score_a) + "     Joueur B : " + str(score_b), align="center", font=("Courier", 18, "normal"))

    # Paddle and ball collisions
    if(ball.xcor() > 340 and ball.xcor() < 350 and ball.ycor() < paddle_b.ycor() + 40 and ball.ycor() > paddle_b.ycor() - 40):
        ball.dx *= -1

    if(ball.xcor() < -340 and ball.xcor() > -350 and ball.ycor() < paddle_a.ycor() + 40 and ball.ycor() > paddle_a.ycor() - 40):
        ball.dx *= -1