feat: improve the way that pieces get generated

This commit is contained in:
Giovani
2021-06-08 10:10:56 -04:00
parent ca1cd14d80
commit 92057af31c

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)