obstacle.py

import random

class Obstacle:
    def __init__(self, image, type, SCREEN_WIDTH, game_speed, obstacles):
        self.image = image
        self.type = type
        self.rect = self.image[self.type].get_rect()
        self.rect.x = SCREEN_WIDTH
        self.game_speed = game_speed
        self.obstacles = obstacles

    def update(self):
        self.rect.x -= self.game_speed
        if self.rect.x < -self.rect.width:
            self.obstacles.pop()
    
    def getX(self):
        return self.rect.x
    
    def getY(self):
        return self.rect.y

    def draw(self, SCREEN):
        SCREEN.blit(self.image[self.type], self.rect)

class SmallCactus(Obstacle):
    def __init__(self, image, SCREEN_WIDTH, Y_POS, game_speed, obstacles):
        self.type = random.randint(0, 2)
        super().__init__(image, self.type, SCREEN_WIDTH, game_speed, obstacles)
        self.rect.y = Y_POS+15     

class LargeCactus(Obstacle):
    def __init__(self, image, SCREEN_WIDTH, Y_POS, game_speed, obstacles):
        self.type = random.randint(0, 2)
        super().__init__(image, self.type, SCREEN_WIDTH, game_speed, obstacles)
        self.rect.y = Y_POS-10

class Bird(Obstacle):
    def __init__(self, image, SCREEN_WIDTH, Y_POS, game_speed, obstacles):
        self.type = 0
        super().__init__(image, self.type, SCREEN_WIDTH, game_speed, obstacles)
        self.rect.y = Y_POS-60
        self.index = 0

    def draw(self, SCREEN):
        if self.index >= 9:
            self.index = 0
        SCREEN.blit(self.image[self.index//5], self.rect)
        self.index += 1