initial commit

main.py

@@ -0,0 +1,94 @@
+import os
+import discord
+from discord.ext import commands, tasks
+from dotenv import load_dotenv
+from datetime import datetime, timedelta
+import asyncio
+from typing import Dict, Set
+
+# Load environment variables
+load_dotenv()
+
+# Bot configuration
+intents = discord.Intents.default()
+intents.message_content = True
+intents.members = True
+intents.voice_states = True
+
+bot = commands.Bot(command_prefix='!', intents=intents)
+
+# Store user cooldowns and experience points
+user_cooldowns: Dict[int, datetime] = {}
+user_exp_points: Dict[int, int] = {}
+voice_time: Dict[int, datetime] = {}
+active_voice_users: Set[int] = set()
+
+# Constants
+CHAT_EXP_POINTS = 1
+VOICE_EXP_POINTS_PER_MINUTE = 2
+CHAT_COOLDOWN = 60  # seconds
+
+@bot.event
+async def on_ready():
+    print(f'{bot.user} has connected to Discord!')
+    update_voice_exp.start()
+
+@bot.event
+async def on_message(message):
+    if message.author.bot:
+        return
+
+    # Handle chat experience points
+    user_id = message.author.id
+    current_time = datetime.now()
+    
+    if user_id not in user_cooldowns or (current_time - user_cooldowns[user_id]).total_seconds() >= CHAT_COOLDOWN:
+        user_exp_points[user_id] = user_exp_points.get(user_id, 0) + CHAT_EXP_POINTS
+        user_cooldowns[user_id] = current_time
+        print(f"Added {CHAT_EXP_POINTS} exp points to {message.author.name}. Total: {user_exp_points[user_id]}")
+
+    await bot.process_commands(message)
+
+@bot.event
+async def on_voice_state_update(member, before, after):
+    if member.bot:
+        return
+
+    # User joined a voice channel
+    if before.channel is None and after.channel is not None:
+        voice_time[member.id] = datetime.now()
+        active_voice_users.add(member.id)
+        print(f"{member.name} joined voice channel {after.channel.name}")
+
+    # User left a voice channel
+    elif before.channel is not None and after.channel is None:
+        if member.id in active_voice_users:
+            active_voice_users.remove(member.id)
+            if member.id in voice_time:
+                time_spent = datetime.now() - voice_time[member.id]
+                minutes = int(time_spent.total_seconds() / 60)
+                exp_points = minutes * VOICE_EXP_POINTS_PER_MINUTE
+                user_exp_points[member.id] = user_exp_points.get(member.id, 0) + exp_points
+                print(f"{member.name} spent {minutes} minutes in voice. Added {exp_points} exp points. Total: {user_exp_points[member.id]}")
+                del voice_time[member.id]
+
+@tasks.loop(minutes=1)
+async def update_voice_exp():
+    current_time = datetime.now()
+    for user_id in list(active_voice_users):
+        if user_id in voice_time:
+            time_spent = current_time - voice_time[user_id]
+            if time_spent.total_seconds() >= 60:
+                user_exp_points[user_id] = user_exp_points.get(user_id, 0) + VOICE_EXP_POINTS_PER_MINUTE
+                voice_time[user_id] = current_time
+                print(f"Added {VOICE_EXP_POINTS_PER_MINUTE} exp points to user {user_id} for voice time")
+
+@bot.command(name='exp')
+async def show_exp(ctx):
+    user_id = ctx.author.id
+    exp = user_exp_points.get(user_id, 0)
+    await ctx.send(f"{ctx.author.mention} has {exp} experience points!")
+
+print(f"Token: {repr(os.getenv('BOT_TOKEN'))}")
+# Run the bot
+bot.run(os.getenv('BOT_TOKEN'))