'commit'

src/main/java/com/dsideal/QingLong/Collect/Controller/CollectController.java

@@ -2184,5 +2184,31 @@ public class CollectController extends Controller {
         kv.set("message", "保存成功！");
         renderJson(kv);
     }
+
+    /**
+     * 功能：获取指定任务的模板与学段的对应关系
+     *
+     * @param job_id
+     */
+    @Before({GET.class})
+    @IsLoginInterface({})
+    public void getSheetStage(int job_id) {
+        List<Record> list = cm.getSheetStage(job_id);
+        renderJson(CommonUtil.renderJsonForLayUI(list));
+    }
+
+    /**
+     * 功能：一键智能处理哪些Sheet对应哪些学段
+     *
+     * @param job_id
+     */
+    @Before({POST.class})
+    @IsLoginInterface({})
+    public void intelligenceSheetStage(int job_id) {
+        cm.intelligenceSheetStage(job_id);
+        Kv kv = Kv.by("success", true);
+        kv.set("message", "保存成功！");
+        renderJson(kv);
+    }
 }
 

src/main/java/com/dsideal/QingLong/Collect/Model/CollectModel.java

@@ -3131,6 +3131,47 @@ public class CollectModel {
         Db.update(sql, job_id, sheet_index, stage_id);
     }
 
+    /**
+     * 功能：获取指定任务的模板与学段的对应关系
+     *
+     * @param job_id
+     */
+    public List<Record> getSheetStage(int job_id) {
+        String sql = "select * from t_collect_sheet_stage where job_id=?";
+        return Db.find(sql, job_id);
+    }
+
+    /**
+     * 功能：一键智能处理哪些Sheet对应哪些学段
+     *
+     * @param job_id
+     */
+    public void intelligenceSheetStage(int job_id) {
+        //1、整体上都修改为0，表示一个都不要
+        String sql = "update t_collect_sheet_stage set b_use=0 where job_id=?";
+        Db.update(sql, job_id);
+
+        //2、遍历这个任务的所有Sheet表
+        sql = "select sheet_index,sheet_name from t_collect_job_sheet where job_id=?";
+        List<Record> list = Db.find(sql, job_id);
+
+        //3、如果Sheet表的表名中存在小学，初中，高中字样，那么认为这个Sheet表对应小学，初中，高中
+        for (Record record : list) {
+            int sheet_index = record.getInt("sheet_index");
+            String sheetName = record.getStr("sheet_name");
+
+            sql = "update t_collect_sheet_stage set b_use=1 where job_id=? and sheet_index=? and stage_id=";
+            if (sheetName.indexOf("小学") >= 0) {
+                sql += 4;
+            } else if (sheetName.indexOf("初中") >= 0) {
+                sql += 5;
+            } else if (sheetName.indexOf("高中") >= 0) {
+                sql += 6;
+            }
+            Db.update(sql, job_id, sheet_index);
+        }
+    }
+
     /**
      * 功能：通过用户设置过的此模板中，学段与Sheet表的对应关系，删除一些不在上面学段内需要填报的表格，提供新的上报模板文件给用户
      *
@@ -3151,28 +3192,13 @@ public class CollectModel {
         for (Record record : list) {
             sheetIndexList.add(record.getInt("sheet_index"));
         }
-        try {
-            File file = new File(excelPath);
-            FileInputStream fis = new FileInputStream(file);
-            XSSFWorkbook wb = new XSSFWorkbook(fis);
-            for (int i = wb.getNumberOfSheets() - 1; i >= 0; i--) {
-                if (!sheetIndexList.contains(i)) {
-                    wb.removeSheetAt(i);
-                }
-            }
-            // 生成随机UUID
-            UUID uuid = UUID.randomUUID();
-            // 将UUID转换为字符串
-            String uuidString = uuid.toString();
-            // 创建临时文件路径
-            File tempFile = new File(System.getProperty("java.io.tmpdir") + File.separator + uuidString+".xlsx");
-            FileOutputStream fos = new FileOutputStream(tempFile);
-            wb.write(fos);
-            return tempFile;
-        } catch (FileNotFoundException e) {
-            throw new RuntimeException(e);
-        } catch (IOException e) {
-            throw new RuntimeException(e);
-        }
+        // 生成随机UUID
+        UUID uuid = UUID.randomUUID();
+        // 将UUID转换为字符串
+        String uuidString = uuid.toString();
+        // 创建临时文件路径
+        File tempFile = new File(System.getProperty("java.io.tmpdir") + File.separator + uuidString + ".xlsx");
+
+
     }
 }