Ruby-Cogs/referrals/db/utils.py
2025-05-23 02:30:00 -04:00

12 lines
415 B
Python

import discord
from ..db.tables import GuildSettings
class DBUtils:
async def get_create_guild(self, guild: discord.Guild | int) -> GuildSettings:
guild_id = guild.id if isinstance(guild, discord.Guild) else guild
settings = await GuildSettings.objects().get_or_create(
where=(GuildSettings.id == guild_id), defaults={GuildSettings.id: guild_id}
)
return settings