Scribd
Upload
11 views3 pages

Message

Uploaded by

justzetsu666
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as TXT, PDF, TXT or read online on Scribd
11 views3 pages

Message

Uploaded by

justzetsu666
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as TXT, PDF, TXT or read online on Scribd
You are on page 1/ 3

import discord

from discord.ext import commands


import os
import asyncio
from concurrent.futures import ThreadPoolExecutor
import aiofiles
import json

with open('config.json') as f:
config = json.load(f)

TOKEN = config["token"]
rmk1337 = commands.Bot(command_prefix="+", intents=discord.Intents.all())

DUMP_FOLDER = 'dbip'
CANAL_ID = int(config["allowed_channel_id"])
ROLE_ID = int(config["required_role_id"])
EMBED_COLOR = int(config["embed_color"], 16)
ENCODINGS = ['utf-8', 'latin-1', 'cp1252']

MAX_CONCURRENT_TASKS = 20
BLOCK_SIZE = 1024 * 1024 # 1 Mo
FILE_PROCESSING_TIMEOUT = 5
executor = ThreadPoolExecutor(max_workers=MAX_CONCURRENT_TASKS)

async def read_file_in_chunks(file_path, license, found_event):


for encoding in ENCODINGS:
try:
async with aiofiles.open(file_path, mode='r', encoding=encoding,
errors='ignore') as f:
while True:
content = await f.read(BLOCK_SIZE)
if not content:
break
if license in content:
return content
except Exception as e:
print(f"Erreur avec l'encodage {encoding} pour le fichier {file_path}:
{e}")
return None

async def process_file(file_path, license, results, found_event):


if found_event.is_set():
return

try:
content = await asyncio.wait_for(read_file_in_chunks(file_path, license,
found_event), timeout=FILE_PROCESSING_TIMEOUT)
if content:
associated_info = next((line.strip() for line in content.split('\n') if
license in line), None)
if associated_info:
results.append(associated_info)
found_event.set()
except asyncio.TimeoutError:
print(f"Le traitement du fichier {file_path} a dépassé le temps limite.")
except Exception as e:
print(f"Erreur lors du traitement du fichier {file_path}: {e}")
@rmk1337.tree.command(name="vip", description="Rechercher des informations pour une
licence FiveM")
async def vip(interaction: discord.Interaction, license: str):
if not any(role.id == ROLE_ID for role in interaction.user.roles):
await interaction.response.send_message("Vous n'avez pas le bon rôle pour
utiliser cette commande.", ephemeral=True)
return

if interaction.channel.id != CANAL_ID:
await interaction.response.send_message("Veuillez faire la commande dans le
salon VIP.", ephemeral=True)
return

embed = discord.Embed(title="VIP Recherche", description="Recherche en


cours...", color=EMBED_COLOR)
await interaction.response.send_message(embed=embed)

found_event = asyncio.Event()
results = []

semaphore = asyncio.Semaphore(MAX_CONCURRENT_TASKS)

async def sem_task(file_path):


async with semaphore:
await process_file(file_path, license, results, found_event)

files_to_process = [os.path.join(root, file)


for root, _, files in os.walk(DUMP_FOLDER)
for file in files if file.endswith(('.txt', '.sql'))]

tasks = [sem_task(file_path) for file_path in files_to_process]


await asyncio.gather(*tasks, return_exceptions=True)

if results:
results_message = results[0]
embed = discord.Embed(
title="Résultat de la Recherche",
description=results_message,
color=EMBED_COLOR
)
embed.set_footer(text=config["footer_text"])

try:
await interaction.user.send(embed=embed)
await interaction.followup.send("Les résultats ont été envoyés en
message privé.")
except discord.Forbidden:
await interaction.followup.send("Je ne peux pas vous envoyer de message
privé. Veuillez vérifier vos paramètres de confidentialité.")
except discord.HTTPException as e:
print(f"Erreur lors de l'envoi du message: {e}")
await interaction.followup.send("Erreur lors de l'envoi du message
privé. Veuillez réessayer plus tard.")
else:
embed = discord.Embed(
title="Aucune Information Trouvée",
description="Aucune information trouvée pour cette licence.",
color=EMBED_COLOR
)
embed.set_footer(text=config["footer_text"])
await interaction.followup.send(embed=embed)

@rmk1337.event
async def on_ready():
print(f"{rmk1337.user} est connecté et prêt à l'emploi. 🚀")
print("🌟 Bot développé par rmk1337 pour LookData 💻✨")

await rmk1337.change_presence(activity=discord.Game(name=config["bot_status"]))
await rmk1337.tree.sync()

rmk1337.run(TOKEN)

You might also like