SZI-Smieciarka/modele.py

import pygame

# This sets the WIDTH and HEIGHT of each grid location
WIDTH = 60
HEIGHT = 60

# This sets the margin between each cell
MARGIN = 5


class Smieciarka(pygame.sprite.Sprite):
    def __init__(self, x, y):
        self.x = x
        self.y = y
        self.image = pygame.image.load('resources/smieciarka.png')
        self.ruch = 0
        pygame.sprite.Sprite.__init__(self)
        self.rect = pygame.Rect(self.x * WIDTH + MARGIN * self.x + MARGIN, self.y * HEIGHT + MARGIN * self.y, WIDTH,
                                HEIGHT)

    def w_lewo(self):
        if self.x > 0:
            self.x -= 1
            self.rect.x = MARGIN + self.x * WIDTH + self.x * MARGIN
            if self.ruch == 2:
                self.image = pygame.image.load('resources/smieciarka.png')
            self.ruch = 1

    def w_prawo(self):
        if self.x < 14:
            self.x += 1
            self.rect.x = MARGIN + self.x * WIDTH + self.x * MARGIN
            if self.ruch == 1:
                self.image = pygame.transform.flip(self.image, True, False)
            self.ruch = 2

    def w_gore(self):
        if self.y > 0:
            self.y -= 1
            self.rect.y = self.y * HEIGHT + self.y * MARGIN

    def w_dol(self):
        if self.y < 14:
            self.y += 1
            self.rect.y = self.y * HEIGHT + self.y * MARGIN


class Kontener(pygame.sprite.Sprite):
    def __init__(self, x, y):
        self.x = x
        self.y = y
        pygame.sprite.Sprite.__init__(self)
        self.image = pygame.image.__init__()
        self.rect = pygame.Rect(self.x * WIDTH + MARGIN * self.x + MARGIN, self.y * HEIGHT + MARGIN * self.y + MARGIN,
                                WIDTH, HEIGHT)


class Kratka(pygame.sprite.Sprite):
    def __init__(self, poz_x, poz_y):
        self.pozX = poz_x
        self.pozY = poz_y
        pygame.sprite.Sprite.__init__(self)
        self.image = pygame.image.__init__()
        self.rect = pygame.Rect(self.pozX * WIDTH + MARGIN * self.pozX + MARGIN, self.pozY * HEIGHT + MARGIN * self.pozY + MARGIN,
                                WIDTH, HEIGHT)

    def setImage(self, image):
        self.image = image