KaniumCogs/trafficCog/trafficCog.py

import discord
from redbot.core import Config, commands  # This line imports commands as well
from datetime import datetime, timedelta
import pytz

class TrafficCog(commands.Cog):
    def __init__(self, bot):
        self.bot = bot
        self.config = Config.get_conf(self, identifier=123456789, force_registration=True)
        default_guild = {
            "traffic_channel": None,
            "daily_stats": {"joined": 0, "left": 0, "banned": 0},
            "total_stats": {"joined": 0, "left": 0, "banned": 0},
            "last_reset": datetime.now(pytz.UTC).isoformat(),
            "admin_roles": ['Developer', 'admin', 'Council'],
            "stats_thumbnail_url": 'https://example.com/default-thumbnail.png',
        }
        self.config.register_guild(**default_guild)

    async def __check_reset(self, guild_id):
        async with self.config.guild_from_id(guild_id).all() as guild_data:
            last_reset = datetime.fromisoformat(guild_data['last_reset'])
            if last_reset.date() < datetime.now(pytz.UTC).date():
                guild_data['daily_stats'] = {"joined": 0, "left": 0, "banned": 0}
                guild_data['last_reset'] = datetime.now(pytz.UTC).isoformat()

    async def __update_stat(self, guild_id, stat_type):
        async with self.config.guild_from_id(guild_id).all() as guild_data:
            guild_data['daily_stats'][stat_type] += 1
            guild_data['total_stats'][stat_type] += 1

    @commands.group(name='traffic', invoke_without_command=True)
    @commands.has_permissions(administrator=True)
    async def traffic_commands(self, ctx):
        """Base command for managing TrafficCog settings."""
        await ctx.send_help(ctx.command)

    @traffic_commands.command(name='setchannel')
    async def set_traffic_channel(self, ctx, channel: discord.TextChannel):
        """Sets the channel for traffic logs."""
        await self.config.guild(ctx.guild).traffic_channel.set(channel.id)
        await ctx.send(f"Traffic logs will now be sent to {channel.mention}.")

    @traffic_commands.command(name='stats')
    async def show_stats(self, ctx):
        """Displays current traffic statistics."""
        await self.__check_reset(ctx.guild.id)
        guild_data = await self.config.guild(ctx.guild).all()
        daily_stats = guild_data['daily_stats']
        total_stats = guild_data['total_stats']

        embed = discord.Embed(title="Server Traffic Stats", description="Statistics on server activity", color=0x3399ff)
        embed.set_thumbnail(url=guild_data['stats_thumbnail_url'])
        embed.add_field(name="Daily Joined", value=daily_stats['joined'], inline=True)
        embed.add_field(name="Daily Left", value=daily_stats['left'], inline=True)
        embed.add_field(name="Daily Banned", value=daily_stats['banned'], inline=True)
        embed.add_field(name="Total Joined", value=total_stats['joined'], inline=True)
        embed.add_field(name="Total Left", value=total_stats['left'], inline=True)
        embed.add_field(name="Total Banned", value=total_stats['banned'], inline=True)
        await ctx.send(embed=embed)

    @commands.Cog.listener()
    async def on_member_join(self, member):
        if member.guild.id not in self.config.all_guilds():
            return
        await self.__check_reset(member.guild.id)
        await self.__update_stat(member.guild.id, 'joined')
        channel_id = await self.config.guild(member.guild).traffic_channel()
        if channel_id:
            channel = member.guild.get_channel(channel_id)
            if channel:
                await channel.send(f"{member.display_name} has joined the server.")

    @commands.Cog.listener()
    async def on_member_remove(self, member):
        await self.__check_reset(member.guild.id)
        await self.__update_stat(member.guild.id, 'left')
        channel_id = await self.config.guild(member.guild).traffic_channel()
        if channel_id:
            channel = member.guild.get_channel(channel_id)
            if channel:
                await channel.send(f"{member.display_name} has left the server.")

    @commands.Cog.listener()
    async def on_member_ban(self, guild, member):
        await self.__check_reset(guild.id)
        await self.__update_stat(guild.id, 'banned')
        channel_id = await self.config.guild(guild).traffic_channel()
        if channel_id:
            channel = guild.get_channel(channel_id)
            if channel:
                await channel.send(f"{member.display_name} has been banned from the server.")

def setup(bot):
    bot.add_cog(TrafficCog(bot))