import pygame
import math
pygame.init()

pygame.font.SysFont('arial', 36)
sc = pygame.display.set_mode((1000, 600))
running = True

class Wall():
    def __init__(self, x, y, len1, len2, sdvig):
        self.x = x
        self.y = y
        self.len1 = len1
        self.len2 = len2
        self.sdvig = sdvig
        self.color = (0, 125, 255)

    def move(self):
        self.y += self.sdvig

    def draw1(self,sc):
        pygame.draw.rect(sc, (128, 0, 128), (0, 0, 12, 600))
        pygame.draw.rect(sc, (128, 0, 128), (988, 0, 12, 600))

    def draw(self, screen):
        pygame.draw.rect(screen, self.color, (self.x, self.y, self.len1, self.len2))


class Ball():
    def __init__(self, x, y, rad, sdvigx, sdvigy):
        self.x = x
        self.y = y
        self.rad = rad
        self.sdvigx = sdvigx
        self.sdvigy = sdvigy
        self.x1 = 0
        self.x2 = 0
        self.color = (255, 0, 0)

    def move(self, pl1, pl2):
        self.x = self.x + self.sdvigx
        self.y = self.y + self.sdvigy
        if (self.x + self.rad >= 1000) or (self.x + self.rad <= 0):
            self.sdvigx *= -1
        if (self.y + self.rad >= 600) or (self.y + self.rad <= 0):
            self.sdvigy *= -1
        if (self.x - 10 <= pl1.x + 25 and self.y >= pl1.y and self.y < pl1.y + 200):
            self.sdvigx *= -1
        if (self.x + 10 >= pl2.x and self.y >= pl2.y and self.y <= pl2.y + 200):
            self.sdvigx *= -1
        if self.x <= 10:
            self.x2 += 1
            self.x = 500
            self.y = 300
        if self.x >= 990:
            self.x1 += 1
            self.x = 500
            self.y = 30

    def draw(self, screen):
        pygame.draw.circle(screen, self.color, (self.x, self.y), self.rad)

clock = pygame.time.Clock()

b = []
player1 = Wall(0, 80, 25, 200, 0)
player2 = Wall(975, 80, 25, 200, 0)

while running:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
        if event.type == pygame.MOUSEBUTTONDOWN:
            if event.button == 1:
                b.append(Ball(pygame.mouse.get_pos()[0], pygame.mouse.get_pos()[1], 10, -20, -20))
        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_UP:
                if player2.y > 10:
                    player2.sdvig = -30
                    player2.move()
            if event.key == pygame.K_DOWN:
                if player2.y < 390:
                    player2.sdvig = 30
                    player2.move()
            if event.key == pygame.K_w:
                if player1.y > 10:
                    player1.sdvig = -30
                    player1.move()
            if event.key == pygame.K_s:
                if player1.y < 390:
                    player1.sdvig = 30
                    player1.move()

    sc.fill((0, 0, 0))

    player1.draw1(sc)
    for i in b:
        i.draw(sc)
        i.move(player1, player2)
        player1.draw(sc)
        player2.draw(sc)
        font = pygame.font.Font(None, 25)
        text = font.render(str(b[0].x1) + " : " + str(b[0].x2), 1, (0, 100, 0))
        place = text.get_rect(center=(500, 30))
        sc.blit(text, place)
    pygame.display.flip()
    pygame.time.wait(90)