autobotrobot/src/heavserver.py

import util
import random
import logging
import discord
import prometheus_client

heavserver_members = prometheus_client.Gauge("abr_heavserver_members", "Current member count of heavserver")

def setup(bot):
	@bot.listen()
	async def on_member_join(member):
		if member.guild.id == util.config["heavserver"]["id"]:
			logging.info("%s (%d) joined heavserver", member.display_name, member.id)
			heavserver_members.set(len(bot.get_guild(util.config["heavserver"]["id"]).members))
			if member.bot:
				await member.add_roles(discord.utils.get(member.guild.roles, id=util.config["heavserver"]["quarantine_role"]))
			await member.add_roles(discord.utils.get(member.guild.roles, id=random.choice(util.config["heavserver"]["moderator_roles"][:])))