mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-18 02:30:08 +08:00
refactor(user_console): streamline user retrieval and creation logic
This commit is contained in:
@@ -40,11 +40,8 @@ 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.create(
|
user, _ = await cls.get_or_create(
|
||||||
user_id=user_id,
|
user_id=user_id,
|
||||||
defaults={
|
defaults={
|
||||||
"platform": platform,
|
"platform": platform,
|
||||||
|
|||||||
Reference in New Issue
Block a user