diff --git a/sorteo.py b/sorteo.py index c506874..8bf9b58 100644 --- a/sorteo.py +++ b/sorteo.py @@ -20,16 +20,20 @@ from random import shuffle # load csv files -def cvsReader(filename:str) -> list: +def cvsReader(filename:str) -> list: + ''' + Read CSV + + >>> os.path.isfile('./participants.csv') + True + >>> type(cvsReader('participants')) + + ''' with open('{}.csv'.format(filename),'r') as f: reader = csv.reader(f) data = namedtuple(filename,next(reader)) return [data(*row) for row in reader ] - -participants = cvsReader('participants') -reward = cvsReader('reward') - winners = [] def winner(data:list) -> str: try: @@ -46,7 +50,12 @@ def exit_app(signal,frame): print("\nFelicitaciones a todos los Ganadores\n{}".format(w)) sys.exit(0) -signal.signal(signal.SIGINT, exit_app) -while True: - input("El Ganador es ....") - print(winner(participants)) +if __name__ == "__main__": + signal.signal(signal.SIGINT, exit_app) + + participants = cvsReader('participants') + reward = cvsReader('reward') + + while True: + input("El Ganador es ....") + print(winner(participants))