import datetime as dt import hikari import lightbulb from lightbulb.ext import tasks import config as c plugin = lightbulb.Plugin("activity") # Check every minute if inactive (or config duration if under 60 secs) @tasks.task(s=60 if c.config["duration"] >= 60 else c.config["duration"]) async def check_activity(): for author_id, timestamp in c.db.items(): # If time between now and timestamp >= duration if dt.datetime.now(dt.timezone.utc) - timestamp >= dt.timedelta(seconds=c.config["duration"]): try: # Acquire member object member = plugin.bot.cache.get_member( c.config["guild"], author_id ) or await plugin.bot.rest.fetch_member(c.config["guild"], author_id) # Delete member from db if it has excluded role if c.config["exclude"] in member.role_ids: del c.db[author_id] continue # Enforce activity roles if c.config["active"] in member.role_ids: await member.remove_role(c.config["active"]) if c.config["inactive"] not in member.role_ids: await member.add_role(c.config["inactive"]) # Delete member from db if not found except hikari.NotFoundError: del c.db[author_id] # Listener for bot ready @plugin.listener(hikari.StartedEvent) async def on_ready(event): check_activity.start() # Check activity and update timestamp async def update_activity(event, member): # Exclude bots and other guilds if member.is_bot or event.guild_id != c.config["guild"]: return # Exclude and remove activity roles from excluded role if c.config["exclude"] in member.role_ids: if c.config["active"] in member.role_ids: await member.remove_role(c.config["active"]) if c.config["inactive"] in member.role_ids: await member.remove_role(c.config["inactive"]) return # Insert current timestamp into db c.db[member.id] = dt.datetime.now(dt.timezone.utc) # Toggle activity roles if c.config["active"] not in member.role_ids: await member.add_role(c.config["active"]) if c.config["inactive"] in member.role_ids: await member.remove_role(c.config["inactive"]) # Listener for guild messages @plugin.listener(hikari.GuildMessageCreateEvent) async def on_message(event): await update_activity(event, event.member) # Listener for guild typing @plugin.listener(hikari.GuildTypingEvent) async def on_typing(event): await update_activity(event, event.member) # Listener for voice state @plugin.listener(hikari.VoiceStateUpdateEvent) async def on_voice(event): await update_activity(event, event.state.member) def load(bot): bot.add_plugin(plugin) def unload(bot): bot.remove_plugin(plugin)