Initial commit: bot Discord Vicy V1.8

commandes/retirer_argent.py

@@ -0,0 +1,147 @@
+import discord
+from discord import app_commands
+from discord.ext import commands
+import json
+from datetime import datetime
+
+BANQUIER_ROLE_ID = 1360314740567576616  # ID du rôle des banquiers  # ID du rôle banquier  # ID du rôle banquier
+
+class RetirerArgent(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    banque_choices = [
+        app_commands.Choice(name="Victoria's Credit Bank", value="Victoria's Credit Bank"),
+        app_commands.Choice(name="NGC Bank", value="NGC Bank"),
+    ]
+
+    @app_commands.command(
+        name="retirer_argent",
+        description="Retire de l'argent d'un compte via son RIB (utilisateur ou entreprise)."
+    )
+    @app_commands.choices(banque=banque_choices)
+    @app_commands.describe(
+        rib="Le RIB du compte",
+        montant="Montant à retirer",
+        banque="Banque cible (facultatif)"
+    )
+    async def retirer_argent(
+        self,
+        interaction: discord.Interaction,
+        rib: str,
+        montant: float,
+        banque: app_commands.Choice[str] = None
+    ):
+        if not any(role.id == BANQUIER_ROLE_ID for role in interaction.user.roles):
+            await interaction.response.send_message("❌ Vous n'avez pas la permission de faire cela.", ephemeral=True)
+            return
+
+        if montant <= 0:
+            await interaction.response.send_message("❌ Le montant doit être supérieur à zéro.", ephemeral=True)
+            return
+
+        banque_nom = banque.value if banque else None
+
+        # Charger les fichiers
+        try:
+            with open("comptes_bancaires.json", "r", encoding="utf-8") as f:
+                comptes_bancaires = json.load(f)
+            with open("comptes_entreprises.json", "r", encoding="utf-8") as f:
+                comptes_entreprises = json.load(f)
+        except Exception:
+            await interaction.response.send_message("❌ Erreur lors de la lecture des fichiers JSON.", ephemeral=True)
+            return
+
+        compte = None
+        source = None  # "utilisateur" ou "entreprise"
+        user_id = None
+
+        # Recherche dans la banque spécifiée (si précisée)
+        if banque_nom and banque_nom in comptes_bancaires:
+            for uid, data in comptes_bancaires[banque_nom].items():
+                if data.get("rib") == rib:
+                    compte = data
+                    source = "utilisateur"
+                    user_id = uid
+                    break
+
+        if not compte and banque_nom and banque_nom in comptes_entreprises:
+            for entreprise, data in comptes_entreprises[banque_nom].items():
+                if data.get("rib") == rib:
+                    compte = data
+                    source = "entreprise"
+                    user_id = entreprise
+                    break
+
+        # Si banque non précisée ou compte non trouvé, recherche dans toutes les banques
+        if not compte and not banque_nom:
+            # Recherche dans toutes les banques utilisateurs
+            for banque_courante, comptes_banque in comptes_bancaires.items():
+                for uid, data in comptes_banque.items():
+                    if data.get("rib") == rib:
+                        compte = data
+                        source = "utilisateur"
+                        user_id = uid
+                        banque_nom = banque_courante
+                        break
+                if compte:
+                    break
+
+        if not compte:
+            # Recherche dans toutes les banques entreprises
+            for banque_courante, entreprises_banque in comptes_entreprises.items():
+                for entreprise, data in entreprises_banque.items():
+                    if data.get("rib") == rib:
+                        compte = data
+                        source = "entreprise"
+                        user_id = entreprise
+                        banque_nom = banque_courante
+                        break
+                if compte:
+                    break
+
+        if not compte:
+            await interaction.response.send_message("❌ Aucun compte trouvé avec ce RIB.", ephemeral=True)
+            return
+
+        if compte["solde"] < montant:
+            await interaction.response.send_message("❌ Solde insuffisant pour retirer cette somme.", ephemeral=True)
+            return
+
+        # Appliquer le retrait
+        compte["solde"] -= montant
+        transaction = {
+            "date": datetime.now().strftime("%d/%m/%Y %H:%M:%S"),
+            "type": "Retrait",
+            "source": "Banque",
+            "montant": -montant,
+            "emetteur": interaction.user.id,
+            "destinataire": user_id
+        }
+        compte.setdefault("transactions", []).append(transaction)
+
+        # Sauvegarde dans le bon fichier
+        if source == "utilisateur":
+            comptes_bancaires[banque_nom][user_id] = compte
+            with open("comptes_bancaires.json", "w", encoding="utf-8") as f:
+                json.dump(comptes_bancaires, f, indent=4, ensure_ascii=False)
+        else:
+            comptes_entreprises[banque_nom][user_id] = compte
+            with open("comptes_entreprises.json", "w", encoding="utf-8") as f:
+                json.dump(comptes_entreprises, f, indent=4, ensure_ascii=False)
+
+        # Message de confirmation
+        embed = discord.Embed(
+            title="💸 Retrait effectué",
+            description=f"💼 Banque : **{banque_nom}**\n📌 Type : **{source.capitalize()}**",
+            color=discord.Color.red()
+        )
+        embed.add_field(name="RIB", value=rib, inline=False)
+        embed.add_field(name="Montant retiré", value=f"{montant:.2f}€", inline=True)
+        embed.add_field(name="Effectué par", value=interaction.user.mention, inline=True)
+        embed.set_footer(text="Transaction enregistrée ✅")
+
+        await interaction.response.send_message(embed=embed, ephemeral=False)
+
+async def setup(bot):
+    await bot.add_cog(RetirerArgent(bot))