From 363c8f708275836372e940a87244ec2a313022cd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E2=80=98KimigaiiWuyi=E2=80=99?= <444835641@qq.com>
Date: Wed, 16 Aug 2023 15:07:41 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20=E8=A1=A5=E5=85=85=E7=BC=BA?=
 =?UTF-8?q?=E5=A4=B1=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 gsuid_core/utils/database/base_models.py | 6 ++----
 gsuid_core/utils/database/dal.py         | 6 ++++--
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/gsuid_core/utils/database/base_models.py b/gsuid_core/utils/database/base_models.py
index 260c28c..d295f0d 100644
--- a/gsuid_core/utils/database/base_models.py
+++ b/gsuid_core/utils/database/base_models.py
@@ -55,10 +55,8 @@ class BaseIDModel(SQLModel):
 
     @classmethod
     @with_session
-    async def full_insert_data(
-        cls, session: AsyncSession, model: Type["BaseIDModel"], **data
-    ) -> int:
-        session.add(model(**data))
+    async def full_insert_data(cls, session: AsyncSession, **data) -> int:
+        session.add(cls(**data))
         await session.commit()
         return 0
 
diff --git a/gsuid_core/utils/database/dal.py b/gsuid_core/utils/database/dal.py
index 36faf0b..e1a7725 100644
--- a/gsuid_core/utils/database/dal.py
+++ b/gsuid_core/utils/database/dal.py
@@ -351,7 +351,6 @@ class SQLA:
     #####################
     async def insert_push_data(self, uid: str):
         await GsPush.full_insert_data(
-            GsPush,
             bot_id=self.bot_id,
             uid=uid,
             coin_push='off',
@@ -403,4 +402,7 @@ class SQLA:
                 await session.close()
 
     async def insert_new_bind(self, **kwargs):
-        await GsBind.full_insert_data(GsBind, **kwargs)
+        await GsBind.full_insert_data(**kwargs)
+
+    async def insert_new_user(self, **kwargs):
+        await GsUser.full_insert_data(**kwargs)