HealthBar.py

import pygame


vec = pygame.math.Vector2


class HealthBar(pygame.sprite.Sprite):
    def __init__(self, x, y):
        super().__init__()
        self.load_animations()

        self.health = 5
        self.image = self.health_animations[self.health]
        self.pos = vec(x, y)

    def render(self, display):
        display.blit(self.image, self.pos)

    def takeDamage(self, damage):
        self.health -= damage
        if self.health < 0: self.health = 0
        
        self.image = self.health_animations[self.health]
        
    def Heal(self, heal):
        self.health += damage
        if self.health > 5: self.health = 5
        
        self.image = self.health_animations[self.health]

    def load_animations(self):
        
        self.health_animations = [pygame.image.load("Images/WEED0.png").convert_alpha(),
                             pygame.image.load("Images/WEED1.png").convert_alpha(),
                             pygame.image.load("Images/WEED2.png").convert_alpha(),
                             pygame.image.load("Images/WEED3.png").convert_alpha(),
                             pygame.image.load("Images/WEED4.png").convert_alpha(),
                             pygame.image.load("Images/WEED5.png").convert_alpha()]