added possibility of choosing the data list
This commit is contained in:
parent
d19cd60d5d
commit
73009d72fe
16
poc_hugo.py
16
poc_hugo.py
|
@ -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:
|
||||
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)
|
||||
|
|
Loading…
Reference in New Issue