player.py

import pygame
from ball import Ball


class Player:
	def __init__(self, image, x_pos, y_pos, speed, width, height):
		self.speed = speed
		self.x_pos = x_pos
		self.y_pos = y_pos
		self.speed = speed
		self.width = width
		self.height = height
		self.score = 0
		self.rounds = 0
		self.rect = pygame.Rect(x_pos, y_pos, width, height)
		self.image = pygame.image.load(image)
		self.image = pygame.transform.scale(self.image, (width, height))

	def move_up(self):
		if self.rect.top > 0:
			self.rect.y -= self.speed

	def move_down(self):
		if self.rect.bottom < pygame.display.get_surface().get_height():
			self.rect.y += self.speed

	def reset_position(self, y_position):
		self.rect.y = y_position


class AutoPlayer(Player):
	def __init__(self, image, x_pos, y_pos, speed, width, height):
		super().__init__(image, x_pos, y_pos, speed, width, height)
		self.target = y_pos
		self.delay = 0

	def predict_ball(self, ball, screen_width, screen_height):
		position_x = ball.rect.x
		position_y = ball.rect.y
		velocity_x = ball.speed_x
		velocity_y = ball.speed_y

		while 0 < position_x < screen_width:
			position_x += velocity_x
			position_y += velocity_y
			if position_y <= 0 or position_y >= screen_height:
				velocity_y *= -1

		return position_y

	def movement(self, ball, screen_width, screen_height):
		player_center = self.rect.height // 2

		if self.delay > 0:
			self.delay -= 1
		else:
			self.target = self.predict_ball(ball, screen_width, screen_height)
			self.delay = 100

		if self.rect.centery < self.target - player_center and self.rect.bottom < screen_height:
			self.rect.y += self.speed
		elif self.rect.centery > self.target + player_center and self.rect.top > 0:
			self.rect.y -= self.speed