from datetime import datetime from typing import Literal import dateutil.parser import discord from redbot.core import Config, commands from redbot.core.bot import Red from redbot.core.commands import Cog from redbot.core.utils import AsyncIter class LastSeen(Cog): """ Report when a user was last seen online """ online_status = discord.Status.online offline_status = discord.Status.offline def __init__(self, bot: Red): super().__init__() self.bot = bot self.config = Config.get_conf(self, identifier=9811198108111121, force_registration=True) default_global = {} default_guild = {"enabled": False} default_member = {"seen": None} self.config.register_global(**default_global) self.config.register_guild(**default_guild) self.config.register_member(**default_member) async def red_delete_data_for_user( self, *, requester: Literal["discord_deleted_user", "owner", "user", "user_strict"], user_id: int, ): all_members = await self.config.all_members() async for guild_id, guild_data in AsyncIter(all_members.items(), steps=100): if user_id in guild_data: await self.config.member_from_ids(guild_id, user_id).clear() @staticmethod def get_date_time(s): d = dateutil.parser.parse(s) return d @commands.group(aliases=["setlseen"], name="lseenset") async def lset(self, ctx: commands.Context): """Change settings for lseen""" if ctx.invoked_subcommand is None: pass @lset.command(name="toggle") async def lset_toggle(self, ctx: commands.Context): """Toggles tracking seen for this server""" enabled = not await self.config.guild(ctx.guild).enabled() await self.config.guild(ctx.guild).enabled.set(enabled) await ctx.maybe_send_embed( "Seen for this server is now {}".format("Enabled" if enabled else "Disabled") ) @commands.command(aliases=["lastseen"]) async def lseen(self, ctx: commands.Context, member: discord.Member): """ Just says the time the user was last seen """ if member.status != self.offline_status: last_seen = datetime.utcnow() else: last_seen = await self.config.member(member).seen() if last_seen is None: await ctx.maybe_send_embed( embed=discord.Embed(description="I've never seen this user") ) return last_seen = self.get_date_time(last_seen) # embed = discord.Embed( # description="{} was last seen at this date and time".format(member.display_name), # timestamp=last_seen) embed = discord.Embed(timestamp=last_seen) await ctx.send(embed=embed) @commands.Cog.listener() async def on_member_update(self, before: discord.Member, after: discord.Member): if before.status != self.offline_status and after.status == self.offline_status: if await self.bot.cog_disabled_in_guild(self, after.guild): return if not await self.config.guild(after.guild).enabled(): return await self.config.member(before).seen.set(datetime.utcnow().isoformat())