import pygame

pygame.init()

class Slab:

def __init__(self, win, size, pos, player, minPos, maxPos):

self.win = win

self.size = size

self.pos = pos

self.player = player # player = 1 or 2

self.minPos = minPos

self.maxPos = maxPos

def draw(self):

pygame.draw.rect(

self.win,

(255, 255, 255),

(self.pos[0], self.pos[1], self.size[0], self.size[1]),

)

def getCoords(self):

return [

self.pos[0],

self.pos[1],

self.pos[0] + self.size[0],

self.pos[1] + self.size[1],

]

def updatePos(self):

keys = pygame.key.get_pressed()

if self.player == 1:

if keys[pygame.K_UP] and self.getCoords()[1] > self.minPos[1]:

self.pos[1] -= 0.3

if keys[pygame.K_DOWN] and self.getCoords()[3] < self.maxPos[1]:

self.pos[1] += 0.3

else:

if keys[pygame.K_w] and self.getCoords()[1] > self.minPos[1]:

self.pos[1] -= 0.3

if keys[pygame.K_s] and self.getCoords()[3] < self.maxPos[1]:

self.pos[1] += 0.3