diff --git a/src/cartelas.py b/src/cartelas.py index 36cfd66..94de380 100644 --- a/src/cartelas.py +++ b/src/cartelas.py @@ -1,25 +1,16 @@ from src.funcoes import aleatorio def listas(): -#Geração de listas a partir do arquivo de cartelas. Essas listas são separadas em -#listas menores, que são separadas a partir da vírgula. - i = 0 - vazio = '' - arquivo = open('cartelas.txt', 'r', encoding='utf-8') - linhas = arquivo.readlines() - tamanho = len(linhas) - arquivo.close() - vet = [0] * tamanho - arquivo = open('cartelas.txt', 'r', encoding='utf-8') - while True: - lelinha = arquivo.readline().rstrip() - if lelinha == vazio: - break - vet[i] = lelinha.split(',') - for a in range(5): - vet[i][a] = int(vet[i][a]) - i += 1 - arquivo.close() - return vet + # Open file and read lines + with open('cartelas.txt', 'r', encoding='utf-8') as f: + lines = f.readlines() + + # Convert lines to list of lists of integers + cards = [] + for line in lines: + card = [int(number) for number in line.strip().split(',')] + cards.append(card) + + return cards def cartelas_show(): #Sorteia 4 cartelas dentre as 20 disponíveis