赋能卷生成功能适配mariadb保存的材料

工具v4/赋能卷生成.py

@@ -32,7 +32,6 @@ class MyWindow_fnsc(QWidget,Ui_Form):
         mydb = connect(hostname = db_host, port = db_port, username=db_user, pwd=db_pwd, db = self.database_name)
         mycursor = mydb.cursor()
         pro_dict,obj_dict,bn_dict,unit_obj_dict = generateDictsfromMariaDB(mycursor)
-        mydb.close()
         units_list = []
         for self.checkbox in self.findChildren(QCheckBox):
             if self.checkbox.isChecked():
@@ -48,7 +47,8 @@ class MyWindow_fnsc(QWidget,Ui_Form):
         elif self.radioButton_median.isChecked():
             method = "median"
         sim_threshold = float(self.lineEdit_threshold.text())
-        excludeids = findxiaobenIDs("../备课组",grade = self.lineEdit_grade.text())
+        excludeids = findxiaobenIDs(mycursor,grade = self.lineEdit_grade.text())
+        mydb.close()
         papercount = int(self.lineEdit_papercount.text())
         output = ""
         for i in range(papercount):