From 73009d72fe20e7fdd7e086b21cfe40e9a1e958b1 Mon Sep 17 00:00:00 2001
From: Ezekiel <ezekiel@debian.local>
Date: Mon, 21 Jun 2021 14:59:56 +0200
Subject: [PATCH] added possibility of choosing the data list

---
 poc_hugo.py | 22 +++++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

diff --git a/poc_hugo.py b/poc_hugo.py
index b69a1e0..fe4f79c 100644
--- a/poc_hugo.py
+++ b/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:
-        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)