tuxbot-bot/tuxbot/cogs/Mod/functions/utils.py

35 lines
971 B
Python

from tuxbot.cogs.Mod.models.rules import Rule
from tuxbot.core.config import set_for_key
from tuxbot.core.config import Config
from tuxbot.core.bot import Tux
from tuxbot.core.utils.functions.extra import ContextPlus
async def save_lang(bot: Tux, ctx: ContextPlus, lang: str) -> None:
set_for_key(bot.config.Servers, ctx.guild.id, Config.Server, locale=lang)
async def get_server_rules(guild_id: int) -> list[Rule]:
return await Rule.filter(server_id=guild_id).all()
def get_most_recent_server_rules(rules: list[Rule]) -> Rule:
return sorted(rules, key=lambda r: r.updated_at)[0]
def paginate_server_rules(rules: list[Rule]) -> list[str]:
body = [""]
for rule in rules:
if len(body[-1] + format_rule(rule)) > 2000:
body.append(format_rule(rule) + "\n")
else:
body[-1] += format_rule(rule) + "\n"
return body
def format_rule(rule: Rule) -> str:
return f"**{rule.rule_id}** - {rule.content}"