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