chore: code clean up

This commit is contained in:
2021-07-13 15:15:18 -04:00
parent c51b4eef69
commit c2b51947ae

View File

@@ -1,7 +1,6 @@
import copy
import random
import pygame
from pygame import mixer
from typing import List, Tuple
from types import FunctionType
from tetri5.util import ConfigurationManager
@@ -395,7 +394,8 @@ class PieceGenerator:
random_number = random.randint(0, 6 - len(bucket))
bucket.append(piece_types.pop(random_number))
def _get_piece_shape(piece_number: int) -> Tuple:
@classmethod
def _get_piece_shape(cls, piece_number: int) -> Tuple:
if piece_number == 0:
return Piece.I_SHAPE
if piece_number == 1:
@@ -413,7 +413,8 @@ class PieceGenerator:
return None
def _get_piece_color(is_player_two: bool) -> Tuple:
@classmethod
def _get_piece_color(cls, is_player_two: bool) -> Tuple:
random_number = random.randint(1, 3)
player_mod = "player-1" if not is_player_two else "player-2"