import logging from discord import Cog, Message from discord.ext import commands from libbot.pycord.classes import PycordBot from modules.database import col_analytics logger = logging.getLogger(__name__) class Analytics(commands.Cog): def __init__(self, client: PycordBot): self.client = client @Cog.listener() async def on_message(self, message: Message): if ( (message.author != self.client.user) and (message.author.bot == False) and (message.author.system == False) ): stickers = [] for sticker in message.stickers: stickers.append( { "id": sticker.id, "name": sticker.name, "format": sticker.format, "url": sticker.url, } ) attachments = [] for attachment in message.attachments: attachments.append( { "content_type": attachment.content_type, "description": attachment.description, "filename": attachment.filename, "is_spoiler": attachment.is_spoiler(), "size": attachment.size, "url": attachment.url, "width": attachment.width, "height": attachment.height, } ) col_analytics.insert_one( { "user": message.author.id, "channel": message.channel.id, "content": message.content, "stickers": stickers, "attachments": attachments, } ) def setup(client: PycordBot): client.add_cog(Analytics(client))