DrawTiles.py

import pygame as pg
import pygame.mouse

from pygame import Vector2
from Settings import *
from main import *
from Tools import *

sprites = pg.sprite.Group()


class Tile(pg.sprite.Sprite):
    def __init__(self, pos, size, color):
        super().__init__()
        self.pos = pos
        self.size = size
        self.color = color

        self.image = pg.Surface((size.x, size.y))
        self.image.fill(self.color)
        self.rect = self.image.get_rect(center=self.pos)

        for tile in sprites.sprites():
            if isinstance(tile, Tile):
                if tile.pos == self.pos:
                    sprites.remove(tile)# noqa

    def preview_col(self, selected_color):
        self.image.fill(selected_color)
        self.rect = self.image.get_rect(center=self.pos)
        self.image.set_alpha(100)


class Tiles:
    def __init__(self):
        self.selected_color = "yellow"
        self.size = Vector2(WIDTH/NUMBER_OF_UNITS, HEIGHT/NUMBER_OF_UNITS)
        self.cords = []
        self.coordinates()
        self.previewer = Tile((0, 0), self.size, self.selected_color)
        self.pen_type = "pen"
        self.pen_size = 1

    def coordinates(self):
        for i in range(1, NUMBER_OF_UNITS+1):
            for j in range(1, NUMBER_OF_UNITS+1):
                self.cords.append(Vector2((self.size.x * i) - (self.size.x/2), (self.size.y * j) - (self.size.y/2)))
                #self.add_tiles() # noqa

    def add_tiles(self):
        for i in self.cords:
            sprites.add(Tile(i, self.size, (255, 255, 255)))# noqa

    def paint_tiles(self, mouse_pos, redraw_background, background):
        self.previewer.preview_col(self.selected_color)
        for i in self.cords:
            if abs(i.x-mouse_pos.x) < (self.size.x/2) * self.pen_size and abs(i.y-mouse_pos.y) < (self.size.y/2) * self.pen_size and mouse_pos.x < WIDTH and mouse_pos.y < HEIGHT:
                self.previewer.pos = i
                if pygame.mouse.get_pressed()[0]:
                    if self.pen_type == "pen":
                        sprites.add(Tile(i, self.size, self.selected_color))# noqa
                    elif self.pen_type == "rubber":
                        for sprite in sprites.sprites():
                            if isinstance(sprite, Tile):
                                if sprite.pos == i:
                                    sprites.remove(sprite)# noqa
                                    redraw_background()
                                    sprites.draw(background)
                    elif self.pen_type == "bucket":
                        color = ""
                        for sprite in sprites.sprites():
                            if isinstance(sprite, Tile):
                                if sprite.pos == i:
                                    color = sprite.color
                        if color == "":
                            for j in self.cords:
                                can_paint = True
                                for sprite in sprites.sprites():
                                    if j == sprite.pos:
                                        can_paint = False
                                if can_paint:
                                    sprites.add(Tile(j, self.size, self.selected_color))  # noqa
                        else:
                            for sprite in sprites.sprites():
                                if isinstance(sprite, Tile) and sprite.color == color:
                                    sprites.add(Tile(sprite.pos, self.size, self.selected_color))  # noqa
            if self.pen_type == "trash":
                for sprite in sprites.sprites():
                    sprites.remove(sprite)
                redraw_background()
                sprites.draw(background)
                self.pen_type = "pen"