Added the ability to save a steam account per user
This commit is contained in:
parent
938337688b
commit
cf06356482
1 changed files with 25 additions and 4 deletions
|
@ -35,8 +35,8 @@ class Steam:
|
||||||
|
|
||||||
@commands.command(pass_context=True)
|
@commands.command(pass_context=True)
|
||||||
@checks.custom_perms(send_messages=True)
|
@checks.custom_perms(send_messages=True)
|
||||||
async def csgo(self, ctx, profile: str):
|
async def steam(self, ctx, profile: str):
|
||||||
"""This command can be used to lookup csgo stats for a user"""
|
"""This command can be used to link a steam profile to your user"""
|
||||||
# Attempt to find the user/steamid based on the url provided
|
# Attempt to find the user/steamid based on the url provided
|
||||||
# If a url is not provided that matches steamcommunity.com, assume they provided just the user/id
|
# If a url is not provided that matches steamcommunity.com, assume they provided just the user/id
|
||||||
try:
|
try:
|
||||||
|
@ -55,6 +55,27 @@ class Steam:
|
||||||
await self.bot.say("Sorry, couldn't find that Steam user!")
|
await self.bot.say("Sorry, couldn't find that Steam user!")
|
||||||
return
|
return
|
||||||
|
|
||||||
|
# Save the author's steam ID, ensuring to only overwrite the steam id if they already exist
|
||||||
|
author = ctx.message.author
|
||||||
|
steam_users = config.get_content('steam_users') or {}
|
||||||
|
if steam_users.get(author.id):
|
||||||
|
steam_users[author.id]['steam_id'] = steam_id
|
||||||
|
else:
|
||||||
|
steam_users[author.id] = {'steam_id': steam_id}
|
||||||
|
|
||||||
|
config.save_content('steam_users', steam_users)
|
||||||
|
await self.bot.say("I have just saved your steam account, you should now be able to view stats for your account!")
|
||||||
|
|
||||||
|
@commands.command(pass_context=True)
|
||||||
|
@checks.custom_perms(send_messages=True)
|
||||||
|
async def csgo(self, ctx, member: discord.Member):
|
||||||
|
"""This command can be used to lookup csgo stats for a user"""
|
||||||
|
try:
|
||||||
|
steam_id = config.get_content('steam_users').get(ctx.message.author.id).get('steam_id')
|
||||||
|
except AttributeError:
|
||||||
|
await self.bot.say("Sorry, but I don't have that user's steam account saved!")
|
||||||
|
return
|
||||||
|
|
||||||
url = "{}&appid=730&steamid={}".format(base_url, steam_id)
|
url = "{}&appid=730&steamid={}".format(base_url, steam_id)
|
||||||
async with self.session.get(url, headers=self.headers) as response:
|
async with self.session.get(url, headers=self.headers) as response:
|
||||||
data = await response.json()
|
data = await response.json()
|
||||||
|
|
Loading…
Reference in a new issue