mirror of
https://github.com/LuisMayo/ace-attorney-discord-bot.git
synced 2025-06-18 06:15:37 -04:00
c066daeb36
Linked images (like to imgur or even discord uploads) don't show up in attachments, so this will pick them up. Uses proxy_url.
54 lines
2.1 KiB
Python
54 lines
2.1 KiB
Python
from discord import Message
|
||
import re
|
||
from emoji.core import demojize
|
||
import requests
|
||
from objection_engine.beans.comment import Comment
|
||
|
||
class Message:
|
||
def __init__(self, update: Message):
|
||
try:
|
||
self.user = User(update.guild.get_member(update.author.id))
|
||
except Exception as e:
|
||
self.user = User(update.author)
|
||
print(e)
|
||
self.evidence = None
|
||
tmp = update.clean_content
|
||
tmp = re.sub(r'(https?)\S*', '(link)', tmp) # links
|
||
tmp = demojize(tmp)
|
||
tmp = re.sub(r'<[a]?:\w{2,32}:\d{18}>', '', tmp) # custom static and animated emojis
|
||
tmp = re.sub(r':\w{2,32}:', '', tmp) # stock emojis
|
||
tmp = re.sub(r'', '', tmp) # @everyone, @here
|
||
for file in update.attachments: # attachments
|
||
if file.filename.split('.')[-1] in {'jpg', 'jpeg', 'JPG', 'JPEG', 'png', 'PNG'}:
|
||
tmp += ' (image)'
|
||
name = str(file.id) + '.png'
|
||
response = requests.get(file.url)
|
||
with open(name, 'wb') as file:
|
||
file.write(response.content)
|
||
self.evidence = name
|
||
elif file.filename.split('.')[-1] in {'gif', 'gifv'}:
|
||
tmp += ' (gif)'
|
||
elif file.filename.split('.')[-1] in {'mp4', 'webm'}:
|
||
tmp += ' (video)'
|
||
elif file.filename.split('.')[-1] in {'mp3', 'wav', 'ogg'}:
|
||
tmp += ' (audio)'
|
||
else:
|
||
tmp += ' (file)'
|
||
for embed in update.embeds:
|
||
if embed.type == 'image':
|
||
tmp += ' (image)'
|
||
url = embed.thumbnail.proxy_url
|
||
name = url.split('/')[-1]
|
||
response = requests.get(url)
|
||
with open(name, 'wb') as file:
|
||
file.write(response.content)
|
||
self.evidence = name
|
||
self.text = tmp
|
||
def to_Comment(self):
|
||
return Comment(user_id=self.user.id, user_name=self.user.name, text_content=self.text, evidence_path=self.evidence)
|
||
|
||
class User:
|
||
def __init__(self, user):
|
||
self.name = user.display_name
|
||
self.id = user.id
|