98 lines
3.4 KiB
Python
98 lines
3.4 KiB
Python
from discord import TextChannel, app_commands, Interaction
|
||
from discord.ext import commands, tasks
|
||
|
||
from bot import db
|
||
|
||
import schedule
|
||
from loguru import logger
|
||
from datetime import datetime
|
||
|
||
col = db.guild_settings
|
||
|
||
@app_commands.guild_only()
|
||
@app_commands.default_permissions(administrator=True)
|
||
class ServerSettings(app_commands.Group, name="настройки_сервера"):
|
||
def add_server_settings(self, server_id: int):
|
||
if not col.find_one({"id": server_id}):
|
||
col.insert_one({
|
||
"id": server_id,
|
||
"levelup": "none",
|
||
"flood_channels": [],
|
||
"commands_channels": [],
|
||
"AFK": {
|
||
"role": None,
|
||
"time": -1,
|
||
"outsaider_help": False,
|
||
"outsaider_help_time": -1
|
||
},
|
||
"clear_top": {
|
||
"enable": False,
|
||
"channel": None,
|
||
"active_role": None,
|
||
"voice_active_role": None
|
||
},
|
||
"aux": {
|
||
"talk": False
|
||
}
|
||
})
|
||
|
||
#@app_commands.command()
|
||
#async def обзор(self, inter: Interaction):
|
||
# await inter.response.send_message("Meow")
|
||
|
||
@app_commands.command(name="назначить_флудилкой")
|
||
async def add_flood_channel(self, inter: Interaction, channel: TextChannel = None):
|
||
self.add_server_settings(inter.guild_id)
|
||
|
||
if channel is None:
|
||
channel = inter.channel
|
||
|
||
col.update_one(
|
||
{"id": inter.guild_id},
|
||
{"$push": {"flood_channels": channel.id}}
|
||
)
|
||
|
||
await inter.response.send_message(f"Канал {channel.mention} успешно стал каналом для флуда")
|
||
|
||
@app_commands.command(name="управление_ответами")
|
||
async def set_aux_talk(self, inter: Interaction, status: bool):
|
||
self.add_server_settings(inter.guild_id)
|
||
|
||
col.update_one({'id': inter.guild_id}, {'$set': {'aux.talk': status}})
|
||
if status:
|
||
await inter.response.send_message("Теперь бот будет отвечать")
|
||
else:
|
||
await inter.response.send_message("Бот больше не будет отвечать")
|
||
|
||
#class NiceName(commands.Cog):
|
||
# def __init__(self, bot):
|
||
# self.bot = bot
|
||
#
|
||
# schedule.every().day.at("00:00:00").do(self.trigger)
|
||
# self.pending.start()
|
||
#
|
||
# @tasks.loop(seconds=1)
|
||
# async def pending(self):
|
||
# #schedule.run_pending()
|
||
# trigger
|
||
#
|
||
# def trigger(self):
|
||
# asyncio.new_event_loop().run_untill_complete(clear_top())
|
||
#
|
||
# async def clear_top(self):
|
||
# if datetime.now().day == 1:
|
||
# logger.debug("Очистка топа!!!!")
|
||
#
|
||
# for data in col.find({"clear_top.enable": True}):
|
||
# server = self.bot.get_guild(data['id'])
|
||
# channel = self.bot.get_channel(data['clear_top']['channel'])
|
||
# active = server.get_member(db.members.find({"guild_stat.822157545622863902": {"$exists": True}}).sort(f"guild_stat.{data['id']}.exp", -1)[0]['id'])
|
||
# active_role = server.get_role(data['clear_top']['active_role'])
|
||
# await active.add_roles(active_role)
|
||
#
|
||
# db.members.update_many({}, {"$unset": f"guild_stat.{server_id}"})
|
||
|
||
|
||
async def setup(bot):
|
||
bot.tree.add_command(ServerSettings())
|