added possibility of choosing the data list

This commit is contained in:
Ezekiel 2021-06-21 14:59:56 +02:00
parent d19cd60d5d
commit 73009d72fe
1 changed files with 17 additions and 5 deletions

View File

@ -9,10 +9,21 @@ from rich.markdown import Markdown
# Open the CSV file and create a dictionary with all the questions
def _create_dictionary():
with open(path) as file:
reader = csv.DictReader(file)
data = list(reader)
return data
answers = input("Choose your question list : \n 1: LPI Essentials \n 2: LPIC 1 101-500 \n")
if answers == "1":
with open(path1) as file:
reader = csv.DictReader(file)
data = list(reader)
return data
elif answers == "2":
with open(path2) as file:
reader = csv.DictReader(file)
data = list(reader)
return data
else:
print("Please enter a valid value")
exit()
def _print_question(question_asked):
@ -87,7 +98,8 @@ def print_stats(counter):
if __name__ == "__main__":
path = "./data/list_book1.csv"
path1 = "./data/list_book1.csv"
path2 = "./data/lpic1_book_101-500.csv"
counter = []
dictionary = _create_dictionary()
random.shuffle(dictionary)