chore: code clean up

This commit is contained in:
Giovani Rodriguez
2021-07-13 15:15:18 -04:00
parent c134a14d3c
commit a5eeb39993

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"