dankmeme.py

#!/usr/bin/env python3
# Author : upils

import os
import discord
import asyncio
from discord.ext import commands

client = discord.Client()

DEBUG = False

# Load the file containing memes
def loadFile():
  data = {}
  with open('memes.txt') as file:
    for line in file:
      name = line.split('\t')[0]
      url = line.split('\t')[1].strip()
      data[name] = url
  return data

# Get a name by it's name
def getMemeByName(memeName):
  memesDict = loadFile()
  if memeName in memesDict:
    link = memesDict[memeName]
  else:
    link = "no link know for this meme."
  return link

# List known memes
def getMemeList():
  memesDict = loadFile()
  list = ['']
  i = 0
  for k in memesDict:
    if list[i].count('\n') > 41:
      i = i + 1
      list.append('')
    list[i] = list[i] + "{}\t <{}>\n".format(k, memesDict.get(k))
  print("A list have been asked.")
  return list

# Add a meme
def addMeme(name, url):
  with open('memes.txt', 'a') as file:
    file.write('{}\t{}\n'.format(name, url))

# Delete a meme
def delMeme(name):
  with open('memes.txt', 'r') as file:
    lines = file.readlines()
  with open('memes.txt', 'w') as file:
    for line in lines:
        if line.split('\t')[0] != name:
          file.write(line)


@client.event
async def on_ready():
  print('Logged in as')
  print(client.user.name)
  print(client.user.id)
  for channel in client.get_all_channels():
    if DEBUG == True:
      if "text" in str(channel.type) and channel.name == "danktest":
        await client.send_message(channel, "Hey everyone !\n\nI'm here to send memes.\n\
Type !help to get the list of currently available commands.\n\
Version 0.2")

@client.event
async def on_message(message):
  if message.content.startswith('! '):

    tmp = await client.send_message(message.channel, 'Meme INCOMING...')
    try:
      memeName = message.content[len('! '):].strip()
      meme = getMemeByName(memeName)
      await client.edit_message(tmp, meme)
    except IndexError:
      await client.edit_message(tmp,'You have to give the name of the meme you want !')

  elif message.content.startswith('!list'):

    tmp = await client.send_message(message.author, 'List INCOMING...')
    liste = getMemeList()
    for chunk in liste:
     await client.send_message(message.author, chunk)

  elif message.content.startswith('!add '):

    tmp = await client.send_message(message.channel, 'Adding a meme...')
    try:
      memeName = message.content[len('!add '):].split()[0]
      memeURL = message.content[len('!add '):].split()[1]
      addMeme(memeName, memeURL)
      await client.edit_message(tmp,'Meme has been added !')
    except IndexError:
      await client.edit_message(tmp,'You have to give a name AND a link to your meme !')

  elif message.content.startswith('!del '):

    tmp = await client.send_message(message.channel, 'Deleting a meme...')
    try:
      memeName = message.content[len('!del '):].split()[0]
      delMeme(memeName)
      await client.edit_message(tmp,'The meme has been deleted !')
    except IndexError:
      await client.edit_message(tmp,'You have to give the name of the meme you want to delete !')

  elif message.content.startswith('!help'):

    await client.send_message(message.channel, "Commands: \n \
! <meme> to send a meme on the channel \n \
!list to list the available memes \n \
!add <name> <url> to add a new meme \n \
!del <name> to delete a meme \n")

#Load API key
APIKEY = os.environ['DISCORD_KEY']


#Run the bot
restart = 5

while restart>0:
  try:
    client.run(APIKEY)
  except ConnectionReset:
    restart = 5
  except:
    restart-=1