pong/paddle.py at main · piekarskipawel/pong · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
import pygame
from settings import Settings

class PaddleOne(pygame.sprite.Sprite):
    def __init__(self):
        super().__init__()

        self.settings = Settings()
        self.image = pygame.Surface((20, 100))
        self.image.fill(self.settings.paddle_color)
        self.rect = self.image.get_rect()
        self.rect.center = (50, self.settings.screen_height // 2)

    def update(self):

        keys = pygame.key.get_pressed()
        if keys[pygame.K_w] and self.rect.top > 0:
            self.rect.y -= self.settings.speed_paddle_one
        if keys[pygame.K_s] and self.rect.bottom < self.settings.screen_height:
            self.rect.y += self.settings.speed_paddle_one

    def reset_paddle_one(self):
        self.settings = Settings()
        self.image = pygame.Surface((20, 100))
        self.image.fill(self.settings.paddle_color)
        self.rect = self.image.get_rect()
        self.rect.center = (50, self.settings.screen_height // 2)


class PaddleTwo(pygame.sprite.Sprite):
    def __init__(self):
        super().__init__()

        self.settings = Settings()
        self.image = pygame.Surface((20, 100))
        self.image.fill(self.settings.paddle_color)
        self.rect = self.image.get_rect()
        self.rect.center = (self.settings.screen_width - 50, self.settings.screen_height // 2)

    def update(self):
        keys = pygame.key.get_pressed()
        if keys[pygame.K_UP] and self.rect.top > 0:
            self.rect.y -= self.settings.speed_paddle_two
        if keys[pygame.K_DOWN] and self.rect.bottom < self.settings.screen_height:
            self.rect.y += self.settings.speed_paddle_two

    def reset_paddle_two(self):
        self.settings = Settings()
        self.image = pygame.Surface((20, 100))
        self.image.fill(self.settings.paddle_color)
        self.rect = self.image.get_rect()
        self.rect.center = (self.settings.screen_width - 50, self.settings.screen_height // 2)