for i in range(25):
str2 += ",PLAY"+str(i)+" TEXT"
c.execute("CREATE TABLE IF NOT EXISTS PLAYERS(%s)"%str2)
#get beatmaps of similar users
counter = 0
for player in similar_users:
counter += 1
if counter % 50 == 0:
print("Creating record for %d"%counter)
conn.commit()
c.execute("SELECT * FROM PLAYERS WHERE PLAYER_ID=?",(player,))
data=c.fetchone()
if data is None:
results = api.get_user_best(int(player), limit=25)
scores = [player]
for x in results:
scores.append(str(x.beatmap_id) + " " + str(x.enabled_mods))
try:
c.execute("INSERT INTO PLAYERS VALUES(?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)", tuple(scores))
except:
pass
conn.commit()
results = api.get_user_best(playername, limit=25)
userscores = []
scoreidonly = []
for x in results:
score = str(x.beatmap_id) + " " + str(x.enabled_mods)
userscores.append(score)
class OsuTracker(commands.Cog):
def __init__(self, bot):
self.bot = bot
self.config = confparser.get("config.json")
self.api = OsuApi(key=self.config.osu_token, connector=ReqConnector())
self.last_check_datetime = datetime.datetime.utcnow()
self.tracking_channel = None
self.tracked_users = self.get_tracked_users()
self.tracker_task_loop.start()
def cog_unload(self):
self.tracker_task_loop.cancel()
@tasks.loop(seconds=30)
async def tracker_task_loop(self):
for user in self.tracked_users:
scores = self.api.get_user_best(user['name'], limit=20)
for s in [s for s in scores if s.date > self.last_check_datetime]:
bmap = self.api.get_beatmaps(beatmap_id=s.beatmap_id)[0]
msg = f"{user['nick']} abım {bmap.title} - {bmap.version} mapınde {int(round(s.pp))} pp play yapmıs\n" \
f"helal abım <@{user['discord'].id}>\n".lower().replace('i', 'ı')
await self.tracking_channel.send(msg)
self.last_check_datetime = datetime.datetime.utcnow()
def get_tracked_users(self):
with open("tracked_users.json", "r") as f:
data = json.load(f) if not json.load(f) == {} else None
users = []
if data:
for u in data['users']:
users.append({
"name": u['name'],
"discord": self.bot.get_user(id=u['discord_id']),
"nick": u['nick']
})
return users
def add_to_tracked_users_file(self, osu_username, discord_id, nick):
with open("tracked_users.json", "w") as f:
data = json.load(f)
if data == {}:
data = {
"users": [{
"name": osu_username,
"discord_id": discord_id,
"nick": nick
}]
}
else:
data['users'].append({
"name": osu_username,
"discord_id": discord_id,
"nick": nick
})
data.update()
print(data)
json.dump(data, f)
@commands.command()
@commands.check(permissions.is_owner)
async def osutrack(self, ctx, mem: discord.Member, osu_username, nick):
if osu_username not in [u['name'] for u in self.tracked_users]:
try:
osu_id = self.api.get_user(osu_username)[0].user_id
self.tracked_users.append({
"name": osu_username,
"nick": nick,
"discord": mem
})
self.add_to_tracked_users_file(osu_username,
discord_id=mem.id,
nick=nick)
await ctx.send(
f"tracking started : {str(mem)}, osu id : {osu_id}")
except:
await ctx.send(f"couldn't start tracking")
else:
await ctx.send(f"already tracking : {str(mem)}")
@commands.command()
@commands.check(permissions.is_owner)
async def setch(self, ctx):
self.tracking_channel = ctx
await ctx.send(f"ok")
class Osu(commands.Cog):
def __init__(self, discordClient):
with open("config.json") as f:
config = json.loads(f.read())
self.bot = discordClient
self.osuApi = OsuApi(config["osu_key"], connector=ReqConnector())
self.modeMap = {'standard': enums.OsuMode.osu, 'mania': enums.OsuMode.mania}
self.db = 'guild_settings.db'
self.table = 'settings'
self.settings = {'guild_id': 'guild_id',
'guild_name': 'guild_name',
'commands_disabled': 'commands_disabled',
'roll_channel': 'roll_channel',
'osu_channel': 'osu_channel',
'admin_role': 'admin_role'}
def check_correct_channel(self, message):
conn = sqlite3.connect(self.db)
c = conn.cursor()
with conn:
c.execute("SELECT {} FROM {} WHERE guild_id=?".format(self.settings['osu_channel'], self.table), (message.guild.id,))
channel = c.fetchone()
conn.close()
return(not channel[0] == message.channel.name)
@commands.group()
async def osu(self, ctx):
if ctx.invoked_subcommand is None:
return(await ctx.send('Invalid subcommand for osu passed...'))
@osu.command()
async def profile(self, ctx, profile, mode_arg="standard"):
if self.check_correct_channel(ctx.message):
return(await ctx.send('You cannot use this command in this channel'))
if mode_arg not in self.modeMap:
return(await ctx.send('Not a valid game mode'))
else:
result = self.osuApi.get_user(str(profile),mode=self.modeMap[mode_arg], event_days=31)
if not result:
return(await ctx.send("User doesn't exist"))
data = result[0] #get first and only result
acc = str(math.ceil(data.accuracy*100)/100) + "%"
lvl = str(math.ceil(data.level*100)/100)
profile_embed = discord.Embed(
url = "https://osu.ppy.sh/users/{}".format(data.user_id),
color = 0xea1595
)
profile_embed.add_field(
name = "Performance",
value = "Global PP Rank: {0}\nPP: {1}\nAccuracy: {2}".format(data.pp_rank, data.pp_raw, acc),
inline = True
)
profile_embed.add_field(
name = "Experience",
value = "Playcount: {0}\nLevel: {1}\n".format(data.playcount, lvl),
inline = True
)
profile_embed.add_field(
name = "Country",
value = emoji.emojize("Country PP Rank: {0}\n:flag_{1}:"
.format(data.pp_country_rank, data.country.lower())
),
inline = True
)
profile_embed.set_author(
name = data.username,
url = "https://osu.ppy.sh/users/{}".format(data.user_id),
icon_url = "http://s.ppy.sh/a/{}".format(data.user_id)
)
return(await ctx.send(embed=profile_embed))
@osu.command()
async def best(self, ctx, username, mode_arg="standard"):
if self.check_correct_channel(ctx.message):
return(await ctx.send('You cannot use this command in this channel'))
if mode_arg not in self.modeMap:
return(await ctx.send('Not a valid game mode'))
else:
result = self.osuApi.get_user_best(username,mode=self.modeMap[mode_arg], limit=1)
if not result:
return(await ctx.send("User doesn't exist"))
data = result[0] #get first and only result
best_embed = discord.Embed(
url = "https://osu.ppy.sh/users/{}".format(data.user_id),
color = 0xea1595
)
best_embed.add_field(
name="Score",
value="Score: {0}\nMax Combo: {1}\nPerfect: {2}".format(data.score, data.maxcombo, data.perfect),
inline = True
)
best_embed.add_field(
name="Count",
value="Count50: {0}\nCount100: {1}\nCount300: {2}\nMisses: {3}".format(data.count50, data.count100, data.count300, data.countmiss),
inline=True
)
best_embed.set_author(
name = username,
url = "https://osu.ppy.sh/users/{}".format(data.user_id),
icon_url = "http://s.ppy.sh/a/{}".format(data.user_id)
)
return(await ctx.send(embed=best_embed))
@best.error
@profile.error
async def error(self, ctx, error):
return(await ctx.send(str(error).capitalize()))
class OsuModule:
def __init__(self, client):
self.client = client
self.database = self.client.database
self.api = OsuApi(self.database.get_osu_api_key(),
connector=ReqConnector())
# Completed with Rich embed.
@commands.command(name='osu')
async def osu(self, ctx, *params, member: discord.Member = None):
if member is None and len(params) == 0:
discord_id = ctx.message.author.id
self.database.cursor.execute(
"SELECT `osu_id` FROM `users` WHERE `discord_id` = ?",
(discord_id, ))
osu_id = self.database.cursor.fetchone()[0]
embed = self.display_profile(osu_id)
await ctx.send(embed=embed)
elif len(params) > 0:
for osu_id in params:
embed = self.display_profile(osu_id)
await ctx.send(embed=embed)
if member is not None:
print(member.id)
self.database.cursor.execute(
"SELECT `osu_id` FROM `users` WHERE `discord_id` = ?",
(member.id, ))
osu_id = self.database.cursor.fetchone()[0]
embed = self.display_profile(osu_id)
await ctx.send(embed=embed)
@commands.command(name='top')
async def top(self, ctx, *params):
(user, amt) = self.params_seperator(ctx, *params)
embed = self.top_scores(user, amt)
await ctx.send(embed=embed)
@commands.command(name='recent') # Completed With Rich Embed.
async def recent(self, ctx, *params):
(user, amt) = self.params_seperator(ctx, *params)
embed = self.recent_scores(user, amt)
await ctx.send(embed=embed)
@commands.command(name='set')
async def set(self, ctx, param):
discord_id = ctx.message.author.id
osu_id = self.api.get_user(param)[0].user_id
self.database.cursor.execute(
"SELECT `osu_id` FROM `users` WHERE `discord_id` = ?",
(discord_id, ))
data = self.database.cursor.fetchone()
if data is None:
self.database.cursor.execute(
"INSERT INTO `users`(`discord_id`, `osu_id`, `days`, `total`) VALUES (?, ?, 0, 0)",
(discord_id, osu_id))
self.database.db.commit()
title = 'Succesfully set {} IGN as {}'.format(
ctx.message.author, param)
embed = discord.Embed(title=title, colour=0xDEADBF)
await ctx.send(embed=embed)
else:
await ctx.send("Record Already Exists")
@commands.command(name='compare')
async def compare(self, ctx, user1, user2):
embed = self.check(user1, user2)
await ctx.send(embed=embed)
@commands.command(name='topr')
async def topr(self, ctx, *params):
(user, amt) = self.params_seperator(ctx, *params)
embed = self.recent_top(user, amt)
await ctx.send(embed=embed)
def display_profile(self, param):
# Obtaining Profile from Paramerter.
profile = self.api.get_user(param)
# Local Stuff
title = "osu! Profile for " + profile[0].username
thumbnail = "https://a.ppy.sh/" + str(profile[0].user_id)
user_url = "https://osu.ppy.sh/users/" + str(profile[0].user_id)
# Embed Creation.
embed = discord.Embed(title=title,
timestamp=datetime.utcnow(),
url=user_url,
color=0xFF0418,
footer="Osu India bot v.1.0")
# thumbnails
embed.set_thumbnail(url=thumbnail)
# PP
embed.add_field(name="PP", value=str(profile[0].pp_raw), inline=True)
# Rank
embed.add_field(name="Rank",
value='#' + str(profile[0].pp_rank),
inline=True)
# Playcount
embed.add_field(name="Playcount",
value=str(profile[0].playcount),
inline=True)
# Accuracy
embed.add_field(name="Accuracy",
value=str(profile[0].accuracy)[:6],
inline=True)
# Country Indentification.
embed.add_field(name="Country",
value=str(profile[0].country),
inline=True)
# Country Rank.
embed.add_field(name="Country Rank",
value='#' + str(profile[0].pp_country_rank),
inline=True)
return embed
def top_scores(self, user, amt):
# Api Call
scores = self.api.get_user_best(user, limit=amt)
# Local Stuff.
title = "Top {} Scores for {}".format(amt, user)
count = 1
# Embed
embed = discord.Embed(title=title,
timestamp=datetime.utcnow(),
color=0xFF0418,
footer="Osu India bot v.1.0")
# Fields.
for score in scores:
beatmap = self.api.get_beatmaps(beatmap_id=score.beatmap_id)
Title = "#{}. {}[{}] +**{}**".format(count, beatmap[0].title,
beatmap[0].version,
score.enabled_mods)
Value = "PP:{}\n Played {}".format(
score.pp, self.time_elapsed(str(score.date)))
embed.add_field(name=Title, value=Value, inline=False)
count += 1
return embed
def recent_scores(self, user, amt):
scores = self.api.get_user_recent(user, limit=amt)
title = "Recent {} scores for {}".format(amt, user)
count = 1
# Discord Embed Creation.
embed = discord.Embed(title=title,
timestamp=datetime.utcnow(),
color=0xFF0418,
footer="Osu India bot v0.2.0")
# Looping over all Scores and adding fields.
for score in scores:
beatmap = self.api.get_beatmaps(beatmap_id=score.beatmap_id)
name = "#{}. {}[{}] +**{}**".format(count, beatmap[0].title,
beatmap[0].version,
score.enabled_mods)
time_delta = str(score.date)
value = "*Played {}\n SR: {}".format(
self.time_elapsed(time_delta),
str(beatmap[0].difficultyrating)[:5])
embed.add_field(name=name, value=value, inline=False)
count += 1
return embed
# Hacked this together using 3 lists, modifications needed
def recent_top(self, user, amt):
scores_list = []
scores = self.api.get_user_best(user, limit=100)
for i in range(100):
scores_list.append((i + 1, scores[i]))
# sort according to date
scores_sorted = sorted(scores_list,
key=lambda score: score[1].date,
reverse=True)
title = "Recent {} top scores for {}".format(amt, user)
count = 1
# Embed
embed = discord.Embed(title=title,
timestamp=datetime.utcnow(),
color=0xFF0418,
footer="Osu India bot v0.2.0")
# Fields.
for score in scores_sorted:
if count > amt:
break
beatmap = self.api.get_beatmaps(beatmap_id=score[1].beatmap_id)
name = "#{}. {}[{}] +**{}**".format(score[0], beatmap[0].title,
beatmap[0].version,
score[1].enabled_mods)
value = "PP:{}\n Played {}".format(
score[1].pp, self.time_elapsed(str(score[1].date)))
embed.add_field(name=name, value=value, inline=False)
count += 1
return embed
def check(self, user1, user2):
# major formatting required
# to add avatar pic no clue how to
p1 = self.api.get_user(user1)
p2 = self.api.get_user(user2)
title = "Comparing stats for " + user1 + " and " + user2
desc = "\t\t" + user1 + " | " + user2 + "\t\t\n" # \t or multiple spaces not working
desc += "**Rank :**\t " + str(p1[0].pp_rank) + " | " + str(
p2[0].pp_rank) + "\n"
desc += "**Country Rank :**\t " + str(
p1[0].pp_country_rank) + " | " + str(
p2[0].pp_country_rank) + "\n"
desc += "**PP :**\t " + str(p1[0].pp_raw) + " | " + str(
p2[0].pp_raw) + "\n"
desc += "**Accuracy :**\t " + str(p1[0].accuracy)[:5] + " | " + str(
p2[0].accuracy)[:5] + "\n"
score1 = self.api.get_user_best(user1, limit=1)
score2 = self.api.get_user_best(user2, limit=1)
desc += "**Top Play :**\t " + str(score1[0].pp) + " | " + str(
score2[0].pp) + "\n"
desc += "**Playcount :**\t " + str(p1[0].playcount) + " | " + str(
p2[0].playcount) + "\n"
embed = discord.Embed(title=title, description=desc, colour=0xDEADBF)
return embed
def time_elapsed(self, datestr):
parsed_date = datetime.strptime(datestr, "%Y-%m-%d %H:%M:%S")
today = datetime.utcnow()
time_delta = relativedelta(today, parsed_date)
years = abs(time_delta.years)
months = abs(time_delta.months)
days = abs(time_delta.days)
hours = abs(time_delta.hours)
minutes = abs(time_delta.minutes)
seconds = abs(time_delta.seconds)
time_elapsed = ""
if (years > 0):
time_elapsed += "{} year{}, ".format(years,
"s" if years != 1 else "")
if (months > 0):
time_elapsed += "{} month{}, ".format(months,
"s" if months != 1 else "")
if (days > 0):
time_elapsed += "{} day{}, ".format(days, "s" if days != 1 else "")
if (hours > 0):
time_elapsed += "{} hour{}, ".format(hours,
"s" if hours != 1 else "")
if (minutes > 0):
time_elapsed += "{} minute{}, ".format(minutes,
"s" if minutes != 1 else "")
if (seconds > 0):
time_elapsed += "{} second{} ago".format(
seconds, "s" if seconds != 1 else "")
return time_elapsed
def params_seperator(self, ctx, *params):
# Default user AND default amount
if len(params) == 0:
self.database.cursor.execute(
"SELECT `osu_id` FROM `users` WHERE `discord_id` = ?",
(ctx.message.author.id, ))
osu_id = self.database.cursor.fetchone()[0]
if osu_id is None:
return (None, None)
user = self.api.get_user(osu_id)[0].username
amt = 5
# Only default amount
elif len(params) == 1:
user = self.tag_to_id(ctx)
if user is False:
return (None, None)
if user is None:
user = params[0]
amt = 5
# No defaults
else:
user = self.tag_to_id(ctx)
if user is False:
return (None, None)
if user is None:
user = params[0]
amt = params[1]
return (user, amt)
def tag_to_id(self, ctx):
user_list = ctx.message.mentions
if len(user_list) == 0:
return None
else:
user_id = user_list[0].id
self.database.cursor.execute(
"SELECT `osu_id` FROM `users` WHERE `discord_id` = ?",
(user_id, ))
osu_id = self.database.cursor.fetchone()[0]
if osu_id is None:
return False
return self.api.get_user(osu_id)[0].username
def new_recent(self, user_id):
scores = self.api.get_user_best(user_id, limit=50)
new_scores = []
last_checked = datetime.now(
) # need to save the last checked to a file
scores_sorted = sorted(scores, key=lambda s: s.date,
reverse=True) # sort according to date
for score in scores_sorted:
if score.date > last_checked:
new_scores.append(score)
else:
break
print("New Recent was executed")
return new_scores
