iris-discord-bot/bot/commands/admin.py

#import discord
import secrets
from discord.ext import commands
from local_settings import WEB_SCHEME, WEB_HOST
from query.guild_access_token import get_active_token, upsert_token
from common.datetime import plus10min
from common.logging import report
from common.settings import check_ignore


async def setup(bot: commands.Bot):
	await bot.add_cog(Admin(bot))


class Admin(commands.Cog):
	"""Administrative functionality."""

	def __init__(self, bot: commands.Bot):
		self.bot = bot

	@commands.command(
		description="Modify channel settings via a browser",
		brief="Set channel specific settings via the webgui",
		help="Sends a single-use time based token to the webportal"
	)
	@commands.has_guild_permissions(administrator=True)
	async def webset(self, ctx: commands.Context):

		# Halt on ignore list.
		if await check_ignore(self.bot.pg, ctx.author):
			return

		record = await get_active_token(self.bot.pg, ctx.guild.id)

		if record:  # Check for active token
			await ctx.send(f"Token {record['id']} is in use by {record['user']} until {plus10min(record['created'])}.")
			await report(self.bot, f"`{ctx.author}` has requested a token for `{ctx.guild.name}`, but got rejected until {plus10min(record['created'])} for use by `{record['user']}`.", ctx.guild)
		else:  # No active token
			token = secrets.token_urlsafe(40)[:40]
			await upsert_token(self.bot.pg, ctx.guild.id, ctx.author.id, token)

			await ctx.author.send(f"{WEB_SCHEME}://{WEB_HOST}/config/channel-settings/{ctx.channel.id}/{token}")	# DM token
			await ctx.send("Your access token has been sent to you in a private DM.")
			await report(self.bot, f"`{ctx.author}` has requested a token for `{ctx.guild.name}`.", ctx.guild)