déplacement de citation vers son cogs
This commit is contained in:
parent
ca845fe704
commit
664fdfcdb6
2 changed files with 97 additions and 81 deletions
96
src/cogs/citation.py
Normal file
96
src/cogs/citation.py
Normal file
|
|
@ -0,0 +1,96 @@
|
||||||
|
import discord
|
||||||
|
import re
|
||||||
|
import os
|
||||||
|
from discord.ext import commands
|
||||||
|
customTimezone = os.environ['TIMEZONE']
|
||||||
|
from utils.core import goodTimezone, userOrNick
|
||||||
|
|
||||||
|
def setup(client):
|
||||||
|
client.add_cog(Citation(client))
|
||||||
|
|
||||||
|
class Citation(commands.Cog):
|
||||||
|
"""Gère le système de citation."""
|
||||||
|
def __init__(self, client):
|
||||||
|
self.client = client
|
||||||
|
|
||||||
|
@commands.Cog.listener()
|
||||||
|
async def on_message(self, message):
|
||||||
|
urls = re.findall(r'http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+', message.content)
|
||||||
|
httpsString = "https://"
|
||||||
|
channelsString = "discord.com/channels/"
|
||||||
|
for i in range(len(urls)):
|
||||||
|
if urls[i].startswith(f"{httpsString}{channelsString}") or urls[i].startswith(f"{httpsString}ptb.{channelsString}") or urls[i].startswith(f"{httpsString}canary.{channelsString}"):
|
||||||
|
link = urls[i]
|
||||||
|
linkURL = link
|
||||||
|
if link.startswith(f"{httpsString}{channelsString}"):
|
||||||
|
link = f'000{link}'
|
||||||
|
if link.startswith(f"{httpsString}ptb.{channelsString}"):
|
||||||
|
link = link[1:]
|
||||||
|
if link.startswith(f"{httpsString}canary.{channelsString}"):
|
||||||
|
link = link[4:]
|
||||||
|
if "@me" in urls[i]:
|
||||||
|
return await message.channel.send("Je ne cite pas les messages privés.", delete_after = 5)
|
||||||
|
try:
|
||||||
|
if int(link[32:-38]) == message.guild.id:
|
||||||
|
msgID = await self.client.get_channel(int(link[51:-19])).fetch_message(int(link[70:]))
|
||||||
|
couleur = 0x2f3136
|
||||||
|
msgFiles = msgID.attachments
|
||||||
|
imageExtensions = ["jpg", "jpeg", "png", "webp", "gif"]
|
||||||
|
desc = msgID.content
|
||||||
|
if len(msgFiles) > 1:
|
||||||
|
listOfFiles = ""
|
||||||
|
for i in range(0, len(msgFiles)):
|
||||||
|
listOfFiles = f"{listOfFiles}, {msgFiles[i].filename}"
|
||||||
|
listOfFiles = listOfFiles[2:]
|
||||||
|
if len(msgID.content) > 0:
|
||||||
|
desc = f"{msgID.content}\n\nIl y a plusieurs fichiers dans ce message : {listOfFiles}"
|
||||||
|
else:
|
||||||
|
desc = f"Il y a plusieurs fichiers dans ce message : {listOfFiles}"
|
||||||
|
else:
|
||||||
|
if len(msgFiles) == 1:
|
||||||
|
if msgFiles[0].filename[-4:].split('.')[1] in imageExtensions:
|
||||||
|
if not len(msgID.content) > 0:
|
||||||
|
desc = f"Une image jointe : {msgFiles[0].filename}"
|
||||||
|
else:
|
||||||
|
linkFile = msgFiles[0].url
|
||||||
|
if not len(msgID.content) > 0:
|
||||||
|
desc = f"Un fichier joint : {msgFiles[0].filename}"
|
||||||
|
embed = discord.Embed(description = desc, colour = couleur)
|
||||||
|
auteur = "Auteur"
|
||||||
|
if message.author == msgID.author:
|
||||||
|
auteur = "Auteur & Citateur"
|
||||||
|
embed.add_field(name = auteur, value = msgID.author.mention, inline=True)
|
||||||
|
try:
|
||||||
|
if len(msgFiles) == 1:
|
||||||
|
if msgFiles[0].filename[-4:].split('.')[1] in imageExtensions:
|
||||||
|
embed.set_image(url=msgFiles[0].url)
|
||||||
|
else:
|
||||||
|
embed.add_field(name = "Fichier", value = f"[Lien]({linkFile})", inline=True)
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
embed.add_field(name = "Message", value = f"{msgID.channel.mention} - [Lien Message]({linkURL})", inline=True)
|
||||||
|
embed.set_author(name = "Citation", icon_url = msgID.author.avatar_url)
|
||||||
|
icon_url = message.author.avatar_url
|
||||||
|
|
||||||
|
date_1 = goodTimezone(msgID.created_at, customTimezone)
|
||||||
|
edit = ""
|
||||||
|
if msgID.edited_at:
|
||||||
|
date_edit = goodTimezone(msgID.edited_at, customTimezone)
|
||||||
|
edit = f" et modifié le {date_edit[0][8:]}/{date_edit[0][5:-3]}/{date_edit[0][:4]} à {date_edit[1]}"
|
||||||
|
messageDuBas = f"Posté le {date_1[0][8:]}/{date_1[0][5:-3]}/{date_1[0][:4]} à {date_1[1]}{edit}"
|
||||||
|
|
||||||
|
date_2 = goodTimezone(message.created_at, customTimezone)
|
||||||
|
date_2 = f"{date_2[0][8:]}/{date_2[0][5:-3]}/{date_2[0][:4]} à {date_2[1]}"
|
||||||
|
|
||||||
|
if auteur == "Auteur":
|
||||||
|
messageDuBas = messageDuBas + f"\nCité par {userOrNick(message.author)} le {date_2}"
|
||||||
|
embed.set_footer(icon_url = icon_url, text = messageDuBas)
|
||||||
|
if message.content == linkURL.replace(' ',''):
|
||||||
|
await message.channel.send(embed = embed)
|
||||||
|
await message.delete()
|
||||||
|
else:
|
||||||
|
await message.reply(embed = embed, mention_author = False)
|
||||||
|
except Exception as e:
|
||||||
|
e = str(e)
|
||||||
|
if not "invalid literal for int() with base 10:" in e or not "404 Not Found (error code: 10008)" in e: # faute de frappe / message supprimé
|
||||||
|
print(e)
|
||||||
82
src/main.py
82
src/main.py
|
|
@ -20,6 +20,7 @@ client.load_extension("cogs.music")
|
||||||
client.load_extension("cogs.games")
|
client.load_extension("cogs.games")
|
||||||
client.load_extension("cogs.fun")
|
client.load_extension("cogs.fun")
|
||||||
client.load_extension("cogs.school")
|
client.load_extension("cogs.school")
|
||||||
|
client.load_extension("cogs.citation")
|
||||||
client.load_extension("cogs.confreriedukassoulait") # you can remove this cogs, only for my private guild
|
client.load_extension("cogs.confreriedukassoulait") # you can remove this cogs, only for my private guild
|
||||||
Reminder().creationTable()
|
Reminder().creationTable()
|
||||||
print("Terminé !")
|
print("Terminé !")
|
||||||
|
|
@ -51,87 +52,6 @@ async def on_message(message):
|
||||||
ctx = await client.get_context(message)
|
ctx = await client.get_context(message)
|
||||||
prefix = await client.get_prefix(message)
|
prefix = await client.get_prefix(message)
|
||||||
await ctx.send(f">>> Coucou !\nMon préfix est `{prefix}` et ma commande d'aide est `{prefix}help`")
|
await ctx.send(f">>> Coucou !\nMon préfix est `{prefix}` et ma commande d'aide est `{prefix}help`")
|
||||||
|
|
||||||
"""citation"""
|
|
||||||
urls = re.findall(r'http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+', message.content)
|
|
||||||
httpsString = "https://"
|
|
||||||
channelsString = "discord.com/channels/"
|
|
||||||
for i in range(len(urls)):
|
|
||||||
if urls[i].startswith(f"{httpsString}{channelsString}") or urls[i].startswith(f"{httpsString}ptb.{channelsString}") or urls[i].startswith(f"{httpsString}canary.{channelsString}"):
|
|
||||||
link = urls[i]
|
|
||||||
linkURL = link
|
|
||||||
if link.startswith(f"{httpsString}{channelsString}"):
|
|
||||||
link = f'000{link}'
|
|
||||||
if link.startswith(f"{httpsString}ptb.{channelsString}"):
|
|
||||||
link = link[1:]
|
|
||||||
if link.startswith(f"{httpsString}canary.{channelsString}"):
|
|
||||||
link = link[4:]
|
|
||||||
if "@me" in urls[i]:
|
|
||||||
return await message.channel.send("Je ne cite pas les messages privés.", delete_after = 5)
|
|
||||||
try:
|
|
||||||
if int(link[32:-38]) == message.guild.id:
|
|
||||||
msgID = await client.get_channel(int(link[51:-19])).fetch_message(int(link[70:]))
|
|
||||||
couleur = 0x2f3136
|
|
||||||
msgFiles = msgID.attachments
|
|
||||||
imageExtensions = ["jpg", "jpeg", "png", "webp", "gif"]
|
|
||||||
desc = msgID.content
|
|
||||||
if len(msgFiles) > 1:
|
|
||||||
listOfFiles = ""
|
|
||||||
for i in range(0, len(msgFiles)):
|
|
||||||
listOfFiles = f"{listOfFiles}, {msgFiles[i].filename}"
|
|
||||||
listOfFiles = listOfFiles[2:]
|
|
||||||
if len(msgID.content) > 0:
|
|
||||||
desc = f"{msgID.content}\n\nIl y a plusieurs fichiers dans ce message : {listOfFiles}"
|
|
||||||
else:
|
|
||||||
desc = f"Il y a plusieurs fichiers dans ce message : {listOfFiles}"
|
|
||||||
else:
|
|
||||||
if len(msgFiles) == 1:
|
|
||||||
if msgFiles[0].filename[-4:].split('.')[1] in imageExtensions:
|
|
||||||
if not len(msgID.content) > 0:
|
|
||||||
desc = f"Une image jointe : {msgFiles[0].filename}"
|
|
||||||
else:
|
|
||||||
linkFile = msgFiles[0].url
|
|
||||||
if not len(msgID.content) > 0:
|
|
||||||
desc = f"Un fichier joint : {msgFiles[0].filename}"
|
|
||||||
embed = discord.Embed(description = desc, colour = couleur)
|
|
||||||
auteur = "Auteur"
|
|
||||||
if message.author == msgID.author:
|
|
||||||
auteur = "Auteur & Citateur"
|
|
||||||
embed.add_field(name = auteur, value = msgID.author.mention, inline=True)
|
|
||||||
try:
|
|
||||||
if len(msgFiles) == 1:
|
|
||||||
if msgFiles[0].filename[-4:].split('.')[1] in imageExtensions:
|
|
||||||
embed.set_image(url=msgFiles[0].url)
|
|
||||||
else:
|
|
||||||
embed.add_field(name = "Fichier", value = f"[Lien]({linkFile})", inline=True)
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
embed.add_field(name = "Message", value = f"{msgID.channel.mention} - [Lien Message]({linkURL})", inline=True)
|
|
||||||
embed.set_author(name = "Citation", icon_url = msgID.author.avatar_url)
|
|
||||||
icon_url = message.author.avatar_url
|
|
||||||
|
|
||||||
date_1 = goodTimezone(msgID.created_at, customTimezone)
|
|
||||||
edit = ""
|
|
||||||
if msgID.edited_at:
|
|
||||||
date_edit = goodTimezone(msgID.edited_at, customTimezone)
|
|
||||||
edit = f" et modifié le {date_edit[0][8:]}/{date_edit[0][5:-3]}/{date_edit[0][:4]} à {date_edit[1]}"
|
|
||||||
messageDuBas = f"Posté le {date_1[0][8:]}/{date_1[0][5:-3]}/{date_1[0][:4]} à {date_1[1]}{edit}"
|
|
||||||
|
|
||||||
date_2 = goodTimezone(message.created_at, customTimezone)
|
|
||||||
date_2 = f"{date_2[0][8:]}/{date_2[0][5:-3]}/{date_2[0][:4]} à {date_2[1]}"
|
|
||||||
|
|
||||||
if auteur == "Auteur":
|
|
||||||
messageDuBas = messageDuBas + f"\nCité par {userOrNick(message.author)} le {date_2}"
|
|
||||||
embed.set_footer(icon_url = icon_url, text = messageDuBas)
|
|
||||||
if message.content == linkURL.replace(' ',''):
|
|
||||||
await message.channel.send(embed = embed)
|
|
||||||
await message.delete()
|
|
||||||
else:
|
|
||||||
await message.reply(embed = embed, mention_author = False)
|
|
||||||
except Exception as e:
|
|
||||||
e = str(e)
|
|
||||||
if not "invalid literal for int() with base 10:" in e or not "404 Not Found (error code: 10008)" in e: # faute de frappe / message supprimé
|
|
||||||
print(e)
|
|
||||||
|
|
||||||
print("Connexion à Discord...", end = " ")
|
print("Connexion à Discord...", end = " ")
|
||||||
client.run(os.environ['TOKEN_DISCORD'])
|
client.run(os.environ['TOKEN_DISCORD'])
|
||||||
|
|
|
||||||
Reference in a new issue