src/ui/widgets/elsa_main.py

@@ -1,24 +1,26 @@
 import os
-from .widget_sources.elsa_maindialog_ui import Ui_Dialog
-from PySide6 import QtCore, QtWidgets, QtGui
-from PySide6.QtGui import QRegularExpressionValidator
-from PySide6.QtCore import QDate
-from src import Icon
-from src.backend import Semester, Database
-from src.logic import elsa_word_to_csv, Prof
-from src.ui.dialogs import ElsaAddEntry, popus_confirm
-from src.ui.widgets import FilePicker, DataQtGraph
-from src.backend import recreateElsaFile
-import loguru
 import sys
-from src import LOG_DIR
+
+import loguru
+from PySide6 import QtCore, QtGui, QtWidgets
+from PySide6.QtCore import QDate
+from PySide6.QtGui import QRegularExpressionValidator
+
+from src import LOG_DIR, Icon
+from src.backend import Database, Semester, recreateElsaFile
+from src.logic import Prof, elsa_word_to_csv
+from src.ui.dialogs import ElsaAddEntry, popus_confirm
+from src.ui.widgets.filepicker import FilePicker
+from src.ui.widgets.graph import DataQtGraph
+
+from .widget_sources.elsa_maindialog_ui import Ui_Dialog
+
 log = loguru.logger
 log.remove()
 log.add(sys.stdout, level="INFO")
 log.add(f"{LOG_DIR}/application.log", rotation="1 MB", retention="10 days")
 
 
-
 class ElsaDialog(QtWidgets.QDialog, Ui_Dialog):
     def __init__(self):
         super().__init__()
@@ -240,7 +242,7 @@ class ElsaDialog(QtWidgets.QDialog, Ui_Dialog):
 
         if prof_id is None:
             self.db.createProf(profdata)
-            prof_id = self.db.getProfId(prof)
+            prof_id = self.db.getProfId(profdata)
             self.profs.append(
                 "f{}, {}".format(profdata.lastname, profdata.firstname), prof_id
             )
@@ -413,6 +415,16 @@ class ElsaDialog(QtWidgets.QDialog, Ui_Dialog):
             log.debug(
                 f"elsa_id: {elsa_id}, prof: {self.elsa_prof.currentText()}, semester: {self.elsa_semester.text()}, date: {self.elsa_date.text()}"
             )
+            self.db.insertElsaFile(
+                [
+                    {
+                        "name": file.split("/")[-1],
+                        "path": file,
+                        "type": file.split(".")[-1],
+                    }
+                ],
+                elsa_id,
+            )
             for row in data:
                 if self.seperateEntries.isChecked():
                     if ";" in row["pages"]:
@@ -426,6 +438,7 @@ class ElsaDialog(QtWidgets.QDialog, Ui_Dialog):
                 else:
                     self.setElsaRow(row)
                     self.db.addElsaMedia(row, elsa_id)
+
         self.quote_entry.setEnabled(True)
 
     def openDocumentElsa(self):