Add handlers for exp and mee6

1 files changed, 54 insertions, 0 deletions

diff --git a/exp.py b/exp.py
new file mode 100644
index 0000000..7583ecd
--- /dev/null
+++ b/exp.py
@@ -0,0 +1,54 @@
+import datetime
+import random
+from base import config
+from models.level import Level
+from models.user import User
+from discord import TextChannel
+
+
+class Exp:
+    def __init__(self, user: User, bot_channel: TextChannel):
+        self.user = user
+        self.channel = bot_channel
+
+    async def calculate_exp(self):
+        self.add_message(self.user)
+        gained_exp = self.gain_exp(self.user)
+        if gained_exp:
+            if self.check_level_up(self.user):
+                await self.level_up(self.user, self.channel)
+
+    def add_message(self, user: User):
+        messages = user.message_count
+        user.message_count = messages + 1
+        user.save()
+
+    def gain_exp(self, user: User):
+        time_now = datetime.datetime.utcnow()
+        minute_ago = time_now - datetime.timedelta(seconds=60)
+        print(f"now: {time_now}, minute: {minute_ago}, user: {user.exp_gained_time}")
+        if user.exp_gained_time is None or user.exp_gained_time < minute_ago:
+            gained_exp = random.randrange(15, 25)
+            total_exp = user.exp + gained_exp
+            user.exp = total_exp
+            user.exp_gained_time = time_now
+            user.save()
+            return True
+        else:
+            return False
+
+    def check_level_up(self, user: User):
+        next_level = Level.get(user.level + 1)
+        if user.exp > next_level.exp_required:
+            return True
+        else:
+            return False
+
+    async def level_up(self, user: User, channel: TextChannel):
+        user.level += 1
+        level = Level.get(level=user.level)
+        user.currency += level.reward_currency
+        await channel.send(
+            f"Congrats {user.display_name}, you levelled up to {user.level}! You win {level.reward_currency} {config['currency']}!"
+        )
+        user.save()