diff --git a/geeksbot/exts/rcon.py b/geeksbot/exts/rcon.py index 882d758..3fb2e93 100644 --- a/geeksbot/exts/rcon.py +++ b/geeksbot/exts/rcon.py @@ -408,6 +408,20 @@ class Rcon(commands.Cog): else: await ctx.send('I need a list of members to whitelist.') + + @commands.command(name="add_steamid", aliases=['steamid']) + @checks.is_moderator() + async def _add_steam_id_to_user(self, ctx, member: discord.Member, steam_id: int): + if isinstance(member, discord.Member): + resp = await self.bot.aio_session.patch(f'{self.bot.api_base}/users/{member.id}/', + headers=self.bot.auth_header, + json={'steam_id': steam_id}) + if resp.status == 200: + await ctx.message.add_reaction(self.bot.success_emoji) + return + await ctx.send('I couldn\'t update that user for some reason.') + return + await ctx.send('Are you sure that is a valid user?') # # @commands.command(name='new_patron') # @commands.guild_only() diff --git a/geeksbot/imports/geeksbot.py b/geeksbot/imports/geeksbot.py index 406d0af..2b365a7 100644 --- a/geeksbot/imports/geeksbot.py +++ b/geeksbot/imports/geeksbot.py @@ -39,7 +39,7 @@ class Geeksbot(commands.Bot): self.git_url = 'https://github.com/dustinpianalto/geeksbot_v2' self.load_default_extensions() self.owner_id = 351794468870946827 - + self.success_emoji = '\N{WHITE CHECK MARK}' self.book_emojis = { 'unlock': '🔓', 'start': '⏮',