mirror of
https://github.com/wlinator/luminara.git
synced 2024-10-02 22:23:13 +00:00
118 lines
4 KiB
Python
118 lines
4 KiB
Python
import time
|
|
|
|
import discord
|
|
from discord.ext import commands
|
|
|
|
from data.Currency import Currency
|
|
from data.Xp import Xp
|
|
from sb_tools import embeds, universal, level_messages, reactions, xp_functions
|
|
|
|
|
|
class Leveling(commands.Cog):
|
|
def __init__(self, sbbot):
|
|
self.bot = sbbot
|
|
|
|
@commands.Cog.listener()
|
|
async def on_message(self, message):
|
|
if message.author.bot:
|
|
return
|
|
|
|
await reactions.check_for_reaction(message)
|
|
|
|
user_id = message.author.id
|
|
current_time = time.time()
|
|
xp_data = Xp.get_user_xp_data(user_id)
|
|
(current_xp, current_level, cooldown) = xp_data[0]
|
|
|
|
if cooldown and current_time < cooldown:
|
|
print(f"XP UPDATE --- {message.author.name} sent a message but is on XP cooldown.")
|
|
return
|
|
|
|
gain_data = Xp.load_gain_data()
|
|
xp_gain = gain_data[0]
|
|
cooldown = gain_data[1]
|
|
|
|
new_xp = current_xp + xp_gain
|
|
needed_xp_for_next_level = xp_functions.xp_needed_for_next_level(current_level)
|
|
|
|
if new_xp >= needed_xp_for_next_level:
|
|
# new level was reached
|
|
new_level = current_level + 1
|
|
new_xp = 0
|
|
|
|
Xp.update_user_xp(user_id, new_xp, new_level, current_time + cooldown)
|
|
print(f"XP UPDATE --- {message.author.name} leveled up; new_level = {new_level}.")
|
|
await message.channel.send(content=f"<@{user_id}> {level_messages.load_level_message(new_level)}")
|
|
|
|
# assign level role if necessary:
|
|
if new_level in [5, 10, 15, 20, 25, 30, 35, 40, 45, 50]:
|
|
await Xp.assign_level_role(message.author, new_level)
|
|
|
|
# Currency handler
|
|
user_currency = Currency(user_id)
|
|
|
|
# award a special_balance_currency
|
|
amount = 1
|
|
user_currency.add_special(amount)
|
|
user_currency.push()
|
|
|
|
else:
|
|
# user gained XP but not new level
|
|
Xp.update_user_xp(user_id, new_xp, current_level, current_time + cooldown)
|
|
print(f"XP UPDATE --- {message.author.name} gained {xp_gain} XP; new_xp = {new_xp}.")
|
|
|
|
"""
|
|
Level calculation
|
|
Linear = 9x + 27
|
|
"""
|
|
|
|
@commands.slash_command(
|
|
name="level",
|
|
description="Displays your level and rank.",
|
|
guild_only=True
|
|
)
|
|
@commands.check(universal.channel_check)
|
|
async def level(self, ctx):
|
|
xp_data = Xp.get_user_xp_data(ctx.author.id)
|
|
(current_xp, current_level, cooldown) = xp_data[0]
|
|
rank = Xp.calculate_rank(ctx.author.id)
|
|
needed_xp_for_next_level = xp_functions.xp_needed_for_next_level(current_level)
|
|
|
|
await ctx.respond(embed=embeds.level_command_message(ctx, current_level, current_xp,
|
|
needed_xp_for_next_level, rank))
|
|
|
|
@commands.slash_command(
|
|
name="leaderboard",
|
|
description="Are ya winnin' son?",
|
|
guild_only=True
|
|
)
|
|
@commands.check(universal.channel_check)
|
|
async def leaderboard(self, ctx):
|
|
leaderboard = Xp.load_leaderboard()
|
|
|
|
embed = discord.Embed(
|
|
color=0xadcca6
|
|
)
|
|
embed.set_author(name="Rave Cave Leaderboard",
|
|
icon_url="https://cdn.discordapp.com/icons/719227135151046699/"
|
|
"49df8c284382af9dbcfd629c8eadc52c.webp?size=96")
|
|
embed.set_footer(text=f"Do /level to see your rank.")
|
|
embed.set_thumbnail(url="https://i.imgur.com/79XfsbS.png")
|
|
for i, (user_id, xp, level, rank, xp_needed_for_next_level) in enumerate(leaderboard[:5], start=1):
|
|
try:
|
|
member = await ctx.guild.fetch_member(user_id)
|
|
name = member.name
|
|
except:
|
|
name = "Unknown User"
|
|
|
|
embed.add_field(
|
|
name=f"#{rank} - {name}",
|
|
value=f"level: `{level}`\nxp: `{xp}/{xp_needed_for_next_level}`",
|
|
inline=False
|
|
)
|
|
|
|
await ctx.respond(embed=embed)
|
|
|
|
|
|
def setup(sbbot):
|
|
sbbot.add_cog(Leveling(sbbot))
|