program doesn't stop when error msg appears, instead asks the question again after the error

This commit is contained in:
theo.canaud 2022-06-20 17:53:02 +02:00
parent 930d91cfa9
commit 8288df18a0

View file

@ -1,8 +1,8 @@
from secrets import choice from secrets import choice
from tracemalloc import stop
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import csv import csv
import pandas as pd
#"C:\Users\quent\OneDrive\Bureau\ENSC\TransD\Framboisier\02400001.TXT" #"C:\Users\quent\OneDrive\Bureau\ENSC\TransD\Framboisier\02400001.TXT"
#! /usr/bin/env python3 #! /usr/bin/env python3
@ -11,13 +11,12 @@ import csv
def function(): def function():
tab = np.zeros([1,9]) tab = np.zeros([1,9])
#def tab_choix():
while True: while True:
choice = input("Choisis un fichier csv de 1 à 1862: \n") choice = input("Choisis un fichier csv de 1 à 1862: \n")
print("\n") print("\n")
if float(choice) in range(1, 1863): if float(choice) in range(1, 1863):
rawdata = '../RawDataFiles/RawData' + choice + '.csv' rawdata = '../RawDataFiles/RawData' + choice + '.csv' #Définit le fichier à ouvrir
with open(rawdata, newline='') as csvfile: with open(rawdata, newline='') as csvfile:
reader = csv.reader(csvfile, quoting=csv.QUOTE_NONNUMERIC) reader = csv.reader(csvfile, quoting=csv.QUOTE_NONNUMERIC)
for row in reader: for row in reader:
@ -28,7 +27,6 @@ def function():
elif choice > '1862' or choice < '1': elif choice > '1862' or choice < '1':
print("{} n'est pas un nombre valide.\n".format(choice)) print("{} n'est pas un nombre valide.\n".format(choice))
exit()
x = tab[:,0] x = tab[:,0]
y = tab[:,1] y = tab[:,1]
@ -50,4 +48,4 @@ def final():
x, y = function() x, y = function()
printFig(x,y) printFig(x,y)
final() final()