2019-05-29 19:23:45 -04:00

1281 lines
54 KiB
Python

###
# Copyright (c) 2019 oddluck
# All rights reserved.
#
#
###
import supybot.ansi as ansi
import supybot.utils as utils
from supybot.commands import *
import supybot.plugins as plugins
import supybot.ircutils as ircutils
import supybot.callbacks as callbacks
import supybot.ircmsgs as ircmsgs
import os
import requests
from PIL import Image, ImageOps, ImageFont, ImageDraw
import numpy as np
import sys, math
from fake_useragent import UserAgent
from colour.difference import *
import re
import pexpect
import time
import random as random
from x256 import x256
import pyimgur
try:
from supybot.i18n import PluginInternationalization
_ = PluginInternationalization('Weed')
except ImportError:
# Placeholder that allows to run the plugin on a bot
# without the i18n module
_ = lambda x: x
class ASCII(callbacks.Plugin):
"""Uses API to retrieve information"""
threaded = True
def __init__(self, irc):
self.__parent = super(ASCII, self)
self.__parent.__init__(irc)
self.colors = 83
self.char = 0
self.stopped = {}
self.rgbColors = [
(255,255,255),
(0,0,0),
(0,0,127),
(0,147,0),
(255,0,0),
(127,0,0),
(156,0,156),
(252,127,0),
(255,255,0),
(0,252,0),
(0,147,147),
(0,255,255),
(0,0,252),
(255,0,255),
(127,127,127),
(210,210,210),
(71,0,0),
(71,33,0),
(71,71,0),
(50,71,0),
(0,71,0),
(0,71,44),
(0,71,71),
(0,39,71),
(0,0,71),
(46,0,71),
(71,0,71),
(71,0,42),
(116,0,0),
(116,58,0),
(116,116,0),
(81,116,0),
(0,116,0),
(0,116,73),
(0,116,116),
(0,64,116),
(0,0,116),
(75,0,116),
(116,0,116),
(116,0,69),
(181,0,0),
(181,99,0),
(181,181,0),
(125,181,0),
(0,181,0),
(0,181,113),
(0,181,181),
(0,99,181),
(0,0,181),
(117,0,181),
(181,0,181),
(181,0,107),
(255,0,0),
(255,140,0),
(255,255,0),
(178,255,0),
(0,255,0),
(0,255,160),
(0,255,255),
(0,140,255),
(0,0,255),
(165,0,255),
(255,0,255),
(255,0,152),
(255,89,89),
(255,180,89),
(255,255,113),
(207,255,96),
(111,255,111),
(101,255,201),
(109,255,255),
(89,180,255),
(89,89,255),
(196,89,255),
(255,102,255),
(255,89,188),
(255,156,156),
(255,211,156),
(255,255,156),
(226,255,156),
(156,255,156),
(156,255,219),
(156,255,255),
(156,211,255),
(156,156,255),
(220,156,255),
(255,156,255),
(255,148,211),
(0,0,0),
(19,19,19),
(40,40,40),
(54,54,54),
(77,77,77),
(101,101,101),
(129,129,129),
(159,159,159),
(188,188,188),
(226,226,226),
(255,255,255)]
self.ircColors= {
(100.0, 0.0053, -0.0104):0,
(0.0, 0.0, 0.0):1,
(12.8119, 47.2407, -64.3396):2,
(52.8041, -57.1624, 55.1703):3,
(53.2329, 80.1093, 67.2201):4,
(25.2966, 47.7847, 37.7562):5,
(36.8705, 68.0659, -42.1489):6,
(66.4237, 42.1616, 73.4335):7,
(97.1382, -21.5559, 94.4825):8,
(86.8105, -85.4149, 82.4382):9,
(55.0455, -31.8888, -9.3772):10,
(91.1165, -48.0796, -14.1381):11,
(31.8712, 78.4892, -106.9003):12,
(60.3199, 98.2542, -60.843):13,
(53.1928, 0.0029, -0.0061):14,
(84.1985, 0.0045, -0.0089):15,
(11.5497, 31.8768, 18.1739):16,
(17.5866, 15.7066, 25.9892):17,
(29.0208, -8.5776, 37.5533):18,
(27.2543, -19.015, 35.3673):19,
(25.2798, -34.2963, 32.8426):20,
(25.8276, -27.5812, 10.7515):21,
(26.6245, -19.1316, -5.6261):22,
(14.986, 1.2467, -23.6473):23,
(4.1091, 27.6851, -41.3905):24,
(9.2862, 34.8709, -32.6869):25,
(14.3696, 39.0991, -24.2113):26,
(12.6512, 34.8073, -6.066):27,
(22.699, 44.779, 34.3145):28,
(31.2054, 21.9979, 41.7676):29,
(47.2407, -12.0488, 52.8125):30,
(44.5753, -27.9355, 49.6338):31,
(41.9858, -48.1745, 46.4957):32,
(42.7207, -39.1444, 16.0528):33,
(43.8747, -26.8746, -7.9028):34,
(26.5278, 3.8603, -34.8152):35,
(11.0, 44.2673, -60.2918):36,
(19.0423, 48.637, -46.7161):37,
(26.6606, 54.9202, -34.0091):38,
(24.1377, 48.6121, -6.8774):39,
(37.5243, 61.9327, 51.9413):40,
(50.3904, 27.7338, 58.553):41,
(71.4677, -16.6651, 73.0447):42,
(67.5818, -39.987, 68.408):43,
(64.1995, -66.6294, 64.3075):44,
(65.1526, -54.8772, 23.9922):45,
(66.8122, -37.1703, -10.9303):46,
(41.6262, 7.9137, -50.0682):47,
(21.343, 61.2273, -83.3898):48,
(32.0743, 66.9878, -65.2716):49,
(43.0033, 75.9603, -47.0378):50,
(39.3866, 66.9043, -7.4929):51,
(69.4811, 36.8308, 75.4949):53,
(92.125, -51.6335, 88.501):55,
(87.737, -86.1846, 83.1812):56,
(88.9499, -71.2147, 31.6061):57,
(58.0145, 11.3842, -65.6058):59,
(32.3026, 79.1967, -107.8637):60,
(45.9331, 86.4699, -84.8483):61,
(55.6111, 86.4597, -9.1916):63,
(60.8927, 62.8729, 35.0702):64,
(78.8241, 18.6736, 56.0796):65,
(97.6208, -17.6977, 66.4162):66,
(94.1539, -37.4631, 68.7023):67,
(89.8813, -66.1541, 56.3842):68,
(91.0093, -53.1765, 13.8066):69,
(92.571, -38.7824, -11.8131):70,
(70.8615, -4.4808, -45.0866):71,
(47.6091, 49.3215, -82.3961):72,
(58.1323, 68.3853, -64.8302):73,
(68.0079, 76.2368, -48.6298):74,
(63.3723, 71.7112, -18.3923):75,
(74.4686, 36.8822, 15.7988):76,
(87.1187, 8.2035, 33.0227):77,
(98.1056, -13.9188, 47.2642):78,
(96.0337, -24.7493, 44.2003):79,
(92.1264, -48.2196, 38.3812):80,
(93.3211, -36.9827, 8.0947):81,
(94.2302, -28.9926, -9.1665):82,
(82.3123, -6.9657, -27.1167):83,
(68.0684, 23.3938, -49.2364):84,
(73.6833, 41.0464, -40.0349):85,
(77.4342, 51.3197, -33.9217):86,
(74.2811, 48.1595, -14.7725):87,
(5.8822, 0.0022, -0.0022):89,
(16.1144, 0.0022, -0.0033):90,
(22.6151, 0.0018, -0.004):91,
(32.7476, 0.0018, -0.0044):92,
(42.7837, 0.0032, -0.0055):93,
(53.9767, 0.0034, -0.0063):94,
(65.4912, 0.0036, -0.0074):95,
(76.2461, 0.0044, -0.0083):96,
(89.8837, 0.0048, -0.0094):97}
def doPrivmsg(self, irc, msg):
channel = msg.args[0]
self.stopped.setdefault(channel, None)
if msg.args[1].lower().strip()[1:] == 'cq':
self.stopped[channel] = True
def renderImage(self, text, size=18, defaultBg = 1, defaultFg = 0):
try:
if utf8 and not isinstance(text, unicode):
text = text.decode('utf-8')
except:
pass
text = text.replace('\t', ' ')
self.strip_colors_regex = re.compile('(\x03([0-9]{1,2})(,[0-9]{1,2})?)|[\x0f\x02\x1f\x03\x16]').sub
path = os.path.dirname(os.path.abspath(__file__))
defaultFont = "{0}/DejaVuSansMono.ttf".format(path)
def strip_colors(string):
return self.strip_colors_regex('', string)
_colorRegex = re.compile('(([0-9]{1,2})(,([0-9]{1,2}))?)')
IGNORE_CHRS = ('\x16','\x1f','\x02', '\x03', '\x0f')
lineLens = [len(line) for line in strip_colors(text).splitlines()]
maxWidth, height = max(lineLens), len(lineLens)
font = ImageFont.truetype(defaultFont, size)
fontX, fontY = font.getsize(' ')
fontX -= 1
fontY += 3
imageX, imageY = maxWidth * fontX, height * fontY
image = Image.new('RGB', (imageX, imageY), self.rgbColors[defaultBg])
draw = ImageDraw.Draw(image)
dtext, drect, match, x, y, fg, bg = draw.text, draw.rectangle, _colorRegex.match, 0, 0, defaultFg, defaultBg
start = time.time()
for text in text.split('\n'):
ll, i = len(text), 0
while i < ll:
chr = text[i]
if chr == "\x03":
m = match(text[i+1:i+6])
if m:
i+= len(m.group(1))
fg = int(m.group(2))
if m.group(4) is not None:
bg = int(m.group(4))
else:
bg, fg = defaultBg, defaultFg
elif chr == "\x0f":
fg, bg = defaultFg, defaultBg
elif chr not in IGNORE_CHRS:
if bg != defaultBg: # bg is not white, render it
drect((x, y, x+fontX, y+fontY), fill=self.rgbColors[bg])
if bg != fg: # text will show, render it. this saves a lot of time!
dtext((x, y), chr, font=font, fill=self.rgbColors[fg])
x += fontX
i += 1
y += fontY
fg, bg, x = defaultFg, defaultBg, 0
return image, imageX, imageY
def png(self, irc, msg, args, optlist, url):
"""[--bg] [--fg] <url>
Generate PNG from text file
"""
optlist = dict(optlist)
if 'bg' in optlist:
bg = optlist.get('bg')
else:
bg = 1
if 'fg' in optlist:
fg = optlist.get('fg')
else:
fg = 0
if url.startswith("https://paste.ee/p/"):
url = re.sub("https://paste.ee/p/", "https://paste.ee/r/", url)
file = requests.get(url)
file = file.text
im, x, y = self.renderImage(file, 18, bg, fg)
path = os.path.dirname(os.path.abspath(__file__))
filepath = "{0}/tmp/tldr.png".format(path)
im.save(filepath, "PNG")
CLIENT_ID = self.registryValue('imgurAPI')
imgur = pyimgur.Imgur(CLIENT_ID)
uploaded_image = imgur.upload_image(filepath, title=url)
irc.reply(uploaded_image.link, noLengthCheck=True, private=False, notice=False)
png = wrap(png, [getopts({'bg':'int', 'fg':'int'}), ('text')])
def ascii(self, irc, msg, args, channel, optlist, text):
"""[<channel>] [--font <font>] [--color <color1,color2>] [<text>]
Text to ASCII art
"""
if not channel:
channel = msg.args[0]
channel = msg.args[0]
optlist = dict(optlist)
font = None
words = []
if text:
text = text.strip()
if '|' in text:
words = text.split('|')
if 'color' in optlist:
color = optlist.get('color')
if "," in color:
color = color.split(",")
color1 = color[0].strip()
color2 = color[1].strip()
else:
color1 = color
color2 = None
else:
color1 = None
color2 = None
if 'font' in optlist:
font = optlist.get('font')
if words:
for word in words:
if word.strip():
data = requests.get("https://artii.herokuapp.com/make?text={0}&font={1}".format(word.strip(), font))
for line in data.text.splitlines():
if line.strip():
irc.reply(ircutils.mircColor(line, color1, color2), prefixNick=False, private=False, notice=False)
else:
data = requests.get("https://artii.herokuapp.com/make?text={0}&font={1}".format(text, font))
for line in data.text.splitlines():
if line.strip():
irc.reply(ircutils.mircColor(line, color1, color2), prefixNick=False, private=False, notice=False, to=channel)
elif 'font' not in optlist:
if words:
for word in words:
if word.strip():
data = requests.get("https://artii.herokuapp.com/make?text={0}&font=univers".format(word.strip()))
for line in data.text.splitlines():
if line.strip():
irc.reply(ircutils.mircColor(line, color1, color2), prefixNick=False, private=False, notice=False, to=channel)
else:
data = requests.get("https://artii.herokuapp.com/make?text={0}&font=univers".format(text))
for line in data.text.splitlines():
if line.strip():
irc.reply(ircutils.mircColor(line, color1, color2), prefixNick=False, private=False, notice=False, to=channel)
ascii = wrap(ascii, [optional('channel'), getopts({'font':'text', 'color':'text'}), ('text')])
def getAverageC(self, pixel, speed):
"""
Given PIL Image, return average RGB value
"""
pixel = tuple(pixel)
if self.colors == 16:
colors = list(self.ircColors.keys())[:16]
else:
colors = list(self.ircColors.keys())
if pixel not in self.matches:
closest_colors = sorted(colors, key=lambda color: self.distance(color, self.rgb2lab(pixel), speed))
closest_color = closest_colors[0]
self.matches[pixel] = self.ircColors[closest_color]
return self.matches[pixel]
def rgb2lab (self, inputColor) :
num = 0
RGB = [0, 0, 0]
for value in inputColor :
value = float(value) / 255
if value > 0.04045 :
value = ( ( value + 0.055 ) / 1.055 ) ** 2.4
else :
value = value / 12.92
RGB[num] = value * 100
num = num + 1
XYZ = [0, 0, 0,]
X = RGB [0] * 0.4124 + RGB [1] * 0.3576 + RGB [2] * 0.1805
Y = RGB [0] * 0.2126 + RGB [1] * 0.7152 + RGB [2] * 0.0722
Z = RGB [0] * 0.0193 + RGB [1] * 0.1192 + RGB [2] * 0.9505
XYZ[ 0 ] = round( X, 4 )
XYZ[ 1 ] = round( Y, 4 )
XYZ[ 2 ] = round( Z, 4 )
XYZ[ 0 ] = float( XYZ[ 0 ] ) / 95.047 # ref_X = 95.047 Observer= 2°, Illuminant= D65
XYZ[ 1 ] = float( XYZ[ 1 ] ) / 100.0 # ref_Y = 100.000
XYZ[ 2 ] = float( XYZ[ 2 ] ) / 108.883 # ref_Z = 108.883
num = 0
for value in XYZ :
if value > 0.008856 :
value = value ** ( 0.3333333333333333 )
else :
value = ( 7.787 * value ) + ( 16 / 116 )
XYZ[num] = value
num = num + 1
Lab = [0, 0, 0]
L = ( 116 * XYZ[ 1 ] ) - 16
a = 500 * ( XYZ[ 0 ] - XYZ[ 1 ] )
b = 200 * ( XYZ[ 1 ] - XYZ[ 2 ] )
Lab [ 0 ] = round( L, 4 )
Lab [ 1 ] = round( a, 4 )
Lab [ 2 ] = round( b, 4 )
return Lab
def distance(self, c1, c2, speed):
if speed == 'faster':
(r1,g1,b1) = (c1[0], c1[1], c1[2])
(r2,g2,b2) = (c2[0], c2[1], c2[2])
delta_e = math.sqrt((r1 - r2)**2 + (g1 - g2) ** 2 + (b1 - b2) **2)
elif speed == 'fast':
delta_e = delta_E_CIE1976(c1, c2)
elif speed == 'slow':
delta_e = delta_E_CIE1994(c1, c2)
elif speed == 'slower':
delta_e = delta_E_CMC(c1, c2)
elif speed == 'slowest':
delta_e = delta_E_CIE2000(c1, c2)
elif speed == 'insane':
delta_e = delta_E_DIN99(c1, c2)
return delta_e
def img(self, irc, msg, args, channel, optlist, url):
"""[<channel>] [--w <width>] [--16] [--chars <text>] [--ramp <text>] [--bg <0-98>] [--fg <0-99>] [--invert] [--nocolor] <url>
Converts image to ASCII art.
--w set width. Default 100
--16 for 16 colors. Default 99.
--invert inverts default ramp.
--chars <TEXT> color text.
--ramp <TEXT> set ramp. e.g. ".:-=+*#%@"
--bg <0-98> set background color.
--fg <0-99> set foreground color.
--nocolor text only greyscale.
"""
if not channel:
channel = msg.args[0]
optlist = dict(optlist)
if '16' in optlist:
self.colors = 16
else:
self.colors = 93
if 'faster' in optlist:
speed = 'faster'
elif 'fast' in optlist:
speed = 'fast'
elif 'slow' in optlist:
speed = 'slow'
elif 'slower' in optlist:
speed = 'slower'
elif 'slowest' in optlist:
speed = 'slowest'
elif 'insane' in optlist:
speed = 'insane'
else:
speed = 'slowest'
if 'w' in optlist:
cols = optlist.get('w')
else:
cols = 100
if 'invert' in optlist:
gscale = "$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:\"^`'."
elif 'chars' in optlist:
gscale = optlist.get('chars')
elif 'ramp' in optlist:
gscale = optlist.get('ramp')
else:
gscale = ".'`^\":;Il!i><~+_-?][}{1)(|\/tfjrxnuvczXYUJCLQ0OZmwqpdbkhao*#MW&8%B@$"
if 'delay' in optlist:
delay = optlist.get('delay')
else:
delay = self.registryValue('delay', msg.args[0])
if 'bg' in optlist:
bg = optlist.get('bg')
else:
bg = 1
if 'fg' in optlist:
fg = optlist.get('fg')
else:
fg = 99
if 'chars' not in optlist and 'ramp' not in optlist and bg == 0 or bg == 98:
gscale = "$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:\"^`'."
if 'nocolor' in optlist and 'chars' in optlist:
return
elif 'nocolor' in optlist and bg == 0 or bg == 98:
gscale = "@%#*+=-:. "
elif 'nocolor' in optlist and 'invert' in optlist:
gscale = "@%#*+=-:. "
elif 'nocolor' in optlist and 'ramp' not in optlist:
gscale = " .:-=+*#%@"
if not gscale.strip():
gscale = '\xa0'
path = os.path.dirname(os.path.abspath(__file__))
filepath = "{0}/tmp".format(path)
filename = "{0}/{1}".format(filepath, url.split('/')[-1])
ua = UserAgent()
header = {'User-Agent':str(ua.random)}
image_formats = ("image/png", "image/jpeg", "image/jpg", "image/gif")
r = requests.head(url, headers=header)
if r.headers["content-type"] in image_formats:
response = requests.get(url, headers=header)
else:
irc.reply("Invalid file type.", private=False, notice=False)
return
if response.status_code == 200:
with open("{0}".format(filename), 'wb') as f:
f.write(response.content)
# open image and convert to grayscale
image = Image.open(filename).convert('L')
if 'nocolor' not in optlist:
image2 = Image.open(filename)
if image2.mode == 'RGBA':
image2 = Image.alpha_composite(Image.new("RGBA", image2.size, self.rgbColors[bg] + (255,)), image2)
if image2.mode != 'RGB':
image2 = image2.convert('RGB')
try:
os.remove(filename)
except:
pass
# store dimensions
W, H = image.size[0], image.size[1]
# compute width of tile
w = W/cols
# compute tile height based on aspect ratio and scale
scale = 0.5
h = w/scale
# compute number of rows
rows = int(H/h)
image = ImageOps.autocontrast(image)
image = image.resize((cols, rows), Image.LANCZOS)
if 'nocolor' not in optlist:
image2 = ImageOps.autocontrast(image2)
image2 = image2.resize((cols, rows), Image.LANCZOS)
if 'dither' in optlist:
image2 = image2.convert('P', dither=Image.FLOYDSTEINBERG, palette=Image.ADAPTIVE)
else:
image2 = image2.convert('P', dither=None, palette=Image.ADAPTIVE)
image2 = image2.convert('RGB')
colormap = np.array(image2)
self.matches = {}
lumamap = np.array(image)
# ascii image is a list of character strings
aimg = []
# generate list of dimensions
for j in range(rows):
y1 = int(j*h)
y2 = int((j+1)*h)
# correct last tile
if j == rows-1:
y2 = H
# append an empty string
aimg.append("")
old_color = None
for i in range(cols):
# get average luminance
avg = int(np.average(lumamap[j][i]))
# look up ascii char
if 'chars' not in optlist:
gsval = gscale[int((avg * (len(gscale) - 1))/255)]
else:
if self.char < len(gscale):
gsval = gscale[self.char]
self.char += 1
else:
self.char = 0
gsval = gscale[self.char]
self.char += 1
# get color value
if 'nocolor' not in optlist and i == 0:
color = self.getAverageC(colormap[j][i].tolist(), speed)
old_color = color
if 'bg' not in optlist:
if gsval != '\xa0':
if gsval.isdigit():
color = "{:02d}".format(int(color))
aimg[j] += "\x03{0}{1}".format(color, gsval)
else:
aimg[j] += "\x03{0}{1}".format(int(color), gsval)
else:
aimg[j] += "\x030,{0} ".format(int(color))
else:
if gsval != '\xa0':
if gsval.isdigit():
newbg = "{:02d}".format(int(bg))
aimg[j] += "\x03{0},{1}{2}".format(int(color), newbg, gsval)
else:
aimg[j] += "\x03{0},{1}{2}".format(int(color), int(bg), gsval)
else:
aimg[j] += "\x030,{0} ".format(int(color))
elif 'nocolor' not in optlist and gsval != ' ':
color = self.getAverageC(colormap[j][i].tolist(), speed)
if color != old_color:
old_color = color
# append ascii char to string
if 'bg' not in optlist:
if gsval != '\xa0':
if gsval.isdigit():
color = "{:02d}".format(int(color))
aimg[j] += "\x03{0}{1}".format(color, gsval)
else:
aimg[j] += "\x03{0}{1}".format(int(color), gsval)
else:
aimg[j] += "\x030,{0} ".format(int(color))
else:
if gsval != '\xa0':
if gsval.isdigit():
newbg = "{:02d}".format(int(bg))
aimg[j] += "\x03{0},{1}{2}".format(int(color), newbg, gsval)
else:
aimg[j] += "\x03{0},{1}{2}".format(int(color), int(bg), gsval)
else:
aimg[j] += "\x030,{0} ".format(int(color))
else:
aimg[j] += "{0}".format(gsval)
else:
aimg[j] += "{0}".format(gsval)
# return txt image
output = aimg
del image
try:
del image2
del colormap
except:
pass
paste = ""
self.stopped[msg.args[0]] = False
for line in output:
if 'nocolor' in optlist and 'fg' in optlist and 'bg' in optlist:
newbg = "{:02d}".format(int(bg))
line = "\x03{0},{1}{2}".format(int(fg), newbg, line)
elif 'nocolor' in optlist and 'fg' in optlist:
newfg = "{:02d}".format(int(fg))
line = "\x03{0}{1}".format(newfg, line)
elif 'nocolor' in optlist and 'bg' in optlist:
newbg = "{:02d}".format(int(bg))
line = "\x0399,{0}{1}".format(newbg, line)
if self.registryValue('pasteEnable', msg.args[0]):
paste += line + "\n"
if not self.stopped[msg.args[0]]:
time.sleep(delay)
irc.reply(line, prefixNick=False, noLengthCheck=True, private=False, notice=False, to=channel)
if self.registryValue('pasteEnable', msg.args[0]):
try:
apikey = self.registryValue('pasteAPI')
payload = {'description':url,'sections':[{'contents':paste}]}
headers = {'X-Auth-Token':apikey}
post_response = requests.post(url='https://api.paste.ee/v1/pastes', json=payload, headers=headers)
response = post_response.json()
irc.reply(response['link'].replace('/p/', '/r/'), private=False, notice=False, to=channel)
except:
return
#irc.reply("Error. Did you set a valid Paste.ee API Key? https://paste.ee/account/api")
self.char = 0
img = wrap(img,[optional('channel'), getopts({'w':'int', 'invert':'', 'fast':'', 'faster':'', 'slow':'', 'slower':'', 'slowest':'', 'insane':'', '16':'', 'delay':'float', 'dither':'', 'chars':'text', 'bg':'int', 'fg':'int', 'ramp':'text', 'nocolor':''}), ('text')])
def fontlist(self, irc, msg, args):
"""
get list of fonts for text-to-ascii-art
"""
fontlist = requests.get("https://artii.herokuapp.com/fonts_list")
response = sorted(fontlist.text.split('\n'))
irc.reply(str(response).replace('\'', '').replace('[', '').replace(']', ''))
fontlist = wrap(fontlist)
def scroll(self, irc, msg, args, channel, optlist, url):
"""[<channel>] <url>
Play ASCII/ANSI art files from web links
"""
if not channel:
channel = msg.args[0]
optlist = dict(optlist)
self.stopped[msg.args[0]] = False
if 'delay' in optlist:
delay = optlist.get('delay')
else:
delay = self.registryValue('delay', msg.args[0])
if url.startswith("https://paste.ee/p/"):
url = re.sub("https://paste.ee/p/", "https://paste.ee/r/", url)
ua = UserAgent()
header = {'User-Agent':str(ua.random)}
r = requests.head(url, headers=header)
if "text/plain" in r.headers["content-type"]:
file = requests.get(url, headers=header)
else:
irc.reply("Invalid file type.", private=False, notice=False)
return
file = file.text.lstrip().splitlines()
for line in file:
if line.strip() and not self.stopped[msg.args[0]]:
time.sleep(delay)
irc.reply(line, prefixNick = False, noLengthCheck=True, private=False, notice=False, to=channel)
scroll = wrap(scroll, [optional('channel'), getopts({'delay':'float'}), ('text')])
def a2m(self, irc, msg, args, channel, optlist, url):
"""[<channel>] [--l] [--r] [--n] [--p] [--t] [--w] [--delay] <url>
Convert ANSI files to IRC formatted text. https://github.com/tat3r/a2m
"""
if not channel:
channel = msg.args[0]
optlist = dict(optlist)
opts = ''
if 'l' in optlist:
l = optlist.get('l')
opts += '-l {0} '.format(l)
if 'r' in optlist:
r = optlist.get('r')
opts += '-r {0} '.format(r)
if 'n' in optlist:
opts += '-n '.format(n)
if 'p' in optlist:
opts += '-p '.format(p)
if 't' in optlist:
t = optlist.get('t')
opts += '-t {0} '.format(t)
if 'w' in optlist:
w = optlist.get('w')
opts += '-w {0} '.format(w)
else:
opts += '-w 80 '
if 'delay' in optlist:
delay = optlist.get('delay')
else:
delay = self.registryValue('delay', msg.args[0])
ua = UserAgent()
header = {'User-Agent':str(ua.random)}
r = requests.head(url, headers=header)
try:
if "text/plain" in r.headers["content-type"] or "application/octet-stream" in r.headers["content-type"] and int(r.headers["content-length"]) < 1000000:
path = os.path.dirname(os.path.abspath(__file__))
filepath = "{0}/tmp".format(path)
filename = "{0}/{1}".format(filepath, url.split('/')[-1])
r = requests.get(url, headers=header)
open(filename, 'wb').write(r.content)
try:
output = pexpect.run('a2m {0} {1}'.format(opts.strip(), str(filename)))
try:
os.remove(filename)
except:
pass
except:
irc.reply("Error. Have you installed A2M? https://github.com/tat3r/a2m", private=False, notice=False)
return
else:
irc.reply("Invalid file type.")
return
except:
irc.reply("Invalid file type.")
return
paste = ""
self.stopped[msg.args[0]] = False
for line in output.splitlines():
line = line.decode()
if self.registryValue('pasteEnable', msg.args[0]):
paste += line + "\n"
if line.strip() and not self.stopped[msg.args[0]]:
time.sleep(delay)
irc.reply(line, prefixNick = False, noLengthCheck=True, private=False, notice=False, to=channel)
if self.registryValue('pasteEnable', msg.args[0]):
try:
apikey = self.registryValue('pasteAPI')
payload = {'description':url,'sections':[{'contents':paste}]}
headers = {'X-Auth-Token':apikey}
post_response = requests.post(url='https://api.paste.ee/v1/pastes', json=payload, headers=headers)
response = post_response.json()
irc.reply(response['link'].replace('/p/', '/r/'), private=False, notice=False, to=channel)
except:
return
irc.reply("Error. Did you set a valid Paste.ee API Key? https://paste.ee/account/api")
a2m = wrap(a2m, [optional('channel'), getopts({'l':'int', 'r':'int', 't':'int', 'w':'int', 'delay':'float'}), ('text')])
def p2u(self, irc, msg, args, channel, optlist, url):
"""[<channel>] [--b] [--f] [--p] [--s] [--t] [--w] [--delay] <url>
Picture to Unicode. https://git.trollforge.org/p2u/about/
"""
if not channel:
channel = msg.args[0]
optlist = dict(optlist)
opts = ''
if 'b' in optlist:
b = optlist.get('b')
opts += '-b {0} '.format(b)
if 'f' in optlist:
f = optlist.get('f')
opts += '-f {0} '.format(f)
else:
opts += '-f m '
if 'p' in optlist:
p = optlist.get('p')
opts += '-p {0} '.format(p)
else:
opts += '-p x '
if 's' in optlist:
s = optlist.get('s')
opts += '-s {0} '.format(s)
if 't' in optlist:
t = optlist.get('t')
opts += '-t {0} '.format(t)
if 'w' in optlist:
w = optlist.get('w')
opts += '-w {0} '.format(w)
else:
opts += '-w 80 '
if 'delay' in optlist:
delay = optlist.get('delay')
else:
delay = self.registryValue('delay', msg.args[0])
path = os.path.dirname(os.path.abspath(__file__))
filepath = "{0}/tmp".format(path)
filename = "{0}/{1}".format(filepath, url.split('/')[-1])
ua = UserAgent()
header = {'User-Agent':str(ua.random)}
image_formats = ("image/png", "image/jpeg", "image/jpg", "image/gif")
r = requests.head(url, headers=header)
if r.headers["content-type"] in image_formats:
response = requests.get(url, headers=header)
else:
irc.reply("Invalid file type.", private=False, notice=False)
return
if response.status_code == 200:
with open("{0}".format(filename), 'wb') as f:
f.write(response.content)
try:
output = pexpect.run('p2u -f m {0} {1}'.format(opts.strip(), str(filename)))
try:
os.remove(filename)
except:
pass
except:
irc.reply("Error. Have you installed p2u? https://git.trollforge.org/p2u", private=False, notice=False)
return
paste = ""
self.stopped[msg.args[0]] = False
for line in output.splitlines():
line = line.decode()
line = re.sub('^\x03 ', ' ', line)
if self.registryValue('pasteEnable', msg.args[0]):
paste += line + "\n"
if line.strip() and not self.stopped[msg.args[0]]:
time.sleep(delay)
irc.reply(line, prefixNick = False, noLengthCheck=True, private=False, notice=False, to=channel)
if self.registryValue('pasteEnable', msg.args[0]):
try:
apikey = self.registryValue('pasteAPI')
payload = {'description':url,'sections':[{'contents':paste}]}
headers = {'X-Auth-Token':apikey}
post_response = requests.post(url='https://api.paste.ee/v1/pastes', json=payload, headers=headers)
response = post_response.json()
irc.reply(response['link'].replace('/p/', '/r/'), private=False, notice=False, to=channel)
except:
return
#irc.reply("Error. Did you set a valid Paste.ee API Key? https://paste.ee/account/api")
else:
irc.reply("Unexpected file type or link format", private=False, notice=False)
p2u = wrap(p2u, [optional('channel'), getopts({'b':'int', 'f':'text', 'p':'text', 's':'int', 't':'int', 'w':'int', 'delay':'float'}), ('text')])
def tdf(self, irc, msg, args, channel, optlist, text):
"""[<channel>] [--f] [--j] [--w] [--e] [--r] [--delay] <text>
tdfiglet. https://github.com/tat3r/tdfiglet
"""
if not channel:
channel = msg.args[0]
optlist = dict(optlist)
opts = ''
if 'f' in optlist:
f = optlist.get('f')
opts += '-f {0} '.format(f.lower())
else:
opts += '-r '
if 'j' in optlist:
j = optlist.get('j')
opts += '-j {0} '.format(j)
if 'w' in optlist:
w = optlist.get('w')
opts += '-w {0} '.format(w)
else:
opts += '-w 80 '
if 'e' in optlist:
e = optlist.get('e')
opts += '-e {0} '.format(e)
if 'r' in optlist:
opts += '-r '
if 'delay' in optlist:
delay = optlist.get('delay')
else:
delay = self.registryValue('delay', msg.args[0])
try:
output = pexpect.run('tdfiglet -c m {0} {1}'.format(opts.strip(), text))
try:
os.remove(filename)
except:
pass
except:
irc.reply("Error. Have you installed tdfiglet? https://github.com/tat3r/tdfiglet", private=False, notice=False)
return
paste = ""
self.stopped[msg.args[0]] = False
output = output.decode().replace('\r\r\n', '\r\n')
for line in output.splitlines():
line = re.sub('\x03\x03\s*', '\x0F ', line)
line = re.sub('\x0F\s*\x03$', '', line)
if self.registryValue('pasteEnable', msg.args[0]):
paste += line + "\n"
if not line.strip() and not self.stopped[msg.args[0]]:
time.sleep(delay)
irc.reply('\xa0', prefixNick = False, noLengthCheck=True, private=False, notice=False, to=channel)
elif not self.stopped[msg.args[0]]:
time.sleep(delay)
irc.reply(line, prefixNick = False, noLengthCheck=True, private=False, notice=False, to=channel)
if self.registryValue('pasteEnable', msg.args[0]):
try:
apikey = self.registryValue('pasteAPI')
payload = {'description':text,'sections':[{'contents':paste}]}
headers = {'X-Auth-Token':apikey}
post_response = requests.post(url='https://api.paste.ee/v1/pastes', json=payload, headers=headers)
response = post_response.json()
irc.reply(response['link'].replace('/p/', '/r/'), private=False, notice=False, to=channel)
except:
return
#irc.reply("Error. Did you set a valid Paste.ee API Key? https://paste.ee/account/api")
tdf = wrap(tdf, [optional('channel'), getopts({'f':'text', 'j':'text', 'w':'int', 'e':'text', 'r':'', 'delay':'float'}), ('text')])
def toilet(self, irc, msg, args, channel, optlist, text):
"""[<channel>] [--f fontname] [--F filter1,filter2,etc.] [--w] [--delay] <text>
Toilet. -f to select font. -F to select filters. Separate multiple filters with a comma.
"""
if not channel:
channel = msg.args[0]
optlist = dict(optlist)
opts = ''
if 'f' in optlist:
f = optlist.get('f')
opts += '-f {0} '.format(f)
if 'F' in optlist:
filter = optlist.get('F')
if ',' in filter:
filter = filter.split(',')
for i in range(len(filter)):
opts += '-F {0} '.format(filter[i])
else:
opts += '-F {0} '.format(filter)
if 'w' in optlist:
w = optlist.get('w')
opts += '-w {0} '.format(w)
elif 'W' in optlist:
opts += '-W '
else:
opts += '-w 100 '
if 's' in optlist:
opts += '-s '
elif 'k' in optlist:
opts += '-k '
elif 'o' in optlist:
opts += '-o '
elif 'S' in optlist:
opts += '-S '
if 'delay' in optlist:
delay = optlist.get('delay')
else:
delay = self.registryValue('delay', msg.args[0])
try:
output = pexpect.run('toilet --irc {0} {1}'.format(opts.strip(), text))
except:
irc.reply("Error. Have you installed toilet?", private=False, notice=False)
return
paste = ""
self.stopped[msg.args[0]] = False
output = output.decode().replace('\r\r\n', '\r\n')
for line in output.splitlines():
if self.registryValue('pasteEnable', msg.args[0]):
paste += line + "\n"
if not line.strip() and not self.stopped[msg.args[0]]:
time.sleep(delay)
irc.reply('\xa0', prefixNick = False, noLengthCheck=True, private=False, notice=False, to=channel)
elif not self.stopped[msg.args[0]]:
time.sleep(delay)
irc.reply(line, prefixNick = False, noLengthCheck=True, private=False, notice=False, to=channel)
if self.registryValue('pasteEnable', msg.args[0]):
try:
apikey = self.registryValue('pasteAPI')
payload = {'description':text,'sections':[{'contents':paste}]}
headers = {'X-Auth-Token':apikey}
post_response = requests.post(url='https://api.paste.ee/v1/pastes', json=payload, headers=headers)
response = post_response.json()
irc.reply(response['link'].replace('/p/', '/r/'), private=False, notice=False, to=channel)
except:
return
#irc.reply("Error. Did you set a valid Paste.ee API Key? https://paste.ee/account/api")
toilet = wrap(toilet, [optional('channel'), getopts({'f':'text', 'F':'text', 's':'', 'S':'', 'k':'', 'w':'int', 'W':'', 'o':'', 'delay':'float'}), ('text')])
def wttr(self, irc, msg, args, channel, optlist, location):
"""[<channel>] [--16] [--99] <location/moon>
ASCII weather report from wttr.in for <location>.
--16 for 16 colors. Default
--99 for 99 colors.
Get moon phase with 'wttr moon'.
<location>?u (use imperial units).
<location>?m (metric).
<location>?<1-3> (number of days)
"""
if not channel:
channel = msg.args[0]
optlist = dict(optlist)
if 'delay' in optlist:
delay = optlist.get('delay')
else:
delay = self.registryValue('delay', msg.args[0])
if '16' in optlist:
self.colors = 16
elif '99' in optlist:
self.colors = 93
else:
self.colors = 16
if 'faster' in optlist:
speed = 'faster'
elif 'fast' in optlist:
speed = 'fast'
elif 'slow' in optlist:
speed = 'slow'
elif 'slower' in optlist:
speed = 'slower'
elif 'slowest' in optlist:
speed = 'slowest'
elif 'insane' in optlist:
speed = 'insane'
else:
speed = 'slower'
self.matches = {}
file = requests.get("http://wttr.in/{0}".format(location))
output = file.text
for i in range(0, 256):
j = '%03d' % i
output = re.sub('\x1b\[38;5;{0}m|\[38;5;{0};\d+m'.format(j), '\x03{:02d}'.format(self.getAverageC(x256.to_rgb(int(j)), speed)), output)
output = re.sub('\x1b\[38;5;{0}m|\[38;5;{0};\d+m'.format(i), '\x03{:02d}'.format(self.getAverageC(x256.to_rgb(int(i)), speed)), output)
output = output.replace('\x1b[0m', '\x0F')
output = re.sub('\x1b|\x9b|\[\d+m', '', output)
output = re.sub('\x0F(\s*)\x03', '\g<1>\x03', output)
for i in range(0, 99):
if i < 17:
i = '%02d' % i
output = re.sub('(?<=\x03{0}.)\x03{0}'.format(i), '', output)
paste = ""
self.stopped[msg.args[0]] = False
for line in output.splitlines():
line = line.strip('\x0F')
line = line.replace('', '☇☇')
if not line.strip() and not self.stopped[msg.args[0]]:
if self.registryValue('pasteEnable', msg.args[0]):
paste += line + "\n"
time.sleep(delay)
irc.reply('\xa0', prefixNick = False, noLengthCheck=True, private=False, notice=False, to=channel)
elif line.strip() and not self.stopped[msg.args[0]] and not line.startswith("Follow"):
if self.registryValue('pasteEnable', msg.args[0]):
paste += line + "\n"
time.sleep(delay)
irc.reply(line, prefixNick = False, noLengthCheck=True, private=False, notice=False, to=channel)
if self.registryValue('pasteEnable', msg.args[0]):
try:
apikey = self.registryValue('pasteAPI')
payload = {'description':location,'sections':[{'contents':paste}]}
headers = {'X-Auth-Token':apikey}
post_response = requests.post(url='https://api.paste.ee/v1/pastes', json=payload, headers=headers)
response = post_response.json()
irc.reply(response['link'].replace('/p/', '/r/'), private=False, notice=False, to=channel)
except:
return
wttr = wrap(wttr, [optional('channel'), getopts({'delay':'float', '16':'', '99':'', 'fast':'', 'faster':'', 'slow':'', 'slower':'', 'slowest':'', 'insane':''}), ('text')])
def rate(self, irc, msg, args, channel, optlist, coin):
"""[<channel>] [--16] [--99] [--sub <text>] [coin]
Crypto exchange rate info from rate.sx. http://rate.sx/:help. Use --sub to set subdomain e.g. eur, btc, etc.
Get a graph with [coin] e.g. 'rate btc'.
--16 for 16 colors. Default.
--99 for 99 colors.
"""
if not channel:
channel = msg.args[0]
optlist = dict(optlist)
if 'delay' in optlist:
delay = optlist.get('delay')
else:
delay = self.registryValue('delay', msg.args[0])
if '16' in optlist:
self.colors = 16
elif '99' in optlist:
self.colors = 93
else:
self.colors = 16
if 'faster' in optlist:
speed = 'faster'
elif 'fast' in optlist:
speed = 'fast'
elif 'slow' in optlist:
speed = 'slow'
elif 'slower' in optlist:
speed = 'slower'
elif 'slowest' in optlist:
speed = 'slowest'
elif 'insane' in optlist:
speed = 'insane'
else:
speed = 'slower'
if 'sub' in optlist:
sub = optlist.get('sub')
else:
sub = 'usd'
if not coin:
coin = ''
self.matches = {}
file = requests.get("http://{0}.rate.sx/{1}".format(sub, coin))
output = file.text
output = output.replace('\x1b[0m', '\x0F')
output = output.replace('\x1b[2m', '')
output = output.replace('\x1b[47m\x1b[30m', '\x031,0')
output = output.replace('\x1b[30m', '\x0301')
output = output.replace('\x1b[31m', '\x0304')
output = output.replace('\x1b[32m', '\x0309')
output = output.replace('\x1b[33m', '\x0308')
output = output.replace('\x1b[34m', '\x0312')
output = output.replace('\x1b[35m', '\x0306')
output = output.replace('\x1b[36m', '\x0311')
output = output.replace('\x1b[37m', '\x0300')
output = re.sub('\x0F(\s*)\x03', '\g<1>\x03', output)
output = re.sub('\x03\d\d(\s*)\x03', '\g<1>\x03', output)
if coin.strip():
output = output.replace('\x1b(B\x1b[m', '')
for i in range(0,99):
i = '%02d' % i
output = re.sub('\x1b\[38;5;{0}m'.format(i), '\x03{:02d}'.format(self.getAverageC(x256.to_rgb(int(i)), speed)), output)
for i in range(100,255):
output = re.sub('\x1b\[38;5;{0}m'.format(i), '\x03{:02d}'.format(self.getAverageC(x256.to_rgb(int(i)), speed)), output)
paste = ""
self.stopped[msg.args[0]] = False
for line in output.splitlines():
line = line.strip('\x0F')
if not line.strip() and not self.stopped[msg.args[0]]:
if self.registryValue('pasteEnable', msg.args[0]):
paste += line + "\n"
time.sleep(delay)
irc.reply('\xa0', prefixNick = False, noLengthCheck=True, private=False, notice=False, to=channel)
elif line.strip() and not self.stopped[msg.args[0]] and "Follow @igor_chubin" not in line:
if self.registryValue('pasteEnable', msg.args[0]):
paste += line + "\n"
time.sleep(delay)
irc.reply(line, prefixNick = False, noLengthCheck=True, private=False, notice=False, to=channel)
if self.registryValue('pasteEnable', msg.args[0]):
try:
apikey = self.registryValue('pasteAPI')
payload = {'description':coin,'sections':[{'contents':paste}]}
headers = {'X-Auth-Token':apikey}
post_response = requests.post(url='https://api.paste.ee/v1/pastes', json=payload, headers=headers)
response = post_response.json()
irc.reply(response['link'].replace('/p/', '/r/'), private=False, notice=False, to=channel)
except:
return
rate = wrap(rate, [optional('channel'), getopts({'delay':'float', '16':'', '99':'', 'sub':'text', 'fast':'', 'faster':'', 'slow':'', 'slower':'', 'slowest':'', 'insane':''}), optional('text')])
def cow(self, irc, msg, args, channel, optlist, text):
"""[<channel>] [--delay] [--type <character>] <text>
Cowsay
"""
if not channel:
channel = msg.args[0]
optlist = dict(optlist)
if 'delay' in optlist:
delay = optlist.get('delay')
else:
delay = self.registryValue('delay', msg.args[0])
if 'type' in optlist:
type = optlist.get('type')
else:
type = 'default'
data = requests.get("https://easyapis.soue.tk/api/cowsay?text={0}&type={1}".format(text, type))
self.stopped[msg.args[0]] = False
paste = ''
for line in data.text.splitlines():
if self.registryValue('pasteEnable', msg.args[0]):
paste += line + "\n"
if not line.strip() and not self.stopped[msg.args[0]]:
time.sleep(delay)
irc.reply('\xa0', prefixNick = False, noLengthCheck=True, private=False, notice=False, to=channel)
elif line.strip() and not self.stopped[msg.args[0]] and "Follow @igor_chubin" not in line:
time.sleep(delay)
irc.reply(line, prefixNick = False, noLengthCheck=True, private=False, notice=False, to=channel)
if self.registryValue('pasteEnable', msg.args[0]):
try:
apikey = self.registryValue('pasteAPI')
payload = {'description':text,'sections':[{'contents':paste}]}
headers = {'X-Auth-Token':apikey}
post_response = requests.post(url='https://api.paste.ee/v1/pastes', json=payload, headers=headers)
response = post_response.json()
irc.reply(response['link'].replace('/p/', '/r/'), private=False, notice=False, to=channel)
except:
return
cow = wrap(cow, [optional('channel'), getopts({'delay':'float', 'type':'text'}), ('text')])
def fortune(self, irc, msg, args, channel, optlist):
"""[<channel>]
Returns a random ASCII from http://www.asciiartfarts.com/fortune.txt
"""
if not channel:
channel = msg.args[0]
optlist = dict(optlist)
if 'delay' in optlist:
delay = optlist.get('delay')
else:
delay = self.registryValue('delay', msg.args[0])
data = requests.get("http://www.asciiartfarts.com/fortune.txt")
fortunes = data.text.split('%\n')
fortune = random.randrange(0, len(fortunes))
for line in fortunes[fortune].splitlines():
if not line.strip() and not self.stopped[msg.args[0]]:
time.sleep(delay)
irc.reply('\xa0', prefixNick = False, noLengthCheck=True, private=False, notice=False, to=channel)
elif line.strip() and not self.stopped[msg.args[0]] and "Follow @igor_chubin" not in line:
time.sleep(delay)
irc.reply(line, prefixNick = False, noLengthCheck=True, private=False, notice=False, to=channel)
fortune = wrap(fortune, [optional('channel'), getopts({'delay':'float'})])
def fonts(self, irc, msg, args, optlist):
"""[--toilet]
List figlets. Default list are tdf fonts. --toilet for toilet fonts
"""
optlist = dict(optlist)
if 'toilet' in optlist:
try:
reply = ", ".join(sorted(os.listdir("/usr/share/figlet")))
irc.reply(reply, prefixNick=False)
except:
irc.reply("Sorry, unable to access font directory /usr/share/figlet")
else:
try:
reply = ", ".join(sorted(os.listdir("/usr/local/share/tdfiglet/fonts/")))
irc.reply("http://www.roysac.com/thedrawfonts-tdf.html", prefixNick=False)
irc.reply(reply, prefixNick=False)
except FileNotFoundError:
reply = ", ".join(sorted(os.listdir("/usr/share/figlet")))
irc.reply(reply, prefixNick=False)
except:
irc.reply("Sorry, unable to access font directories /usr/local/share/tdfiglet/fonts/ or /usr/share/figlet")
fonts = wrap(fonts, [getopts({'toilet':''})])
def cq(self, irc, msg, args):
"""
Stop the scroll.
"""
if not self.stopped[msg.args[0]]:
self.stopped[msg.args[0]] = True
irc.reply("Stopping.", prefixNick=False, private=False, notice=False)
cq = wrap(cq)
Class = ASCII