Mark Hoekveen
128 lines
4.0 KiB
Python
128 lines
4.0 KiB
Python
import asyncio
|
|
import discord
|
|
import sys, os
|
|
import praw
|
|
import random
|
|
import re
|
|
from discord.ext import commands
|
|
from sympy import latex, symbols, preview, Symbol #for LaTeX images
|
|
import youtube_dl
|
|
import ffmpeg
|
|
|
|
class General(commands.Cog):
|
|
"""Overal toegestaan."""
|
|
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.voice_states = {}
|
|
self.r = praw.Reddit(
|
|
user_agent='StroopwafelBot',
|
|
client_secret=self.bot.tokens["redditSecret"],
|
|
client_id=self.bot.tokens["redditId"]
|
|
)
|
|
self.r.read_only = True
|
|
|
|
def __check(self, ctx): #Todo: Is dit het porno kanaal?
|
|
return True
|
|
|
|
@commands.command()
|
|
@asyncio.coroutine
|
|
def huge(self, ctx, *, arg):
|
|
"""YUGE. Geeft een grote versie van de gegeven emoji terug. Werkt alleen voor custom emoji.
|
|
Bijvoorbeeld:
|
|
!huge :ancilla:
|
|
"""
|
|
id = re.search("\<\:([A-z]*)\:(\d*)\>", arg)
|
|
hugemoji = 'https://cdn.discordapp.com/emojis/' + id.group(2) + '.png'
|
|
em = discord.Embed(title=id.group(1), type='photo')
|
|
em.set_image(url=hugemoji)
|
|
try:
|
|
yield from ctx.message.delete()
|
|
except:
|
|
print("No permission to delete message")
|
|
yield from ctx.channel.send(ctx.author.mention, embed=em)
|
|
|
|
@commands.command()
|
|
@asyncio.coroutine
|
|
def tex(self, ctx, *, arg):
|
|
"""Geeft een mooi LaTeX plaatje terug.
|
|
Bijvoorbeeld:
|
|
!tex Je moeder is lelijk! $a^b b^c \\rightarrow d$
|
|
"""
|
|
preamble = "\\documentclass[10pt]{article}\\pagestyle{empty}\\usepackage[margin=1in]{geometry}\\usepackage{amsmath, amsfonts}\\begin{document}" #\\usepackage{CJKutf8}\\usepackage[T1]{fontenc}\\usepackage[english]{babel}
|
|
options = ["-T", "tight", "-z", "9", "--truecolor", "-D", "512"]
|
|
preview(r''+arg+'', viewer='file', filename='test.png', dvioptions=options, preamble=preamble)
|
|
#file = open('test.png', 'rb')
|
|
#yield from ctx.channel.send(file, filename ='test.png', content='')
|
|
try:
|
|
yield from ctx.message.delete()
|
|
except:
|
|
print("No permission to delete message")
|
|
yield from ctx.channel.send(ctx.author.mention, file=discord.File('test.png', 'tex.png'))
|
|
|
|
@commands.command(pass_context=True)
|
|
@asyncio.coroutine
|
|
def aww(self, ctx):
|
|
""" iets schattigs... """
|
|
print(type(ctx.message.channel))
|
|
raww = self.r.subreddit('aww+catloaf+tuckedinkitties+babyelephantgifs+babybigcatgifs+blep+holdmynip+gingerkitty+cathighfive+catsstandingup+catsareassholes+eyebleach')
|
|
post = raww.top('month', limit=250)
|
|
listing = list(post)
|
|
random.shuffle(listing)
|
|
post = listing[0]
|
|
|
|
try:
|
|
yield from ctx.message.delete()
|
|
except:
|
|
print("No permission to delete message")
|
|
yield from ctx.channel.send(ctx.author.mention + ' ' + post.url)
|
|
|
|
@commands.command(pass_context=True)
|
|
@asyncio.coroutine
|
|
def gif(self, ctx, yt: str, start: str="0", duration: str="5", res: str="320", fps: int=10):
|
|
""" Giffify.
|
|
Syntax: !gif [YT] [START] [DUUR] [RESOLUTIE] [FPS]
|
|
Bv: !gif RPxvTd_jCPQ 10 5 320 10
|
|
Alleen de Youtube ID is nodig.
|
|
Start en duur zijn in seconden.
|
|
Resolutie is de breedte
|
|
"""
|
|
try:
|
|
yield from ctx.message.delete()
|
|
except:
|
|
print("No permission to delete message")
|
|
try:
|
|
os.remove('temp')
|
|
except:
|
|
pass
|
|
ydl_opts = {
|
|
'outtmpl': 'temp',
|
|
'format': 'bestvideo[ext=mp4]/mp4'
|
|
}
|
|
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
|
|
ydl.download([yt])
|
|
split = ffmpeg.input('temp').trim(start=start, duration=duration).filter_('fps',fps=fps).filter_('scale', w=res, h=-1).filter_multi_output('split')
|
|
pal = split.stream(0).filter_('palettegen')
|
|
inputs = {
|
|
'v1':split.stream(1),
|
|
'v2':pal
|
|
}
|
|
gif = (
|
|
ffmpeg
|
|
.filter(stream_spec=[split.stream(1), pal], filter_name='paletteuse')
|
|
.output('out.gif')
|
|
.overwrite_output()
|
|
.run()
|
|
)
|
|
yield from ctx.channel.send(ctx.author.mention + ' hoi ', file=discord.File('out.gif'))
|
|
|
|
@commands.command(pass_context=True, hidden=True)
|
|
@asyncio.coroutine
|
|
def reboot(self, ctx, password=""):
|
|
""" Reboot.. """
|
|
if password == self.bot.passwords["resetpassword"]:
|
|
yield from ctx.channel.send("Rebooting.. brb!")
|
|
os.execv(sys.executable, ['python3'] + sys.argv)
|
|
else:
|
|
yield from ctx.channel.send(ctx.author.mention + " What's the magic word?")
|