Traktor/app/base_field.py

import pygame

from config import *


class BaseField:
    def __init__(self, img_path: str):
        self._img_path = img_path

    def draw_field(self, screen: pygame.Surface, pos_x: int, pos_y: int,
                   is_centered: bool = False, size: tuple = None, angle: float = 0.0):
        img = pygame.image.load(self._img_path)
        img = pygame.transform.rotate(img, angle)

#     def draw_field(self, screen: pygame.Surface, pos_x: int, pos_y: int, is_centered: bool = False, size: tuple = None,
#                    angle: int = 0):
#         pre_img = pygame.image.load(self._img_path)
#         if angle == 90:
#             pre_img = pygame.transform.flip(pre_img, True, False)
#         img = pygame.transform.rotate(pre_img, angle)
        scale = pygame.transform.scale(img, (FIELD_SIZE, FIELD_SIZE))
        rect = img.get_rect()

        if is_centered:
            rect.center = (pos_x, pos_y)
        else:
            rect.topleft = (pos_x, pos_y)

        if size is not None:
            rect.size = (FIELD_SIZE, FIELD_SIZE)

        screen.blit(scale, rect)