import pygame


def load_rocket(path="raketa.png", scale=3):
    original = pygame.image.load(path).convert_alpha()
    return pygame.transform.scale(
        original, (original.get_width() // scale, original.get_height() // scale)
    )


def rescale_rocket(raketa, speed):
    scale_factor = 1 + speed.length() * 0.01
    raketa_w, raketa_h = raketa.get_size()
    raketa_w *= scale_factor
    raketa_h /= scale_factor
    return pygame.transform.scale(raketa, (raketa_h, raketa_w))