You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
3.1 KiB
95 lines
3.1 KiB
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.send(
|
|
"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.send(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 not await self.config.guild(before.guild).enabled():
|
|
return
|
|
await self.config.member(before).seen.set(datetime.utcnow().isoformat())
|