From eab867e85afdd3fcaeb06d4fffe36d43abc58212 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E7=82=9C=E7=BF=94?= <2307953404@qq.com>
Date: Tue, 1 Oct 2024 19:55:06 +0800
Subject: [PATCH] =?UTF-8?q?=E6=88=91=E5=9C=A8=E8=BD=A6=E4=B8=8A=E5=86=99?=
 =?UTF-8?q?=E4=BB=A3=E7=A0=81+w-?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../Controllers/Budget/BgCountController.cs   | 93 +++++++++++++++++++
 asg_form/Controllers/Dbset.cs                 |  2 +
 2 files changed, 95 insertions(+)
 create mode 100644 asg_form/Controllers/Budget/BgCountController.cs

diff --git a/asg_form/Controllers/Budget/BgCountController.cs b/asg_form/Controllers/Budget/BgCountController.cs
new file mode 100644
index 0000000..2b5d292
--- /dev/null
+++ b/asg_form/Controllers/Budget/BgCountController.cs
@@ -0,0 +1,93 @@
+using asg_form.Controllers.Teamregistration;
+using Microsoft.AspNetCore.Authorization;
+using Microsoft.AspNetCore.Identity;
+using Microsoft.AspNetCore.Mvc;
+using System.Security.Claims;
+
+namespace asg_form.Controllers.Budget
+{
+    public class BgDB
+    {
+        public long id { get; set; }
+        public string budget_year { get; set; }
+        public long budget_money { get; set; }
+        public string budget_type { get; set; }
+        public string budget_name { get; set; }
+        public string budget_used { get; set; }
+        public string source_event_name { get; set; }
+        public long source_event_id { get; set; }
+        public string used_event_name { get; set; }
+        public long used_event_id { get; set; }
+        //public string use_person {} 
+        //public long use_person_id { get; set; }
+        public string change_time { get; set; }
+        public string use_status { get; set; }
+        public string? update_person { get; set; }//设置默认
+        public string? update_time { get; set; }
+
+    }
+
+    public class BgCountController : ControllerBase
+    {
+        private readonly RoleManager<Role> roleManager;
+        private readonly UserManager<User> userManager;
+        public BgCountController(
+            RoleManager<Role> roleManager, UserManager<User> userManager)
+        {
+            this.roleManager = roleManager;
+            this.userManager = userManager;
+        }
+        private class bgMsg
+        {
+            public string budgetYear { get; set; }
+            public int budgetMoney { get; set;}
+            public string budgetType { get; set; }
+            public string budgetName { get; set; }
+            public string sourceEventName { get; set; }
+            public long sourceEventId { get; set; }
+            public string budgetUsed {  get; set; }
+            public string usedEventName { get;set; }
+            public long usedEventId { get; set; }
+            public string updatePerson { get; set; }
+            public string updateTime { get; set; }//自动生成时间
+
+        }
+        [Route("api/v1/admin/insertBg")]
+        [HttpPost]
+        [Authorize]
+        public async Task<ActionResult<object>> InBg([FromBody] bgMsg msg)
+        {
+            if (!this.User.FindAll(ClaimTypes.Role).Any(a => a.Value == "admin"))
+            {
+                return Ok(new error_mb { code = 401, message = "无权访问" });
+            }
+            if (msg == null)
+            {
+                return Ok(new error_mb { code = 500, message = "请插入必要字段" });
+            }
+            var dateString = DateTime.Now.ToString(); ;
+            using(TestDbContext sub =  new TestDbContext())
+            {
+                var budget = new BgDB
+                {
+                    budget_year = msg.budgetYear,
+                    budget_money = msg.budgetMoney,
+                    budget_type = msg.budgetType,
+                    budget_name = msg.budgetName,
+                    budget_used = msg.budgetUsed,
+                    source_event_name = msg.sourceEventName,
+                    source_event_id = msg.sourceEventId,
+                    change_time = dateString,
+                    used_event_name = msg.usedEventName,
+                    used_event_id = msg.usedEventId,
+                    use_status = "0",
+                    update_person = null,
+                    update_time = null,
+                };
+            sub.budgetDetails.Add(budget);
+            await sub.SaveChangesAsync();
+            return Ok(new error_mb { code = 200, message = "新增预算成功" });
+        }
+        }
+    }
+}
diff --git a/asg_form/Controllers/Dbset.cs b/asg_form/Controllers/Dbset.cs
index ab8acbd..c0a8f9e 100644
--- a/asg_form/Controllers/Dbset.cs
+++ b/asg_form/Controllers/Dbset.cs
@@ -1,4 +1,5 @@
 
+using asg_form.Controllers.Budget;
 using asg_form.Controllers.Store;
 using asg_form.Controllers.Team;
 using asg_form.Controllers.Teamregistration;
@@ -239,6 +240,7 @@ namespace asg_form.Controllers
         public DbSet<T_config> T_config { get; set; }
         public DbSet<TaskDB> T_Task { get; set; }
         public DbSet<ComformDB> T_Comform { get; set; }
+        public DbSet<BgDB> budgetDetails { get; set; }
 
         protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
         {