Kanium-PUBLIC/KaniumCogs
5
0
Fork 0
Code Issues 6 Pull Requests Packages Projects 1 Releases Wiki Activity

Attempting to fix missing join messages

Browse Source
This commit is contained in:
AllfatherHatt 2025-02-20 15:39:16 +01:00
parent 1c00b81d9a
commit 4639877767
1 changed files with 36 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
trafficCog/trafficCog.py
View File

@ -1,6 +1,6 @@
import discord
from redbot.core import Config, commands # This line imports commands as well
from datetime import datetime, timedelta
from redbot.core import Config, commands
from datetime import datetime
import pytz
class TrafficCog(commands.Cog):
@ -17,14 +17,14 @@ class TrafficCog(commands.Cog):
}
self.config.register_guild(**default_guild)
async def __check_reset(self, guild_id):
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'])
last_reset = datetime.fromisoformat(guild_data.get('last_reset', datetime.now(pytz.UTC).isoformat()))
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 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
@ -44,53 +44,56 @@ class TrafficCog(commands.Cog):
@traffic_commands.command(name='stats')
async def show_stats(self, ctx):
"""Displays current traffic statistics."""
await self.__check_reset(ctx.guild.id)
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']
daily_stats = guild_data.get('daily_stats', {"joined": 0, "left": 0, "banned": 0})
total_stats = guild_data.get('total_stats', {"joined": 0, "left": 0, "banned": 0})
thumbnail_url = guild_data.get('stats_thumbnail_url', 'https://example.com/default-thumbnail.png')
embed = discord.Embed(title="Server Traffic Stats", description="Statistics on server activity", color=0x3399ff)
embed.set_thumbnail(url=guild_data['stats_thumbnail_url'])
embed.set_thumbnail(url=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')
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 not channel_id:
return
channel = member.guild.get_channel(channel_id) or await member.guild.fetch_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')
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 not channel_id:
return
channel = member.guild.get_channel(channel_id) or await member.guild.fetch_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')
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 not channel_id:
return
channel = guild.get_channel(channel_id) or await guild.fetch_channel(channel_id)
if channel:
await channel.send(f"{member.display_name} has been banned from the server.")
def setup(bot):
cog = TrafficCog(bot)
bot.add_cog(cog)