Init commit

2023-05-03 19:53:01 +03:00
commit 12c87b99d3
45 changed files with 3890 additions and 0 deletions
@@ -0,0 +1,107 @@
+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))