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)