feat: improve the way that pieces get generated

This commit is contained in:
2021-06-08 10:10:56 -04:00
parent dc12dac9cc
commit da6f5ef706

View File

@@ -3,12 +3,21 @@ import random
from entity.Piece import Piece from entity.Piece import Piece
from util.ConfigurationManager import ConfigurationManager from util.ConfigurationManager import ConfigurationManager
"""
TODO Add link that goes through random piece generation
"""
class PieceGenerator: class PieceGenerator:
__pieces = []
@classmethod @classmethod
def get_piece(cls, position): def get_piece(cls, position):
piece = Piece(cls.__get_piece_shape(), position, cls.__get_piece_color())
return piece if len(cls.__pieces) == 0:
for _ in range(7):
cls.__pieces.append(Piece(cls.__get_piece_shape(), position, cls.__get_piece_color()))
return cls.__pieces.pop()
def __get_piece_shape(): def __get_piece_shape():
random_number = random.randint(0, 6) random_number = random.randint(0, 6)