pneuma-pygame/effects/magic_effects.py

import pygame
from random import randint

from config.system.window import TILESIZE


class MagicPlayer:
    def __init__(self, animation_player):
        self.animation_player = animation_player

    def heal(self, player, strength, cost, groups):
        if player.stats.energy >= cost:
            self.sounds['heal'].play()
            player.stats.health += strength
            player.stats.energy -= cost
            if player.stats.health >= player.stats.stats['health']:
                player.stats.health = player.stats.stats['health']
            self.animation_player.generate_particles(
                'aura',
                player.animation.rect.center,
                groups)

            self.animation_player.generate_particles(
                'heal',
                player.animation.rect.center + pygame.math.Vector2(0, -50),
                groups)

    def flame(self, player, cost, groups):
        if player.stats.energy >= cost:
            player.stats.energy -= cost

            if player._input.status.split('_')[0] == 'right':
                direction = pygame.math.Vector2(1, 0)
            elif player._input.status.split('_')[0] == 'left':
                direction = pygame.math.Vector2(-1, 0)
            elif player._input.status.split('_')[0] == 'up':
                direction = pygame.math.Vector2(0, -1)
            else:
                direction = pygame.math.Vector2(0, 1)

            for i in range(1, 6):
                if direction.x:
                    offset_x = direction.x * i * TILESIZE
                    x = player.animation.rect.centerx + offset_x + \
                        randint(-TILESIZE // 3, TILESIZE // 3)
                    y = player.animation.rect.centery + \
                        randint(-TILESIZE // 3, TILESIZE // 3)
                    self.animation_player.generate_particles(
                        'flame', (x, y), groups)
                else:
                    offset_y = direction.y * i * TILESIZE
                    x = player.animation.rect.centerx + \
                        randint(-TILESIZE // 3, TILESIZE // 3)
                    y = player.animation.rect.centery + offset_y + \
                        randint(-TILESIZE // 3, TILESIZE // 3)
                    self.animation_player.generate_particles(
                        'flame', (x, y), groups)