Upload files to "/"

marsey_bot.py

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