From e76366561cf9b3527139b3bd367c6bbeab6c4549 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana?= <rmontanana@gmail.com>
Date: Mon, 21 Nov 2022 00:07:29 +0100
Subject: [PATCH] Add be_init_project to scripts

---
 benchmark/scripts/be_init_project.py | 35 ++++++++++++++++++++++++++++
 requirements.txt                     |  2 +-
 setup.py                             |  1 +
 3 files changed, 37 insertions(+), 1 deletion(-)
 create mode 100755 benchmark/scripts/be_init_project.py

diff --git a/benchmark/scripts/be_init_project.py b/benchmark/scripts/be_init_project.py
new file mode 100755
index 0000000..fda74b0
--- /dev/null
+++ b/benchmark/scripts/be_init_project.py
@@ -0,0 +1,35 @@
+#!/usr/bin/env python
+import os
+from benchmark.Utils import Files, Folders
+from benchmark.Arguments import Arguments
+
+
+def main(args_test=None):
+    arguments = Arguments(prog="be_init_project")
+    arguments.add_argument("project_name", help="Project name")
+    args = arguments.parse(args_test)
+    folders = []
+    folders.append(args.project_name)
+    folders.append(os.path.join(args.project_name, Folders.results))
+    folders.append(os.path.join(args.project_name, Folders.hidden_results))
+    folders.append(os.path.join(args.project_name, Folders.exreport))
+    folders.append(os.path.join(args.project_name, Folders.report))
+    folders.append(os.path.join(args.project_name, Folders.img))
+    try:
+        for folder in folders:
+            print(f"Creating folder {folder}")
+            os.makedirs(folder)
+    except FileExistsError as e:
+        print(e)
+    try:
+        env_src = os.path.join(Folders.src(), "..", f"{Files.dot_env}.dist")
+        env_to = os.path.join(args.project_name, Files.dot_env)
+        os.system(f"cp {env_src} {env_to}")
+    except Exception as e:
+        print(e)
+    print("Done!")
+    print(
+        "Please, edit .env file with your settings and add a datasets folder"
+    )
+    print("with an all.txt file with the datasets you want to use.")
+    print("In that folder you have to include all the datasets you'll use.")
diff --git a/requirements.txt b/requirements.txt
index cba834e..fd384e7 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,7 +5,7 @@ odte
 cython
 mdlp-discretization
 mufs
-bayesclass @ git+ssh://git@github.com/doctorado-ml/bayesclass.git
+#bayesclass @ git+ssh://git@github.com/doctorado-ml/bayesclass.git
 xlsxwriter
 openpyxl
 tqdm
diff --git a/setup.py b/setup.py
index c98406f..807a1b8 100644
--- a/setup.py
+++ b/setup.py
@@ -38,6 +38,7 @@ def script_names():
         "print_strees",
         "report",
         "summary",
+        "init_project",
     ]
     result = []
     for script in scripts: