computer_cell_bot.py

import discord
import random


from io import BytesIO, StringIO
import base64
from numpy import *
import numpy as np
import matplotlib.pyplot as plt


from discord.ext import commands

# last one sitting universe
#TODO: Add channel ID HERE
channels = [0000000]

VC = ""

bot = commands.Bot(command_prefix=">")

from corpora import Corpora

Corpora(bot)


"""
------------------------------
Members
------------------------------
"""


@bot.command("docs")
async def hello(ctx):
 with open("help.txt") as afile:
 docs = afile.read()
 await ctx.send(docs)


"""
------------------------------
JOKES
------------------------------
"""


@bot.group()
async def joke(ctx):
 if ctx.invoked_subcommand is None:
 await ctx.send("Invalid command passed...")


@joke.command()
async def comp(ctx):
 with open("computer_jokes.txt") as afile:
 roasts = afile.readlines()
 roast = random.choice(roasts)
 await ctx.send(roast)


@joke.command()
async def git(ctx):
 with open("git_jokes.txt") as afile:
 roasts = afile.readlines()
 roast = random.choice(roasts)
 await ctx.send(roast)


"""
------------------------------
meme name
------------------------------
"""


@bot.group()
async def make(ctx):
 pass


@make.command()
async def doge(ctx, text_1, text_2):
 text_1 = urllib.parse.quote(text_1.replace(" ", "_"))
 text_2 = urllib.parse.quote(text_2.replace(" ", "_"))
 link = "https://memegen.link/doge/{}/{}.jpg".format(text_1, text_2)
 await ctx.send(link)


@make.command()
async def buzz(ctx, text_1, text_2):
 text_1 = urllib.parse.quote(text_1.replace(" ", "_"))
 text_2 = urllib.parse.quote(text_2.replace(" ", "_"))
 link = "https://memegen.link/buzz/{}/{}.jpg".format(text_1, text_2)
 await ctx.send(link)


"""
------------------------------
mock name <>
------------------------------
"""

import urllib


@bot.group()
async def mock(ctx):
 if ctx.invoked_subcommand is None:
 await ctx.send("Invalid command passed...")


@mock.command()
async def name(ctx, name: str):
 with open("name_mock.txt") as afile:
 name_mocks = afile.readlines()
 for line in name_mocks:
 if line.startswith(name.upper()):
 await ctx.send("{}".format(line))
 return


"""
------------------------------
DANGER: Math commands
------------------------------
"""
import sympy


@bot.group()
async def math(ctx):
 if ctx.invoked_subcommand is None:
 await ctx.send("Invalid command passed...")


@math.command()
async def diff(ctx, equation: str, variable: str):
 try:
 ans = sympy.diff(equation, variable)
 except:
 ans = "Erro Occured."
 await ctx.send(
 "=tex \\begin{{align*}} {0} \\end{{align*}}".format(sympy.latex(ans))
 )
 return


@math.command()
async def integrate(ctx, equation: str, variable: str, upperlimit, lowerlimit):
 try:
 ans = sympy.integrate(equation, (variable, upperlimit, lowerlimit))
 except:
 ans = "Erro Occured."
 await ctx.send(
 "=tex \\begin{{align*}} {0} \\end{{align*}}".format(sympy.latex(ans))
 )
 return


@math.command()
async def plot(ctx, equation, x_min, x_max, delta_x):
 plt.clf()
 figure = BytesIO() # File like object to save png
 x_min = float(x_min)
 x_max = float(x_max)
 delta_x = float(delta_x)

 print("Equation:", equation)
 X = np.arange(x_min, x_max, delta_x)
 Y = [eval(equation) for x in X]
 plt.plot(X, Y)
 plt.ylabel(equation)
 plt.xlabel("x")
 plt.savefig(figure, format="png")
 figure.seek(0)
 # Encode data with base64 so it can be served
 # figure = base64.standard_b64encode(figure.read()).decode()

 for i in channels:
 channel = bot.get_channel(i)
 try:
 if channel.name == ctx.message.channel.name:
 file = discord.File(figure, filename="plot.png")
 await channel.send("Powered by matplotlib", file=file)
 except AttributeError:
 print("AttributeError")

API_KEY = "ADD API KEY HERE"
bot.run(API_KEY)