lulzbot/lulzbot.py

import datetime
import discord
from src.flan import flan_speak
from src.cartman import cartman_speak
from src.twitter import get_tweet
from fortune import fortune
import numpy as np


motd = '''
 _          _         ____   ___  _
| |   _   _| |    ___| __ ) / _ \| |_
| |  | | | | |   |_  /  _ \| | | | __|
| |__| |_| | |___ / /| |_) | |_| | |_ 
|_____\__,_|_____/___|____/ \___/ \__|
'''

menu = '''```
Commands:
  fortune: tell a fortune
  chuck: give a Chuck Norris quote
  ligma: LIGMA BALLS
  bofa: deez
  deez: nutz
  limerick: tell a limerick
  prost!: prost!

Contribute!
    https://github.com/adoyle0/lulzbot```'''


chuck_quotes = open('data/chuck_quotes').read().split('\n%\n')
ligma_list = open('data/ligma_list').read().split('\n')
limericks = open('data/limericks').read().split('\n%\n')
aclist = open('data/aclist').read().split('\n')


def show_menu():
    return menu


def musk():
    return get_tweet(44196397)


def ligma():
    return np.random.choice(ligma_list)


def limerick():
    return np.random.choice(limericks)


def prost():
    return 'https://tenor.com/view/prost-christoph-waltz-django-bier-zum-wohle-gif-11041516'

def chuck():
    return np.random.choice(chuck_quotes)


def ac():
    return np.random.choice(aclist)


triggers = {'lulzbot': show_menu,  # these need to be functions
            'musk': musk,
            'deez': ligma,
            'ligma': ligma,
            'bofa': ligma,
            'bopha': ligma,
            'limerick': limerick,
            'limrick': limerick,
            'prost!': prost,
            'fortune': fortune,
            'chuck': chuck,
            'ac': ac,
            }

TOKEN = open('.sekrit/discord_token').read()
intents = discord.Intents.default()
intents.message_content = True
client = discord.Client(activity=discord.Game(
    name='with myself'), intents=intents)


@client.event
async def on_ready():
    print(motd+'\n'+datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S') +
          '\nLogged in as {0.user}'.format(client))
    return


@client.event
async def on_message(message):
    username = str(message.author).split('#')[0]
    user_message = str(message.content)
    channel = str(message.channel.name)
    print(f'{datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")} ({channel}) {username}: {user_message}')

    if message.author == client.user:
        return

    elif message.channel.name == 'cartman':
        await message.channel.send(cartman_speak(user_message))
        # await message.channel.send("I'm broken, come back later.")

    elif message.channel.name == 'flan':
        await message.channel.send(flan_speak(user_message))
        # await message.channel.send('GPU is busy, come back later')

    elif message.channel.name == 'shitposting':
        if user_message.lower() in triggers:
            await message.channel.send(triggers[user_message.lower()]())
    return

client.run(TOKEN)