Possibilité de choisir le nom de la bdd

Author: alex-baiet

badass.pyproject.user

@@ -1,6 +1,6 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <!DOCTYPE QtCreatorProject>
-<!-- Written by QtCreator 7.0.1, 2022-06-29T09:24:45. -->
+<!-- Written by QtCreator 7.0.1, 2022-06-29T12:37:20. -->
 <qtcreator>
  <data>
   <variable>EnvironmentId</variable>

py/create_db.py

@@ -16,7 +16,7 @@
 # Nom par défaut du fichier qgz
 DEFAULT_QGZ_NAME = "Badass"
 # Nom par défaut de la nouvelle bdd
-DEFAULT_DB_NAME = "badass_v2_vierge.sqlite"
+DEFAULT_DB_NAME = "badass"
 
 class CreateDB(QtWidgets.QDialog, FORM_CLASS):
     """
@@ -42,21 +42,22 @@ def __generate_files(self):
             self.labMsg.setText("Le dossier sélectionné n'existe pas.")
             return
 
-        name = self.editName.text()
-        if len(name) == 0:
-            name = DEFAULT_QGZ_NAME
+        qgz_name = self.editQgzName.text()
+        if len(qgz_name) == 0:
+            qgz_name = DEFAULT_QGZ_NAME
 
-        ### Création bdd ###
-        path_db = os.path.join(path_dir, DEFAULT_DB_NAME)
-        db.exec_sql_file(path_db, db.SQL_MAIN)
-        # Création extensions
+        db_name = self.editDbName.text()
+        if len(db_name) == 0:
+            db_name = DEFAULT_DB_NAME
+
+        sql_files = [db.SQL_MAIN]
         if self.checkOfTheDead.isChecked():
-            db.exec_sql_file(path_db, db.SQL_OF_THE_DEAD)
+            sql_files.append(db.SQL_OF_THE_DEAD)
         if self.checkAtHome.isChecked():
-            db.exec_sql_file(path_db, db.SQL_AT_HOME)
+            sql_files.append(db.SQL_AT_HOME)
 
         ### Création qgz ###
-        qgz.create_qgz(os.path.join(path_dir, name) + ".qgz")
-        
+        qgz.create_project(path_dir, qgz_name + ".qgz", db_name + ".sqlite", sql_files)
+
         self.labMsg.setText("Le projet a été créé avec succès.")
 

py/create_db.ui

@@ -6,8 +6,8 @@
    <rect>
     <x>0</x>
     <y>0</y>
-    <width>422</width>
-    <height>207</height>
+    <width>459</width>
+    <height>259</height>
    </rect>
   </property>
   <property name="sizePolicy">
@@ -81,22 +81,32 @@
         </layout>
        </item>
        <item>
-        <widget class="QLabel" name="labName">
+        <widget class="QLabel" name="labQgzName">
          <property name="text">
           <string>Nom du nouveau projet</string>
          </property>
         </widget>
        </item>
        <item>
-        <layout class="QHBoxLayout" name="horizontalLayout_3">
-         <item>
-          <widget class="QLineEdit" name="editName">
-           <property name="text">
-            <string>Badass</string>
-           </property>
-          </widget>
-         </item>
-        </layout>
+        <widget class="QLineEdit" name="editQgzName">
+         <property name="text">
+          <string>Badass</string>
+         </property>
+        </widget>
+       </item>
+       <item>
+        <widget class="QLabel" name="labDbName">
+         <property name="text">
+          <string>Nom de la base de données</string>
+         </property>
+        </widget>
+       </item>
+       <item>
+        <widget class="QLineEdit" name="editDbName">
+         <property name="text">
+          <string>badass</string>
+         </property>
+        </widget>
        </item>
       </layout>
      </item>
@@ -126,6 +136,19 @@
          </property>
         </widget>
        </item>
+       <item>
+        <spacer name="verticalSpacer">
+         <property name="orientation">
+          <enum>Qt::Vertical</enum>
+         </property>
+         <property name="sizeHint" stdset="0">
+          <size>
+           <width>20</width>
+           <height>30</height>
+          </size>
+         </property>
+        </spacer>
+       </item>
       </layout>
      </item>
     </layout>

py/qgz.py

@@ -8,22 +8,42 @@
 import shutil
 from zipfile import ZipFile
 from . import files
+from . import db
 
 ORGINAL_DB_NAME = "badass_v2_vierge.sqlite"
 
-def create_qgz(dest_path: str):
+def create_project(dir_path: str, qgz_name: str, db_name: str, sql_files: str):
+    """Créer un nouveau projet avec sa base de données.
+
+    Args:
+        dir_path (str): Dossier de destination du projet
+        qgz_name (str): Nom du fichier .qgz généré
+        db_name (str): Nom du fichier de la base de données générée
+        sql_files (str): Liste des fichiers sql à exécuter pour générer la base de données
+    """
+    qgz_path = os.path.join(dir_path, qgz_name)
+    db_path = os.path.join(dir_path, db_name)
+    
+    # Création de la bdd
+    for file in sql_files:
+        db.exec_sql_file(db_path, file)
+        
+    # Création du qgz
+    __create_qgz(qgz_path)
+    __edit_qgz_db(qgz_path, db_name)
+
+def __create_qgz(dest_path: str):
     """
     Permet de créer un nouveau fichier QGZ au chemin indiqué.
     Args:
         dest_path (str): Chemin du nouveau fichier QGZ.
     """
     src_path = files.get_file_path("file/model.qgz")
-    
     shutil.copyfile(src_path, dest_path)
 
 def __edit_qgz_db(path: str, new_db_name: str, old_db_name: str = ORGINAL_DB_NAME):
     """
-    Modifie le nom de la base de données dans le projet qgis.
+    Modifie le nom de la base de données utilisé par le projet qgis.
 
     Args:
         path (str): Chemin du fichier .qgz