import os # OS Locations import yaml # YAML parser for Bot config files import asyncio # Discord Py Dependency import discord # Main Lib from discord.ext import commands # Commands module from discord_slash import SlashCommand, SlashContext, cog_ext, utils # Slash Command Library from discord_slash.utils.manage_commands import create_choice, create_option, create_permission # Slash Command features from discord_slash.model import SlashCommandPermissionType, ButtonStyle from discord_slash.client import SlashCommand from discord_slash.utils.manage_components import create_select, create_select_option, create_actionrow, wait_for_component, create_button, create_actionrow from bot import configFile, yaml_load, yaml_dump, cogsDir, unloadCog, dataFile, lookupFile, gmFile, categoriesFile, pitchesFile, loadCog #### Pitch Command class Pitch(commands.Cog, name='Pitch Command'): def __init__(self, client): self.client = client conf=yaml_load(configFile) permissions={} guild_ids = list(set.intersection(set([int(guildKey) for guildKey in yaml_load(configFile) if yaml_load(configFile)[guildKey]['timeslots']]),set([int(guildKey) for guildKey in yaml_load(configFile) if 'bot' in yaml_load(configFile)[guildKey]['roles'] and type(yaml_load(configFile)[guildKey]['roles']['bot']) is int]))) for guildID in guild_ids: permissions[guildID] = [] permissions[guildID].append(create_permission(id=conf[str(guildID)]['owner'],id_type=SlashCommandPermissionType.USER,permission=True)) for admin in conf[str(guildID)]['roles']['admin']: permissions[guildID].append(create_permission(id=admin,id_type=SlashCommandPermissionType.ROLE,permission=True)) permissions[guildID] = [] permissions[guildID].append(create_permission(id=conf[str(guildID)]['owner'],id_type=SlashCommandPermissionType.USER,permission=True)) for admin in conf[str(guildID)]['roles']['admin']: permissions[guildID].append(create_permission(id=admin,id_type=SlashCommandPermissionType.ROLE,permission=True)) @cog_ext.cog_slash( name='pitch', description='Designates the various key roles referenced by the Bot.', default_permission=False, permissions=permissions, guild_ids=guild_ids, ) async def _pitch(self, ctx:SlashContext): await ctx.channel.trigger_typing() conf = yaml_load(configFile) data = yaml_load(dataFile) lookup = yaml_load(lookupFile) gms = yaml_load(gmFile) categories = yaml_load(categoriesFile) guildStr = str(ctx.guild.id) pitches = yaml_load(pitchesFile) if guildStr not in pitches: pitches[guildStr] = {} if pitches[guildStr]: await ctx.send(f'```Error: pitches are already running for the guild `{ctx.guild.name}`. Please close the existing pitches first before issuing this command.```') return if 'timeslots' not in conf[guildStr]: conf[guildStr]['timeslots'] = {} tsDict = {k: conf[guildStr]['timeslots'][k] for k in data[guildStr] if data[guildStr][k]} optionsList = [create_select_option(label=tsDict[x], value=x, description=x) for x in tsDict] try: m = await ctx.send( content='```Select which time slot for which you would like to run pitches for.```', delete_after=5, components=[ create_actionrow( create_select( placeholder='Time Slot', options= optionsList, min_values=1, max_values=1 ) ) ] ) while True: select_ctx = await wait_for_component(self.client,messages=m, timeout=5) if select_ctx.author != ctx.author: await select_ctx.send(f'```Invalid response: you are not the person who issued the command.```', hidden=True) else: break await m.delete() [timeslot] = select_ctx.selected_options except asyncio.TimeoutError: await ctx.send(f'```Error: Command timed out.```', hidden=True) return await ctx.channel.trigger_typing() p = discord.PermissionOverwrite() p.read_messages = False p.send_messages = False await ctx.channel.edit( reason=f'/pitch command issued by {ctx.author.display_name}', overwrites = { ctx.guild.default_role: p } ) if timeslot not in pitches[guildStr]: pitches[guildStr][timeslot] = {} pitches[guildStr][timeslot]['indices'] = {} pitches[guildStr][timeslot]['entries'] = [x for x in data[guildStr][timeslot].values()] pitches[guildStr][timeslot]['entries'].sort(key= lambda x: x['game_title']) header_message = await ctx.channel.send( content=f'**Game listing for {conf[guildStr]["timeslots"][timeslot]}**\n_ _```The following are the games that are being pitched. Please select which game you would like to join by clicking on the `Join` button below.```' ) pitches[guildStr][timeslot]['header_message'] = header_message.id pitches[guildStr][timeslot]['messages'] = [] pitches[guildStr][timeslot]['roles'] = {} for index, element in enumerate(pitches[guildStr][timeslot]['entries']): gm = await self.client.fetch_user(element["gm"]) o = f'_ _\n***{element["game_title"]}*** (GM: {gm.mention})\n```\n' if element['system'] is not None: o = ''.join([o,f'System: {element["system"]}\n']) if element['min_players'] is not None: o = ''.join([o,f'Minimum Players: {str(element["min_players"])} ']) if element['max_players'] is not None: o = ''.join([o,f'Maximum Players: {str(element["max_players"])}\n']) if element['platform'] is not None: o = ''.join([o,f'Platform: {element["platform"]}\n']) o = ''.join([o,f'```']) spaces_remaining = element["max_players"] - element["current_players"] o = ''.join([o,f'~~Spaces Remaining: {str(0)}~~'])if spaces_remaining <= 0 else ''.join([o,f'Spaces Remaining: {str(spaces_remaining)}']) m = await ctx.channel.send( content=o, components=[ create_actionrow( create_button( style=ButtonStyle.green, label='Join', emoji='🉑', custom_id=f'join_{index}' ), create_button( style=ButtonStyle.red, label='Leave', emoji='🈳', custom_id=f'leave_{index}' ) ) ] ) pitches[guildStr][timeslot]['messages'].append(m.id) r = discord.utils.find(lambda x: x.id == element['role'],ctx.guild.roles) pitches[guildStr][timeslot]['roles'][index] = r.id pitches[guildStr][timeslot]['indices'][r.id] = index newcomer = returning_player = None if 'newcomer' in conf[guildStr]['roles']: newcomer = discord.utils.find(lambda x: x.id == conf[guildStr]['roles']['newcomer'], ctx.guild.roles) if 'returning_player' in conf[guildStr]['roles']: returning_player = discord.utils.find(lambda x: x.id == conf[guildStr]['roles']['returning_player'], ctx.guild.roles) buttons = [] if returning_player is not None: buttons.append( create_button( style= ButtonStyle.grey, label= 'Allow Returning Players', emoji= '🔁', custom_id='allow_returning' ) ) if newcomer is not None: buttons.append( create_button( style= ButtonStyle.grey, label= 'Allow Newcomers', emoji= '🆕', custom_id='allow_newcomers' ) ) buttons.append( create_button( style= ButtonStyle.green, label= 'Allow All', emoji='🚪', custom_id='allow_all' ) ) buttons.append( create_button( style= ButtonStyle.red, label= 'Close Pitches', emoji='🔒', custom_id='close_pitches' ) ) control = await ctx.channel.send( content='_ _\n```Control Panel:\nFor Admin Use Only```', components=[ create_actionrow( *buttons ) ] ) pitches[guildStr][timeslot]['control'] = control.id yaml_dump(pitches,pitchesFile) if self.client.get_cog('Pitch Listener') is None: loadCog(f'./{cogsDir}/events/secondary/pitch_listener.py') #### Activate global pitch listener def setup(client): client.add_cog(Pitch(client))