diff --git a/src/main/java/com/gunshi/project/hsz/controller/AttDamBaseController.java b/src/main/java/com/gunshi/project/hsz/controller/AttDamBaseController.java
index d0e5233..595bbd4 100644
--- a/src/main/java/com/gunshi/project/hsz/controller/AttDamBaseController.java
+++ b/src/main/java/com/gunshi/project/hsz/controller/AttDamBaseController.java
@@ -92,13 +92,21 @@ public class AttDamBaseController extends AbstractCommonFileController{
     @Operation(summary = "列表")
     @PostMapping("/list")
     public R<List<AttDamBase>> list() {
-        return R.ok(service.lambdaQuery().list());
+        List<AttDamBase> list = service.lambdaQuery().list();
+        for (AttDamBase attDamBase : list) {
+            attDamBase.setFiles(fileService.getFiles(getGroupId(), attDamBase.getDamCode()));
+        }
+        return R.ok(list);
     }
 
     @Operation(summary = "获取主副坝,拦洪坝信息")
     @GetMapping("/info")
     public R<List<AttDamBase>> info(@Schema(name = "isMain",description = "是否主坝（0否 1是 2拦洪坝）") @RequestParam(name = "isMain") Integer isMain) {
-        return R.ok(service.lambdaQuery().eq(AttDamBase::getIsMain,isMain).list());
+        List<AttDamBase> list = service.lambdaQuery().eq(AttDamBase::getIsMain, isMain).list();
+        for (AttDamBase attDamBase : list) {
+            attDamBase.setFiles(fileService.getFiles(getGroupId(), attDamBase.getDamCode()));
+        }
+        return R.ok(list);
     }
 
     @Operation(summary = "分页")