Feat/refactor

ball_release_event_listener.py

@@ -2,7 +2,7 @@
 from game import Game
 from stoppable_thread import StoppableThread
 from vector import Vector
-from math import (ceil, atan, pi, cos, sin)
+from math import (ceil, atan, cos, sin)
 import pygame
 
 class BallReleaseEventListener(EventListener):

game.py

@@ -1,11 +1,14 @@
 import pygame
 import sys
 
-from event_listener import EventListener
 from player import Player
 from element import Element
 from window import Window
 
+from event_listener import EventListener
+from drag_event_listener import DragEventListener
+from ball_release_event_listener import BallReleaseEventListener
+
 class Game():
     def __init__(self, screen:pygame.Surface, img_location:str, sound_location:str) -> None:
         # get the actual display screen.
@@ -38,6 +41,9 @@ def __init__(self, screen:pygame.Surface, img_location:str, sound_location:str)
         self.__window.register_element("ball", Element(ball, 200, 400))
         # register the placeholder ball.
         self.__window.register_element("placeholder_ball", Element(placeholder_ball, 200, 400))
+        # listen to events.
+        self.listen(pygame.MOUSEMOTION, DragEventListener())
+        self.listen(pygame.MOUSEBUTTONUP, BallReleaseEventListener()) 
     def get_window(self) -> Window:
         """
         :return: the game's window.

main.py

@@ -1,8 +1,5 @@
 import pygame
-
 from game import Game
-from drag_event_listener import DragEventListener
-from ball_release_event_listener import BallReleaseEventListener
 
 # init pygame.
 pygame.init()
@@ -16,9 +13,6 @@
 screen = pygame.display.set_mode((1024,640))
 # create a new instance of the Game.
 game = Game(screen, "img/", None)
-# listen to events.
-game.listen(pygame.MOUSEMOTION, DragEventListener())
-game.listen(pygame.MOUSEBUTTONUP, BallReleaseEventListener())
 # register dummy player.
 game.register_player("Yanis")
 # setup the game.