### # 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 import numpy as np import sys, math from fake_useragent import UserAgent from colour.difference import * import re import pexpect import urllib import time import random as random 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.ircColors= { (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, (53.2329, 80.1093, 67.2201):52, (69.4811, 36.8308, 75.4949):53, (97.1382, -21.5559, 94.4825):54, (92.125, -51.6335, 88.501):55, (87.737, -86.1846, 83.1812):56, (88.9499, -71.2147, 31.6061):57, (91.1165, -48.0796, -14.1381):58, (58.0145, 11.3842, -65.6058):59, (32.3026, 79.1967, -107.8637):60, (45.9331, 86.4699, -84.8483):61, (60.3199, 98.2542, -60.843):62, (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, (0.0, 0.0, 0.0):88, (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, (100.0, 0.0053, -0.0104):98} self.colors16 = { (211, 215, 207):'00', (46, 52, 54):'01', (52, 101, 164):'02', (78, 154, 6):'03', (204, 0, 0):'04', (143, 57, 2):'05', (92, 53, 102):'06', (206, 92, 0):'07', (196, 160, 0):'08', (115, 210, 22):'09', (17, 168, 121):'10', (88, 161, 157):'11', (87, 121, 158):'12', (160, 67, 101):'13', (85, 87, 83): '14', (136, 137, 133):'15'} def ascii(self, irc, msg, args, optlist, text): """[--font ] [--color ] [] Text to ASCII art """ 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) 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) 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) 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) ascii = wrap(ascii, [getopts({'font':'text', 'color':'text'}), ('text')]) def getAverageC(self, pixel, speed): """ Given PIL Image, return average RGB value """ if self.colors == 16: speed = speed colors = list(self.colors16.keys()) closest_colors = sorted(colors, key=lambda color: self.distance(self.rgb2lab(color), self.rgb2lab(pixel), speed)) closest_color = closest_colors[0] return self.colors16[closest_color] else: speed = speed colors = list(self.ircColors.keys()) closest_colors = sorted(colors, key=lambda color: self.distance(color, self.rgb2lab(pixel), speed)) closest_color = closest_colors[0] return self.ircColors[closest_color] 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) else: delta_e = delta_E_DIN99(c1, c2) return delta_e def img(self, irc, msg, args, optlist, url): """[--cols ] [--16] [--fast | --slow | --slower] [--invert] [--delay] Converts image to ASCII art. --16 for 16 colors. --invert to invert luminance character map. Set speed to vary color difference algorithm. """ optlist = dict(optlist) if '16' in optlist: self.colors = 16 else: self.colors = 83 if '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 = 'faster' if 'cols' in optlist: cols = optlist.get('cols') else: cols = 100 if 'invert' in optlist: gscale = "$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,\"^`'." 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]) 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.") 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') image2 = Image.open(filename) try: os.remove(filename) except: pass waitmsgs = self.registryValue("randWait", msg.args[0]) waitmsg = random.choice(waitmsgs) irc.reply(waitmsg) # 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) image2 = image2.convert('RGBA') image2 = image2.convert('RGB') image2 = ImageOps.autocontrast(image2) image2 = image2.resize((cols, rows), Image.LANCZOS) lumamap = np.array(image) colormap = np.array(image2) # 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 gsval = gscale[int((avg*68)/255)] # get color value color = self.getAverageC(colormap[j][i].tolist(),speed) #color = self.getAverageC(img2,speed) if color != old_color: old_color = color # append ascii char to string aimg[j] += "\x03{0}{1}".format(color, gsval) else: aimg[j] += "{0}".format(gsval) # return txt image output = aimg del image del image2 del colormap paste = "" for line in output: if self.registryValue('pasteEnable', msg.args[0]): paste += line + "\n" time.sleep(delay) irc.reply(line, prefixNick=False, noLengthCheck=True) 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/')) except: irc.reply("Error. Did you set a valid Paste.ee API Key? https://paste.ee/account/api") img = wrap(img,[getopts({'cols':'int', 'invert':'', 'fast':'', 'slow':'', 'slower':'', 'slowest':'', 'insane':'', '16':'', 'delay':'float'}), ('text')]) def ansi(self, irc, msg, args, optlist, url): """[--cols ] [--16] [--fast | --slow | --slower] [--invert] [--delay] Converts image to ANSI art. --16 for 16 colors. --invert to invert luminance character map. Set speed to vary color difference algorithm. """ optlist = dict(optlist) if '16' in optlist: self.colors = 16 else: self.colors = 83 if '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 = 'faster' if 'cols' in optlist: cols = optlist.get('cols') else: cols = 80 if 'invert' in optlist: gscale = "█▓▒░" else: gscale = "░▒▓█" 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.") 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') image2 = Image.open(filename) try: os.remove(filename) except: pass waitmsgs = self.registryValue("randWait", msg.args[0]) waitmsg = random.choice(waitmsgs) irc.reply(waitmsg) # 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) image2 = image2.convert('RGBA') image2 = image2.convert('RGB') image2 = ImageOps.autocontrast(image2) image2 = image2.resize((cols, rows), Image.LANCZOS) lumamap = np.array(image) colormap = np.array(image2) # 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 gsval = gscale[int((avg*3)/255)] # get color value color = self.getAverageC(colormap[j][i].tolist(),speed) #color = self.getAverageC(img2,speed) if color != old_color: old_color = color # append ascii char to string aimg[j] += "\x03{0}{1}".format(color, gsval) else: aimg[j] += "{0}".format(gsval) # return txt image output = aimg del image del image2 del colormap paste = "" for line in output: if self.registryValue('pasteEnable', msg.args[0]): paste += line + "\n" time.sleep(delay) irc.reply(line, prefixNick=False, noLengthCheck=True) 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/')) except: irc.reply("Error. Did you set a valid Paste.ee API Key? https://paste.ee/account/api") ansi = wrap(ansi, [getopts({'cols':'int', 'invert':'', 'fast':'', 'slow':'', 'slower':'', 'slowest':'', 'insane':'', '16':'', 'delay':'float'}), ('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, optlist, url): """[--delay] Play ASCII/ANSI art files from web links """ optlist = dict(optlist) 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) file = requests.get(url) if "" in file.text: irc.reply("Error: ansi2irc requires a text file as input.") return elif url.endswith(".txt") or url.startswith("https://pastebin.com/raw/") or url.startswith("https://paste.ee/r/"): for line in file.text.splitlines(): if line.strip(): time.sleep(delay) irc.reply(line, prefixNick = False, noLengthCheck=True) else: irc.reply("Unexpected file type or link format") scroll = wrap(scroll, [getopts({'delay':'float'}), ('text')]) def a2m(self, irc, msg, args, optlist, url): """[--l] [--r] [--n] [--p] [--t] [--w] [--delay] Convert ANSI files to IRC formatted text. https://github.com/tat3r/a2m """ 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]) if url.lower().endswith(".ans") or url.lower().endswith(".asc"): file = requests.get(url) if "" in file.text: irc.reply("Error: ansi2irc requires a text file as input.") return try: path = os.path.dirname(os.path.abspath(__file__)) filepath = "{0}/tmp".format(path) filename = "{0}/{1}".format(filepath, url.split('/')[-1]) urllib.request.urlretrieve(url, filename) 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") return paste = "" for line in output.splitlines(): line = line.decode() if self.registryValue('pasteEnable', msg.args[0]): paste += line + "\n" if line.strip(): time.sleep(delay) irc.reply(line, prefixNick = False, noLengthCheck=True) 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/')) except: 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") a2m = wrap(a2m, [getopts({'l':'int', 'r':'int', 't':'int', 'w':'int', 'delay':'float'}), ('text')]) def p2u(self, irc, msg, args, optlist, url): """[--b] [--f] [--p] [--s] [--t] [--w] [--delay] Picture to Unicode. https://git.trollforge.org/p2u/about/ """ 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.") 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") return paste = "" for line in output.splitlines(): line = line.decode() if self.registryValue('pasteEnable', msg.args[0]): paste += line + "\n" if line.strip(): time.sleep(delay) irc.reply(line, prefixNick = False, noLengthCheck=True) 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/')) except: 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") p2u = wrap(p2u, [getopts({'b':'int', 'f':'text', 'p':'text', 's':'int', 't':'int', 'w':'int', 'delay':'float'}), ('text')]) def cq(self, irc, msg, args): """ Clear the queue """ irc.queue.reset() irc.replySuccess() cq = wrap(cq) Class = ASCII