From 3c32baafd3b851a64ab2dfbcc63af54e0fc3e4fc Mon Sep 17 00:00:00 2001 From: nanometer5088 Date: Mon, 20 Feb 2023 01:04:30 -0300 Subject: [PATCH] More efficient file reading --- src/cartelas.py | 31 +++++++++++-------------------- 1 file changed, 11 insertions(+), 20 deletions(-) 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