mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-12 23:40:03 +08:00
refactor(user_console): improve user retrieval logic by checking existence before creation
This commit is contained in:
@@ -40,8 +40,11 @@ class UserConsole(Model):
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
async def get_user(cls, user_id: str, platform: str | None = None) -> "UserConsole":
|
async def get_user(cls, user_id: str, platform: str | None = None) -> "UserConsole":
|
||||||
|
if user := await cls.get_or_none(user_id=user_id):
|
||||||
|
return user
|
||||||
|
|
||||||
try:
|
try:
|
||||||
user, _ = await cls.get_or_create(
|
user = await cls.create(
|
||||||
user_id=user_id,
|
user_id=user_id,
|
||||||
defaults={
|
defaults={
|
||||||
"platform": platform,
|
"platform": platform,
|
||||||
|
|||||||
Reference in New Issue
Block a user