async def getStats(key, userid):
class Stats:
username = ""
rank = 0
api = OsuApi(key, connector=AHConnector())
Data = await api.get_user(userid)
Stats.username = Data[0].username
Stats.rank = Data[0].pp_rank
api.close()
return (Stats)
async def getResult(key, id, maps):
class Result:
datetime = ""
users = []
wins = []
score = {}
totalScore = {}
scoreDiff = {}
api = OsuApi(key, connector=AHConnector())
Data = await api.get_match(id)
Result.datetime = Data.match.end_time
Games = Data.games
Result.users = [(Games[0].scores[0].user_id), (Games[0].scores[1].user_id)]
for game in Games:
if str(game.beatmap_id) in maps:
stats = game.scores
Result.score[stats[0].user_id] = Result.score.get(
stats[0].user_id, 0)
Result.score[stats[1].user_id] = Result.score.get(
stats[1].user_id, 0)
Result.totalScore[stats[0].user_id] = Result.totalScore.get(
stats[0].user_id, 0) + stats[0].score
Result.totalScore[stats[1].user_id] = Result.totalScore.get(
stats[1].user_id, 0) + stats[1].score
if (stats[0].score > stats[1].score):
Result.wins.append(stats[0].user_id)
Result.score[stats[0].user_id] = Result.score.get(
stats[0].user_id, 0) + 1
Result.scoreDiff[stats[0].user_id] = Result.scoreDiff.get(
stats[0].user_id, 0) + (stats[0].score - stats[1].score)
elif (stats[0].score < stats[1].score):
Result.wins.append(stats[1].user_id)
Result.score[stats[1].user_id] = Result.score.get(
stats[1].user_id, 0) + 1
Result.scoreDiff[stats[1].user_id] = Result.scoreDiff.get(
stats[1].user_id, 0) + (stats[1].score - stats[0].score)
else:
pass
else:
pass
api.close()
return (Result)
user_best.save(parse_user_recent(result, user_id))
collected += 1
'''
else:
user.remove({'user_id': str(user_id)})
user_best.remove({'user_id': str(user_id)})
user_recent.remove({'user_id': str(user_id)})
break
'''
except aiohttp.client_exceptions.ClientPayloadError:
print('get_user_best(): ' + str(user_id) + ' not successful.')
print('user: '******': completed.')
print('Data collected: ', collected)
i += 1
api.close()
'''
user_id = Attribute(int)
username = Attribute(str)
count300 = Attribute(Nullable(int))
count100 = Attribute(Nullable(int))
count50 = Attribute(Nullable(int))
playcount = Attribute(Nullable(int))
ranked_score = Attribute(Nullable(int))
total_score = Attribute(Nullable(int))
pp_rank = Attribute(Nullable(int))
level = Attribute(Nullable(float))
pp_raw = Attribute(Nullable(float))
class Osu:
def __init__(self, bot):
self.bot = bot
if bot.config['osu']:
self.api = OsuApi(bot.config['osu'], connector=AHConnector())
else:
self.api = None
@classmethod
def osu_mode_converter(self, mode=None):
if mode is 0 or 'standard' or 'osu!standard' or 'osu!' or None:
return osuapi.enums.OsuMode.osu
elif mode is 1 or 'ctb' or 'catchthebeat' or 'osu!catch' or 'catch':
return osuapi.enums.OsuMode.catch
elif mode is 2 or 'taiko' or 'osu!taiko':
return osuapi.enums.OsuMode.taiko
elif mode is 3 or 'mania' or 'osu!mania':
return osuapi.enums.OsuMode.mania
else:
return 'Unknown'
def __unload(self):
logging.info('osu[api]: Closing session.')
try:
self.api.close()
except Exception:
logging.error('osu[api]: Wow, amazing! '
'You closed the instance early. GJ.')
else:
logging.info('osu[unload]: OK')
@commands.group()
async def osu(self, ctx):
"""Commands for osu!"""
if ctx.invoked_subcommand is None:
help_em = discord.Embed(title='Commands for osu!', colour=0x690E8)
help_em.add_field(name='user', value='Gets info on osu! players.'
' `^osu user *user*`')
await ctx.send(embed=help_em)
@osu.command()
async def user(self, ctx, u: str, mode=None):
"""Returns information on a osu! player.
If the player name you are searching has spaces, use quotation marks.
e.g. ^osu user "player name with spaces"
Special thanks to khazhyk for the library this command uses.
By default this command defaults to osu!standard.
All modes are supported.
To use osu!standard, leave mode blank, or use 'standard',
'osu!standard', 'osu!' or 0.
To use osu!catch, use 'catch', 'osu!catch', or 1.
To use osu!taiko, use 'taiko', 'osu!taiko', or 2.
To use osu!mania, use 'mania', 'osu!mania', or 3.
Any other modes will return 'Unknown' error. (Service error)
"""
if self.api:
mode = self.osu_mode_converter(mode=mode)
if mode == 'Unknown':
raise utils.errors.ServiceError('Unknown mode')
user = await self.api.get_user(u, mode=mode)
try:
user = user[0]
pp = user.pp_raw # CHAR SAVING.
ss = user.count_rank_ss
s = user.count_rank_s
a = user.count_rank_a
except IndexError:
return await ctx.send('User does not exist, '
'maybe try one that does')
else:
raise utils.errors.ServiceError('osu! api key not configured')
osu_embed = discord.Embed(title=f'osu! stats', colour=0x690E8)
osu_embed.set_author(name=f'{u} ({user.country}'
f' #{user.pp_country_rank}, global'
f' #{user.pp_rank})',
icon_url='https://osu.ppy.sh/images/flags/'
f'{user.country}.png')
osu_embed.set_thumbnail(url=f'https://a.ppy.sh/{user.user_id}')
osu_embed.add_field(name='Hits (300/100/50)', value=f'{user.count300}/'
f'{user.count100}/'
f'{user.count50}')
osu_embed.add_field(name='Play count', value=user.playcount)
osu_embed.add_field(name='Ranked score', value=user.ranked_score)
osu_embed.add_field(name='Total score', value=user.total_score)
osu_embed.add_field(name='Level', value=int(user.level))
osu_embed.add_field(name='Total PP', value=f'{round(pp, 2)} PP')
osu_embed.add_field(name='Accuracy', value=f'{user.accuracy:.1f}%')
osu_embed.add_field(name='Plays (SS/S/A)', value=f'{ss}/{s}/{a}')
await ctx.send(embed=osu_embed)
