108 lines
4.6 KiB
Python
108 lines
4.6 KiB
Python
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
import discord
|
|||
|
|
from discord.ext import commands
|
|||
|
|
from pymongo import MongoClient
|
|||
|
|
from datetime import timedelta
|
|||
|
|
|
|||
|
|
from cogs.translate import *
|
|||
|
|
|
|||
|
|
client = MongoClient('localhost', 27017)
|
|||
|
|
db = client['Koteika']
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Logs(commands.Cog, name="Логи"):
|
|||
|
|
def __init__(self, bot):
|
|||
|
|
self.bot = bot
|
|||
|
|
|
|||
|
|
@commands.command(brief="Управление логами")
|
|||
|
|
async def logs(self, ctx):
|
|||
|
|
info = db.logs.find_many({"type": "log_chat", "guild": ctx.guild})
|
|||
|
|
if not info:
|
|||
|
|
e = discord.Embed(title="Логи сервера", description="На этом сервере нет логов")
|
|||
|
|
else:
|
|||
|
|
e = discord.Embed(title="Логи сервера")
|
|||
|
|
for i in info:
|
|||
|
|
channel = self.bot.get_channel(i["channel"])
|
|||
|
|
listening = [f"`{j}`" for j in i["listen"]]
|
|||
|
|
if len(listening) >= 3:
|
|||
|
|
listening = listening[:2] + [str(len(listening)-2)]
|
|||
|
|
e.add_field(name=channel.mention, value=', '.join(listening))
|
|||
|
|
message = await ctx.send(embed=e)
|
|||
|
|
await message.add_reaction("➕")
|
|||
|
|
|
|||
|
|
def check(reaction, user):
|
|||
|
|
return reaction.message == message and str(reaction.emoji) == "➕" and user == ctx.message.author
|
|||
|
|
|
|||
|
|
reaction, user = await client.wait_for('reaction_add', check=check)
|
|||
|
|
await ctx.send("Done")
|
|||
|
|
|
|||
|
|
@commands.Cog.listener()
|
|||
|
|
async def on_error(self, event, *args, **kwargs):
|
|||
|
|
e = discord.Embed(title="Ошибка", description=event, color=discord.Color(0xff0000))
|
|||
|
|
e.add_field(name="args", value=str(args))
|
|||
|
|
e.add_field(name="kwargs", value=str(kwargs))
|
|||
|
|
await self.bot.get_channel(795050679776968704).send(embed=e)
|
|||
|
|
|
|||
|
|
@commands.Cog.listener()
|
|||
|
|
async def on_guild_join(self, guild):
|
|||
|
|
e = discord.Embed(title=f"Бот зашел на сервер (#{len(self.bot.guilds)})",
|
|||
|
|
description=f"Имя: {guild.name}\n" \
|
|||
|
|
f"Владелец: {guild.owner.name}\n" \
|
|||
|
|
f"Создан: {(guild.created_at + timedelta(hours=3)).isoformat()}\n" \
|
|||
|
|
f"Участников: {len(guild.members)}\n" \
|
|||
|
|
f"ID: {guild.id}",
|
|||
|
|
color=discord.Color(0x388e3c))
|
|||
|
|
e.set_thumbnail(url=guild.icon_url)
|
|||
|
|
mes = await self.bot.get_channel(795050679776968704).send(embed=e)
|
|||
|
|
|
|||
|
|
db.logs.insert_one({"type": "guild",
|
|||
|
|
"id_guild": guild.id,
|
|||
|
|
"id_mes_log": mes.id,
|
|||
|
|
"joined_at": guild.me.joined_at})
|
|||
|
|
|
|||
|
|
@commands.Cog.listener()
|
|||
|
|
async def on_guild_remove(self, guild):
|
|||
|
|
mes_id = db.logs.find_one({"id_guild": guild.id})["id_mes_log"]
|
|||
|
|
joined_at = db.logs.find_one({"id_guild": guild.id})["joined_at"]
|
|||
|
|
was = (datetime.now() - joined_at).total_seconds()
|
|||
|
|
|
|||
|
|
days = int(was / 60 / 60 / 24)
|
|||
|
|
hours = int(was / 60 / 60 - days * 24)
|
|||
|
|
minutes = int(was / 60 - (hours * 60 + days * 24 * 60))
|
|||
|
|
seconds = int(was % 60)
|
|||
|
|
|
|||
|
|
mes = await self.bot.get_channel(795050679776968704).fetch_message(mes_id)
|
|||
|
|
|
|||
|
|
e = discord.Embed(title=f"Бот вышел из сервера (#{len(self.bot.guilds)})",
|
|||
|
|
color=discord.Color(0xd32f2f))
|
|||
|
|
e.add_field(name="был на сервере:",
|
|||
|
|
value=f"Дней: {days}\n"
|
|||
|
|
f"Часов: {hours - 3}\n"
|
|||
|
|
f"Минут: {minutes}\n"
|
|||
|
|
f"Секунд: {seconds}\n")
|
|||
|
|
e.set_thumbnail(url=guild.icon_url)
|
|||
|
|
|
|||
|
|
await mes.reply(embed=e)
|
|||
|
|
|
|||
|
|
db.logs.delete_one({"id_guild": guild.id})
|
|||
|
|
|
|||
|
|
@commands.Cog.listener()
|
|||
|
|
async def on_message_delete(self, message):
|
|||
|
|
info = db.logs.find_many({"type": "log_chat", "guild": message.guild})
|
|||
|
|
if info:
|
|||
|
|
for i in info:
|
|||
|
|
if "message_delete" in i["listen"]:
|
|||
|
|
region = region_to_str(message.guild.region)
|
|||
|
|
e = discord.Embed(title=translate("$log_messageDelete", region))
|
|||
|
|
e.add_field(name=translate("$log_author", region),
|
|||
|
|
value="{0.mention} ({0.id})".format(message.author))
|
|||
|
|
e.add_field(name=translate("$log_channel", region),
|
|||
|
|
value="{0.mention} ({0.id})".format(message.channel))
|
|||
|
|
e.add_field(name=translate("$log_text".region), value=message.content)
|
|||
|
|
self.bot.get_channel(i["channel"]).send(embed=e)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def setup(bot):
|
|||
|
|
bot.add_cog(Logs(bot))
|