diff --git a/test/create_random_db.py b/test/create_random_db.py index fd2eca20ba6e11da3422aae6e7003c8d4aaade7b..a4808a8721d8896f441408809f854b6c6c15331c 100644 --- a/test/create_random_db.py +++ b/test/create_random_db.py @@ -47,8 +47,11 @@ def create_tutos(nb_tutos): def create_labels(nb_labels): values1 = [] for i in range(nb_labels): - values1.append([random_generator(7)]) - request = "INSERT INTO labels (name) VALUES (?)" + father = randint(1, nb_labels) + while (father == i): + father = randint(1, nb_labels) + values1.append([random_generator(7), father]) + request = "INSERT INTO labels (name, father_id) VALUES (?, ?)" send_sql_request(request, many=1, values=values1) def link_tutos_labels(nb_tutos, nb_labels): @@ -91,6 +94,7 @@ elif (len(sys.argv) == 4): nb_labels = int(sys.argv[3]) connection = sqlite3.connect(db_name, check_same_thread=False) create_all_db(db_name, nb_tutos, nb_labels) + print ("Succès") else: print ("bad parameters, use \"create_random_db.py [db_name] [nb_tutos] [nb_labels]\"")