Guzzy
/
marsey_bot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update marsey_bot.py

master
Guzzy 2023-11-15 06:12:38 +00:00
parent 36ed396f78
commit 15993b1693
1 changed files with 90 additions and 90 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

180
marsey_bot.py
View File

@ -1,90 +1,90 @@
#!/usr/bin/env python #!/usr/bin/env python
# encoding: utf-8 # encoding: utf-8
import os import os
import discord import discord
from discord.ext import commands from discord.ext import commands
from discord.ext import tasks from discord.ext import tasks
from PIL import Image, ImageSequence from PIL import Image, ImageSequence
import string import string
def wheres_marsey(message_text): #returns list of potential marsey names def wheres_marsey(message_text): #returns list of potential marsey names
marsey_names = [] marsey_names = []
message_text = message_text.lower().split('!') #convert message to all lowercase and checks for "!" message_text = message_text.lower().split('!') #convert message to all lowercase and checks for "!"
for marsey_name in message_text: for marsey_name in message_text:
marsey_name = marsey_name.translate(str.maketrans('', '', string.punctuation)) #strip punctuation marsey_name = marsey_name.translate(str.maketrans('', '', string.punctuation)) #strip punctuation
marsey_name = marsey_name.split(' ') marsey_name = marsey_name.split(' ')
marsey_names.append(marsey_name[0]) #only look for first word afer "!" marsey_names.append(marsey_name[0]) #only look for first word afer "!"
return [x for x in marsey_names[1:] if x] #ignore empty strings return [x for x in marsey_names[1:] if x] #ignore empty strings
def is_animated(file_path): #check if webp is animated or static def is_animated(file_path): #check if webp is animated or static
try: try:
media = Image.open(file_path) media = Image.open(file_path)
Index = 0 Index = 0
for Frame in ImageSequence.Iterator(media): #check number of frames for Frame in ImageSequence.Iterator(media): #check number of frames
Index += 1 Index += 1
if Index > 1: #if more than one frame return True if Index > 1: #if more than one frame return True
return True return True
else: else:
return False return False
except: except:
return False return False
def webm2gif(abs_media_path): #convert webp to gif def webm2gif(abs_media_path): #convert webp to gif
media = Image.open(abs_media_path) #path of webm file media = Image.open(abs_media_path) #path of webm file
media.info.pop('background', None) media.info.pop('background', None)
abs_media_path = abs_media_path[:-4]+'gif' #change the path from webp to gif abs_media_path = abs_media_path[:-4]+'gif' #change the path from webp to gif
media.save(abs_media_path, 'gif', optimize=False, save_all=True, disposal=2) #temp save a gif file (there might be a better way but i lazy :marseyteehee:) media.save(abs_media_path, 'gif', optimize=False, save_all=True, disposal=2) #temp save a gif file (there might be a better way but i lazy :marseyteehee:)
return(abs_media_path) #return gif path return abs_media_path #return gif path
client = commands.Bot(command_prefix = '!') client = commands.Bot(command_prefix = '!')
@client.event #print ready in command line to verify bot initiation @client.event #print ready in command line to verify bot initiation
async def on_ready(): async def on_ready():
print('ready') print('ready')
@client.event @client.event
async def on_message(message): #bot reads every message in server async def on_message(message): #bot reads every message in server
message_text = message.content message_text = message.content
marsey_names = wheres_marsey(message_text) #make list of called marseys marsey_names = wheres_marsey(message_text) #make list of called marseys
marsey_queue = [] #empty list of marseys to send marsey_queue = [] #empty list of marseys to send
for marsey_name in marsey_names: for marsey_name in marsey_names:
#find directory of marsey #find directory of marsey
bot_dir = os.path.dirname(__file__) bot_dir = os.path.dirname(__file__)
rel_marsey_path = "emojis/"+marsey_name+".webp" rel_marsey_path = "emojis/"+marsey_name+".webp"
abs_marsey_path = os.path.join(bot_dir, rel_marsey_path) abs_marsey_path = os.path.join(bot_dir, rel_marsey_path)
flag = 0 #flag used in webp to gif file conversion flag = 0 #flag used in webp to gif file conversion
if os.path.isfile(abs_marsey_path): #only continue if marsey exists if os.path.isfile(abs_marsey_path): #only continue if marsey exists
#if marsey is animated, turn it into a gif #if marsey is animated, turn it into a gif
if is_animated(abs_marsey_path) is True: if is_animated(abs_marsey_path) is True:
abs_marsey_path = webm2gif(abs_marsey_path) abs_marsey_path = webm2gif(abs_marsey_path)
flag = 1 #flag for deleting temp gif file after sent flag = 1 #flag for deleting temp gif file after sent
#open file and add it to send queue #open file and add it to send queue
with open(abs_marsey_path, 'rb') as f: with open(abs_marsey_path, 'rb') as f:
marsey_queue.append(discord.File(f)) marsey_queue.append(discord.File(f))
#delete temp file #delete temp file
if flag == 1: if flag == 1:
os.remove(abs_marsey_path) os.remove(abs_marsey_path)
flag = 0 #reset flag flag = 0 #reset flag
if marsey_queue: #only send marsey(s) if marsey_queue: #only send marsey(s)
await message.channel.send(files=marsey_queue) await message.channel.send(files=marsey_queue)
#token #token
with open('discord_token.txt') as f: with open('discord_token.txt') as f:
discord_token = f.readline() discord_token = f.readline()
client.run(discord_token) client.run(discord_token)