Add bot activity configuration

This commit is contained in:
Choco
2023-04-09 20:28:26 +08:00
parent b8de512346
commit fd46453567
6 changed files with 89 additions and 8 deletions

View File

@@ -1,2 +1,3 @@
from .lyrics import lyricsPlatform
from .placeholders import Placeholders
from .settings import Settings, TOKENS

42
addons/placeholders.py Normal file
View File

@@ -0,0 +1,42 @@
from discord.ext import commands
from re import findall
from importlib import import_module
class Placeholders:
def __init__(self, bot: commands.Bot) -> None:
self.bot = bot
self.voicelink = import_module("voicelink")
self.placeholders = {
"guilds": self.guilds_count,
"users": self.users_count,
"players": self.players_count,
"nodes": self.nodes_count
}
def guilds_count(self) -> int:
return len(self.bot.guilds)
def users_count(self) -> int:
return len(self.bot.users)
def players_count(self) -> int:
count = 0
for node in self.voicelink.NodePool._nodes.values():
count += len(node._players)
return count
def nodes_count(self):
return len(self.voicelink.NodePool._nodes)
def replace(self, msg: str) -> str:
keys = findall(r'@@(.*?)@@', msg)
for key in keys:
value = self.placeholders.get(key.lower(), None)
if value is None:
continue
msg = msg.replace(f"@@{key}@@", str(value()))
return msg

View File

@@ -7,6 +7,7 @@ class Settings:
self.nodes = settings.get("nodes", {})
self.max_queue = settings.get("default_max_queue", 1000)
self.bot_prefix = settings.get("prefix", "")
self.activity = settings.get("activity", [{"listen": "/help"}])
self.embed_color = int(settings.get("embed_color", "0xb3b3b3"), 16)
self.bot_access_user = settings.get("bot_access_user", [])
self.emoji_source_raw = settings.get("emoji_source_raw", {})