Upload bot.py

bot.py

@@ -0,0 +1,241 @@
+import discord
+from discord.ext import commands, tasks
+from discord import app_commands
+from datetime import datetime, timedelta
+from zoneinfo import ZoneInfo
+import json
+import os
+
+TOKEN = "MTQzNzQyMzM5Mzk5MjYwNTc5OQ.G7YuIP.W-tSbjR5iRgkvWYKwq1z2TKKoVmU6c2qCw4mB8"
+
+CHANNEL_NAME = "︙🏖️︙urlopy"
+BACKUP_CHANNEL = "︙📦︙backup-urlopy"
+
+intents = discord.Intents.default()
+intents.members = True
+intents.voice_states = True
+
+bot = commands.Bot(command_prefix="!", intents=intents)
+
+active_leaves = {}
+voice_time = {}
+backup_message_id = None
+
+
+# ================= BACKUP =================
+
+async def save_backup():
+    global backup_message_id
+
+    for guild in bot.guilds:
+        channel = discord.utils.get(guild.text_channels, name=BACKUP_CHANNEL)
+        if not channel:
+            continue
+
+        data = {
+            str(user_id): {
+                "start": str(start),
+                "end": str(end),
+                "nick": nick
+            }
+            for user_id, (start, end, nick) in active_leaves.items()
+        }
+
+        content = json.dumps(data)
+
+        try:
+            if backup_message_id:
+                msg = await channel.fetch_message(backup_message_id)
+                await msg.edit(content=content)
+                return
+        except:
+            backup_message_id = None
+
+        async for msg in channel.history(limit=10):
+            backup_message_id = msg.id
+            await msg.edit(content=content)
+            return
+
+        msg = await channel.send(content)
+        backup_message_id = msg.id
+
+
+async def load_backup():
+    global backup_message_id
+
+    for guild in bot.guilds:
+        channel = discord.utils.get(guild.text_channels, name=BACKUP_CHANNEL)
+        if not channel:
+            continue
+
+        async for msg in channel.history(limit=10):
+            try:
+                data = json.loads(msg.content)
+                backup_message_id = msg.id
+
+                for user_id, val in data.items():
+                    active_leaves[int(user_id)] = (
+                        datetime.strptime(val["start"], "%Y-%m-%d").date(),
+                        datetime.strptime(val["end"], "%Y-%m-%d").date(),
+                        val["nick"]
+                    )
+                return
+            except:
+                continue
+
+
+# ================= READY =================
+
+@bot.event
+async def on_ready():
+    await bot.tree.sync()
+    await load_backup()
+    leave_manager.start()
+    print(f"Zalogowano jako {bot.user}")
+
+
+# ================= KOMENDA =================
+
+@bot.tree.command(name="urlop")
+async def urlop(interaction: discord.Interaction, od: str, do: str, powod: str):
+
+    await interaction.response.defer()  # 🔥 FIX 404 INTERACTION
+
+    try:
+        now = datetime.now(ZoneInfo("Europe/Warsaw"))
+        today = now.date()
+        year = now.year
+
+        member = interaction.user
+
+        def reject(reason):
+            return (
+                f"❌ {member.mention} próbował wziąć urlop!\n\n"
+                f"Od: **{od}**\n"
+                f"Do: **{do}**\n"
+                f"Powód: {powod}\n\n"
+                f"🚫 Odrzucono: {reason}"
+            )
+
+        channel = discord.utils.get(interaction.guild.text_channels, name=CHANNEL_NAME)
+
+        if interaction.channel.name != CHANNEL_NAME:
+            return await interaction.followup.send("❌ Użyj komendy na kanale urlopy!")
+
+        # ⛔ TYLKO PO 22:00 BLOKADA (PRZED 12:00 JUŻ DZIAŁA)
+        if now.hour >= 22:
+            return await interaction.followup.send(reject("Urlopy zablokowane po 22:00"))
+
+        start = datetime.strptime(f"{od}.{year}", "%d.%m.%Y").date()
+        end_real = datetime.strptime(f"{do}.{year}", "%d.%m.%Y").date()
+
+        if start < today:
+            return await interaction.followup.send(reject("Nie możesz brać urlopu w przeszłości"))
+
+        if start > today + timedelta(days=7):
+            return await interaction.followup.send(reject("Max 7 dni do przodu"))
+
+        dni = (end_real - start).days + 1
+
+        if dni < 2:
+            return await interaction.followup.send(reject("Minimum 2 dni"))
+
+        if dni > 7:
+            return await interaction.followup.send(reject("Max 7 dni"))
+
+        if member.id in active_leaves and "[URLOP]" not in member.display_name:
+            del active_leaves[member.id]
+            await save_backup()
+
+        if "[URLOP]" in member.display_name or member.id in active_leaves:
+            return await interaction.followup.send(reject("Masz już aktywny urlop"))
+
+        remove_date = end_real + timedelta(days=1)
+        active_leaves[member.id] = (start, remove_date, False)
+
+        await save_backup()
+
+        await channel.send(
+            f"📅 {member.mention} zgłosił urlop!\n"
+            f"Od: **{od}**\n"
+            f"Do: **{do}**\n"
+            f"Dni: **{dni}**\n"
+            f"Powód: {powod}"
+        )
+
+        await interaction.followup.send("✅ Urlop zapisany!")
+
+    except Exception as e:
+        await interaction.followup.send(f"❌ Błąd dat! Użyj DD.MM")
+
+
+# ================= SYSTEM =================
+
+@tasks.loop(minutes=5)
+async def leave_manager():
+    today = datetime.now(ZoneInfo("Europe/Warsaw")).date()
+
+    for user_id in list(active_leaves.keys()):
+        start, remove_date, nick_dany = active_leaves[user_id]
+
+        for guild in bot.guilds:
+            member = guild.get_member(user_id)
+            if not member:
+                continue
+
+            channel = discord.utils.get(guild.text_channels, name=CHANNEL_NAME)
+
+            if today >= start and not nick_dany:
+                try:
+                    new_nick = f"{member.display_name} [URLOP]"
+                    new_nick = new_nick[:32]
+
+                    await member.edit(nick=new_nick)
+                    await channel.send(f"🟢 {member.mention} rozpoczął urlop!")
+                except:
+                    pass
+
+                active_leaves[user_id] = (start, remove_date, True)
+                await save_backup()
+
+            if today >= remove_date:
+                try:
+                    await member.edit(nick=member.display_name.replace(" [URLOP]", ""))
+                    await channel.send(f"🔴 {member.mention} zakończył urlop!")
+                except:
+                    pass
+
+                del active_leaves[user_id]
+                await save_backup()
+
+
+# ================= VOICE =================
+
+@bot.event
+async def on_voice_state_update(member, before, after):
+
+    if after.channel and member.id in active_leaves:
+        voice_time[member.id] = datetime.now(ZoneInfo("Europe/Warsaw"))
+
+    if before.channel and not after.channel:
+        if member.id in voice_time:
+            czas = datetime.now(ZoneInfo("Europe/Warsaw")) - voice_time[member.id]
+
+            if czas.total_seconds() >= 3600:
+                for guild in bot.guilds:
+                    channel = discord.utils.get(guild.text_channels, name=CHANNEL_NAME)
+
+                    try:
+                        await member.edit(nick=member.display_name.replace(" [URLOP]", ""))
+                        await channel.send(f"⚡ {member.mention} wrócił wcześniej z urlopu!")
+                    except:
+                        pass
+
+                if member.id in active_leaves:
+                    del active_leaves[member.id]
+                    await save_backup()
+
+            del voice_time[member.id]
+
+
+bot.run(TOKEN)