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))
|