Trashmaster/agent.py

import pygame.image

class trashmaster(pygame.sprite.Sprite):
    
    def __init__(self,x,y,img):
        super().__init__()
        
        self.width=x
        self.height=y

        self.x = 0
        self.y = 0
    
        self.image = pygame.image.load(img)
        self.image = pygame.transform.scale(self.image, (self.width,self.height))
        self.rect = self.image.get_rect()

    def movement(self, key, vel):
        if key == pygame.K_LEFT:
            self.x -= vel
  
        if key == pygame.K_RIGHT:
            self.x += vel
  
        if key == pygame.K_UP:
            self.y -= vel
  
        if key == pygame.K_DOWN:
            self.y += vel
        return (self.x, self.y)