重构了代码

Data/Repositories/VarTableRepository.cs

@@ -1,6 +1,9 @@
+using iNKORE.UI.WPF.Modern.Common.IconKeys;
 using PMSWPF.Data.Entities;
+using PMSWPF.Enums;
 using PMSWPF.Extensions;
 using PMSWPF.Models;
+using SqlSugar;
 
 namespace PMSWPF.Data.Repositories;
 
@@ -15,10 +18,34 @@ public class VarTableRepository
     {
         using (var db = DbContext.GetInstance())
         {
-            return await db.Insertable<DbVariableTable>(varTable.CopyTo<DbVariableTable>()).ExecuteReturnIdentityAsync();
+            return await db.Insertable<DbVariableTable>(varTable.CopyTo<DbVariableTable>())
+                .ExecuteReturnIdentityAsync();
         }
     }
 
+    
+    /// <summary>
+    /// 添加默认变量表
+    /// </summary>
+    /// <param name="db"></param>
+    /// <param name="dbDevice"></param>
+    /// <returns></returns>
+    public async Task<DbVariableTable> AddDeviceDefVarTable(DbDevice dbDevice, SqlSugarClient db)
+    {
+        
+        // 添加默认变量表
+        dbDevice.VariableTables = new List<DbVariableTable>();
+        var dbVariableTable = new DbVariableTable();
+        dbVariableTable.IsActive = true;
+        dbVariableTable.DeviceId=dbDevice.Id;
+        dbVariableTable.Name = "默认变量表";
+        dbVariableTable.Description = "默认变量表";
+        dbVariableTable.ProtocolType = dbDevice.ProtocolType;
+        dbDevice.VariableTables.Add(dbVariableTable);
+        return await db.Insertable<DbVariableTable>(dbVariableTable).ExecuteReturnEntityAsync();;
+    }
+
+
     public async Task<int> Edit(VariableTable variableTable)
     {
         using (var db = DbContext.GetInstance())
@@ -26,4 +53,4 @@ public class VarTableRepository
             return await db.Updateable<DbVariableTable>(variableTable.CopyTo<DbVariableTable>()).ExecuteCommandAsync();
         }
     }
-}   
+}