From 04aae4a3aa354485be4b2a148ac2e2ed7bbf7208 Mon Sep 17 00:00:00 2001 From: Mattias Andrée Date: Sat, 29 Sep 2012 22:55:59 +0200 Subject: rename file ponysay to ponysay.py --- ponysay | 1779 ------------------------------------------------------------ ponysay.py | 1779 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ setup.py | 2 +- 3 files changed, 1780 insertions(+), 1780 deletions(-) delete mode 100755 ponysay create mode 100755 ponysay.py diff --git a/ponysay b/ponysay deleted file mode 100755 index d1571a7..0000000 --- a/ponysay +++ /dev/null @@ -1,1779 +0,0 @@ -#!/usr/bin/env python3 -# -*- coding: utf-8 -*- - -''' -ponysay.py - Ponysay, cowsay reimplementation for ponies -Copyright (C) 2012 Erkin Batu Altunbaş et al. - -This program is free software. It comes without any warranty, to -the extent permitted by applicable law. You can redistribute it -and/or modify it under the terms of the Do What The Fuck You Want -To Public License, Version 2, as published by Sam Hocevar. See -http://sam.zoy.org/wtfpl/COPYING for more details. - - -Authors of ponysay.py: - - Erkin Batu Altunbaş: Project leader, helped write the first implementation - Mattias "maandree" Andrée: Major contributor of both implementions - Elis "etu" Axelsson: Major contributor of current implemention and patcher of the first implementation - Sven-Hendrik "svenstaro" Haase: Major contributor of the first implementation - Jan Alexander "heftig" Steffens: Major contributor of the first implementation - Kyah "L-four" Rindlisbacher: Patched the first implementation - -''' - -import os -import shutil -import sys -import random -from subprocess import Popen, PIPE - - -''' -The version of ponysay -''' -VERSION = 'dev' # this line should not be edited, it is fixed by the build system - - - -''' -Hack to enforce UTF-8 in output (in the future, if you see anypony not using utf-8 in programs by default, report them to Princess Celestia so she can banish them to the moon) -''' -def print(text = '', end = '\n'): - sys.stdout.buffer.write((str(text) + end).encode('utf-8')) - - -''' -This is the mane class of ponysay -''' -class Ponysay(): - ''' - Starts the part of the program the arguments indicate - ''' - def __init__(self, args): - if (args.argcount == 0) and not pipelinein: - args.help() - return - - - if args.opts['-h'] is not None: args.help() - elif args.opts['--quoters'] is not None: self.quoters() - elif args.opts['--onelist'] is not None: self.onelist() - elif args.opts['-v'] is not None: self.version() - elif args.opts['-l'] is not None: self.list() - elif args.opts['-L'] is not None: self.linklist() - elif args.opts['-B'] is not None: self.balloonlist() - elif args.opts['++onelist'] is not None: self.__extraponies(); self.onelist() - elif args.opts['+l'] is not None: self.__extraponies(); self.list() - elif args.opts['+L'] is not None: self.__extraponies(); self.linklist() - elif args.opts['-A'] is not None: self.list(); self.__extraponies(); self.list() - elif args.opts['+A'] is not None: self.linklist(); self.__extraponies(); self.linklist() - else: - self.__extraponies(args) - self.__bestpony(args) - self.__ucsremap(args) - if args.opts['-q'] is not None: self.quote(args) - else: self.print_pony(args) - - - ## - ## Methods that run before the mane methods - ## - - ''' - Use extra ponies - ''' - def __extraponies(self, args = None): - if args is None: - ponydirs[:] = extraponydirs - elif args.opts['-F'] is not None: - args.opts['-f'] = args.opts['-F'] - ponydirs[:] = extraponydirs - - - ''' - Use best.pony if nothing else is set - ''' - def __bestpony(self, args): - if (args.opts['-f'] is None) or (args.opts['-q'] is None) or (len(args.opts['-q']) == 0): - for ponydir in ponydirs: - if os.path.isfile(ponydir + 'best.pony') or os.path.islink(ponydir + 'best.pony'): - pony = os.path.realpath(ponydir + 'best.pony') # Canonical path - args.opts['-f' if args.opts['-q'] is None else '-q'] = [pony] - break - - - ''' - Apply pony name remapping to args according to UCS settings - ''' - def __ucsremap(self, args): - env_ucs = os.environ['PONYSAY_UCS_ME'] if 'PONYSAY_UCS_ME' in os.environ else '' - ucs_conf = 0 - if env_ucs in ('yes', 'y', '1'): ucs_conf = 1 - elif env_ucs in ('harder', 'h', '2'): ucs_conf = 2 - - if ucs_conf == 0: - return - - maplines = [] - for ucsmap in ucsmaps: - if os.path.isfile(ucsmap): - mapfile = None - try: - mapfile = open(ucsmap, 'rb') - maplines += [line.replace('\n', '') for line in mapfile.read().decode('utf8', 'replace').split('\n')] - finally: - if mapfile is not None: - mapfile.close() - - map = {} - stripset = ' \t' # must be string, wtf! and way doesn't python's doc say so - for line in maplines: - if (len(line) > 0) and not (line[0] == '#'): - s = line.index('→') - ucs = line[:s] .strip(stripset) - ascii = line[s + 1:].strip(stripset) - map[ucs] = ascii - - for flag in ('-f', '-q'): - if args.opts[flag] is not None: - for i in range(0, len(args.opts[flag])): - if args.opts[flag][i] in map: - args.opts[flag][i] = map[args.opts[flag][i]] - - - ## - ## Auxiliary methods - ## - - ''' - Apply USC:ise pony names according to UCS settings - ''' - def __ucsise(self, ponies, links = None): - env_ucs = os.environ['PONYSAY_UCS_ME'] if 'PONYSAY_UCS_ME' in os.environ else '' - ucs_conf = 0 - if env_ucs in ('yes', 'y', '1'): ucs_conf = 1 - elif env_ucs in ('harder', 'h', '2'): ucs_conf = 2 - - if ucs_conf == 0: - return - - maplines = [] - for ucsmap in ucsmaps: - if os.path.isfile(ucsmap): - mapfile = None - try: - mapfile = open(ucsmap, 'rb') - maplines += [line.replace('\n', '') for line in mapfile.read().decode('utf8', 'replace').split('\n')] - finally: - if mapfile is not None: - mapfile.close() - - map = {} - stripset = ' \t' # must be string, wtf! and way doesn't python's doc say so - for line in maplines: - if (len(line) > 0) and not (line[0] == '#'): - s = line.index('→') - ucs = line[:s] .strip(stripset) - ascii = line[s + 1:].strip(stripset) - map[ascii] = ucs - - if ucs_conf == 1: - for pony in ponies: - if pony in map: - ponies.append(map[pony]) - if links is not None: - links[map[pony]] = pony - else: - for j in range(0, len(ponies)): - if ponies[j] in map: - ponies[j] = map[ponies[j]] - - - ''' - Returns one file with full path, names is filter for names, also accepts filepaths - ''' - def __getponypath(self, names = None): - ponies = {} - - for ponydir in ponydirs: - for ponyfile in os.listdir(ponydir): - pony = ponyfile[:-5] - if pony not in ponies: - ponies[pony] = ponydir + ponyfile - - if not names == None: - for name in names: - if os.path.exists(name): - ponies[name] = name - - if names == None: - names = list(ponies.keys()) - - pony = names[random.randrange(0, len(names))] - if pony not in ponies: - sys.stderr.write('I have never heard of anypony named %s\n' % (pony)); - exit(1) - else: - return ponies[pony] - - - ''' - Returns a set with all ponies that have quotes and are displayable - ''' - def __quoters(self): - quotes = [] - quoteshash = set() - _quotes = [] - for quotedir in quotedirs: - _quotes += [item[:item.index('.')] for item in os.listdir(quotedir)] - for quote in _quotes: - if not quote == '': - if not quote in quoteshash: - quoteshash.add(quote) - quotes.append(quote) - - ponies = set() - for ponydir in ponydirs: - for pony in os.listdir(ponydir): - if not pony[0] == '.': - p = pony[:-5] # remove .pony - for quote in quotes: - if ('+' + p + '+') in ('+' + quote + '+'): - if not p in ponies: - ponies.add(p) - - return ponies - - - ''' - Returns a list with all (pony, quote file) pairs - ''' - def __quotes(self): - quotes = [] - for quotedir in quotedirs: - quotes += [quotedir + item for item in os.listdir(quotedir)] - rc = [] - - for ponydir in ponydirs: - for pony in os.listdir(ponydir): - if not pony[0] == '.': - p = pony[:-5] # remove .pony - for quote in quotes: - q = quote[quote.rindex('/') + 1:] - q = q[:q.rindex('.')] - if ('+' + p + '+') in ('+' + q + '+'): - rc.append((p, quote)) - - return rc - - - ''' - Gets the size of the terminal in (rows, columns) - ''' - def __gettermsize(self): - termsize = Popen(['stty', 'size'], stdout=PIPE, stdin=sys.stderr).communicate()[0] - termsize = termsize.decode('utf8', 'replace')[:-1].split(' ') # [:-1] removes a \n - termsize = [int(item) for item in termsize] - return termsize - - - - ## - ## Listing methods - ## - - ''' - Columnise a list and prints it - ''' - def __columnise(self, ponies): - termwidth = self.__gettermsize()[1] + 2 - ponies.sort(key = lambda pony : pony[0]) - widths = [UCS.dispLen(pony[0]) for pony in ponies] - width = max(widths) + 2 # longest pony file name + space between columns - - cols = termwidth // width - rows = (len(ponies) + cols - 1) // cols - columns = [] - for c in range(0, cols): columns.append([]) - - (y, x) = (0, 0) - for j in range(0, len(ponies)): - cell = ponies[j][1] + ' ' * (width - widths[j]); - columns[x].append(cell) - y += 1 - if y == rows: - x += 1 - y = 0 - - diff = rows * cols - len(ponies) - if diff > 2: - c = cols - 1 - diff -= 1 - while diff > 0: - columns[c] = columns[c - 1][-diff:] + columns[c] - c -= 1 - columns[c] = columns[c][:-diff] - diff -= 1 - pass - - lines = [] - for r in range(0, rows): - lines.append([]) - for c in range(0, cols): - if r < len(columns[c]): - line = lines[r].append(columns[c][r]) - - print('\n'.join([''.join(line)[:-2] for line in lines])); - print() - - - ''' - Lists the available ponies - ''' - def list(self): - quoters = self.__quoters() - - for ponydir in ponydirs: # Loop ponydirs - _ponies = os.listdir(ponydir) - ponies = [] - for pony in _ponies: - if (len(pony) > 5) and (pony[-5:] == '.pony'): - ponies.append(pony[:-5]) - self.__ucsise(ponies) - - if len(ponies) == 0: - continue - print('\033[1mponies located in ' + ponydir + '\033[21m') - - self.__columnise([(pony, '\033[1m' + pony + '\033[21m' if pony in quoters else pony) for pony in ponies]) - - - ''' - Lists the available ponies with alternatives inside brackets - ''' - def linklist(self): - termsize = self.__gettermsize() - quoters = self.__quoters() - - for ponydir in ponydirs: # Loop ponydirs - _ponies = os.listdir(ponydir) - ponies = [] - for pony in _ponies: - if (len(pony) > 5) and (pony[-5:] == '.pony'): - ponies.append(pony[:-5]) - - if len(ponies) == 0: - continue - print('\033[1mponies located in ' + ponydir + '\033[21m') - - pseudolinkmap = {} - self.__ucsise(ponies, pseudolinkmap) - pairs = [] - for pony in ponies: - if pony in pseudolinkmap: - pairs.append((pony, pseudolinkmap[pony] + '.pony')); - else: - pairs.append((pony, os.path.realpath(ponydir + pony + '.pony') if os.path.islink(ponydir + pony + '.pony') else None)) - - ponymap = {} - for pair in pairs: - if (pair[1] is None) or (pair[1] == ''): - if pair[0] not in ponymap: - ponymap[pair[0]] = [] - else: - target = pair[1][:-5] - if '/' in target: - target = target[target.rindex('/') + 1:] - if target in ponymap: - ponymap[target].append(pair[0]) - else: - ponymap[target] = [pair[0]] - - width = 0 - ponies = {} - for pony in ponymap: - w = UCS.dispLen(pony) - item = '\033[1m' + pony + '\033[21m' if (pony in quoters) else pony - syms = ponymap[pony] - syms.sort() - if len(syms) > 0: - w += 2 + len(syms) - item += ' (' - first = True - for sym in syms: - w += UCS.dispLen(sym) - if first: first = False - else: item += ' ' - item += '\033[1m' + sym + '\033[21m' if (sym in quoters) else sym - item += ')' - ponies[(item.replace('\033[1m', '').replace('\033[21m', ''), item)] = w - if width < w: - width = w - - self.__columnise(list(ponies)) - - - ''' - Lists with all ponies that have quotes and are displayable - ''' - def quoters(self): - last = '' - ponies = [] - for pony in self.__quoters(): - ponies.append(pony) - self.__ucsise(ponies) - ponies.sort() - for pony in ponies: - if not pony == last: - last = pony - print(pony) - - - ''' - Lists the available ponies one one column without anything bold - ''' - def onelist(self): - last = '' - _ponies = [] - for ponydir in ponydirs: # Loop ponydirs - _ponies += os.listdir(ponydir) - ponies = [] - for pony in _ponies: - if (len(pony) > 5) and (pony[-5:] == '.pony'): - ponies.append(pony[:-5]) - self.__ucsise(ponies) - ponies.sort() - for pony in ponies: - if not pony == last: - last = pony - print(pony) - - - - ## - ## Balloon methods - ## - - ''' - Prints a list of all balloons - ''' - def balloonlist(self): - termsize = self.__gettermsize() - balloonset = set() - - for balloondir in balloondirs: - for balloon in os.listdir(balloondir): - if isthink and (len(balloon) > 6) and (balloon[-6:] == '.think'): - balloon = balloon[:-6] - elif (not isthink) and (len(balloon) > 4) and (balloon[-4:] == '.say'): - balloon = balloon[:-4] - else: - continue - if balloon not in balloonset: - balloonset.add(balloon) - - self.__columnise([(balloon, balloon) for balloon in list(balloonset)]) - - - ''' - Returns one file with full path, names is filter for style names, also accepts filepaths - ''' - def __getballoonpath(self, names): - if names is None: - return None - balloons = {} - - for balloondir in balloondirs: - for balloon in os.listdir(balloondir): - balloonfile = balloon - if isthink and (len(balloon) > 6) and (balloon[-6:] == '.think'): - balloon = balloon[:-6] - elif (not isthink) and (len(balloon) > 4) and (balloon[-4:] == '.say'): - balloon = balloon[:-4] - else: - continue - if balloon not in balloons: - balloons[balloon] = balloondir + balloonfile - - for name in names: - if os.path.exists(name): - balloons[name] = name - - if names == None: - names = list(balloons.keys()) - - balloon = names[random.randrange(0, len(names))] - if balloon not in balloons: - sys.stderr.write('That balloon style %s does not exist\n' % (balloon)); - exit(1) - else: - return balloons[balloon] - - - ''' - Creates the balloon style object - ''' - def __getballoon(self, balloonfile): - if balloonfile is None: - if isthink: - return Balloon('o', 'o', '( ', ' )', [' _'], ['_'], ['_'], ['_'], ['_ '], ' )', ' )', ' )', ['- '], ['-'], ['-'], ['-'], [' -'], '( ', '( ', '( ') - return Balloon('\\', '/', '< ', ' >', [' _'], ['_'], ['_'], ['_'], ['_ '], ' \\', ' |', ' /', ['- '], ['-'], ['-'], ['-'], [' -'], '\\ ', '| ', '/ ') - - map = {} - for elem in ('\\', '/', 'ww', 'ee', 'nw', 'nnw', 'n', 'nne', 'ne', 'nee', 'e', 'see', 'se', 'sse', 's', 'ssw', 'sw', 'sww', 'w', 'nww'): - map[elem] = [] - - balloonstream = None - try: - balloonstream = open(balloonfile, 'rb') - data = [line.replace('\n', '') for line in balloonstream.read().decode('utf8', 'replace').split('\n')] - finally: - if balloonstream is not None: - balloonstream.close() - - last = None - for line in data: - if len(line) > 0: - if line[0] == ':': - map[last].append(line[1:]) - else: - last = line[:line.index(':')] - value = line[len(last) + 1:] - map[last].append(value) - - return Balloon(map['\\'][0], map['/'][0], map['ww'][0], map['ee'][0], map['nw'], map['nnw'], map['n'], - map['nne'], map['ne'], map['nee'][0], map['e'][0], map['see'][0], map['se'], map['sse'], - map['s'], map['ssw'], map['sw'], map['sww'][0], map['w'][0], map['nww'][0]) - - - - ## - ## Displaying methods - ## - - ''' - Prints the name of the program and the version of the program - ''' - def version(self): - print('%s %s' % ('ponysay', VERSION)) - - - ''' - Print the pony with a speech or though bubble. message, pony and wrap from args are used. - ''' - def print_pony(self, args): - if args.message == None: - msg = ''.join(sys.stdin.readlines()).rstrip() - else: - msg = args.message - - if args.opts['-c'] is not None: ## This algorithm should give some result as cowsay's (according to tests) - buf = '' - last = ' ' - CHARS = '\t \n' - for c in msg: - if (c in CHARS) and (last in CHARS): - if last == '\n': - buf += last - last = c - else: - buf += c - last = c - msg = buf.strip(CHARS) - buf = '' - for c in msg: - if (c != '\n') or (last != '\n'): - buf += c - last = c - msg = buf.replace('\n', '\n\n') - - pony = self.__getponypath(args.opts['-f']) - - if (len(pony) > 4) and (pony[-4:].lower() == '.png'): - pony = '\'' + pony.replace('\'', '\'\\\'\'') + '\'' - pngcmd = ('img2ponysay -p -- ' if linuxvt else 'img2ponysay -- ') + pony - pngpipe = os.pipe() - Popen(pngcmd, stdout=os.fdopen(pngpipe[1], 'w'), shell=True).wait() - pony = '/proc/' + str(os.getpid()) + '/fd/' + str(pngpipe[0]) - - pony = self.__kms(pony) - - if linuxvt: - print('\033[H\033[2J', end='') - - env_width = os.environ['PONYSAY_FULL_WIDTH'] if 'PONYSAY_FULL_WIDTH' in os.environ else None - if env_width is None: env_width = '' - widthtruncation = self.__gettermsize()[1] if env_width not in ('yes', 'y', '1') else None - messagewrap = int(args.opts['-W'][0]) if args.opts['-W'] is not None else None - - balloon = self.__getballoon(self.__getballoonpath(args.opts['-b'])) - - backend = Backend(message = msg, ponyfile = pony, wrapcolumn = messagewrap if messagewrap is not None else 40, width = widthtruncation, balloon = balloon) - backend.parse() - output = backend.output - if (len(output) > 0) and (output[-1] == '\n'): - output = output[:-1] - - - env_bottom = os.environ['PONYSAY_BOTTOM'] if 'PONYSAY_BOTTOM' in os.environ else None - if env_bottom is None: env_bottom = '' - - env_height = os.environ['PONYSAY_TRUNCATE_HEIGHT'] if 'PONYSAY_TRUNCATE_HEIGHT' in os.environ else None - if env_height is None: env_height = '' - - env_lines = os.environ['PONYSAY_SHELL_LINES'] if 'PONYSAY_SHELL_LINES' in os.environ else None - if (env_lines is None) or (env_lines == ''): env_lines = '2' - - lines = self.__gettermsize()[0] - int(env_lines) - - - if linuxvt or (env_height is ('yes', 'y', '1')): - if env_bottom is ('yes', 'y', '1'): - for line in output.split('\n')[: -lines]: - print(line) - else: - for line in output.split('\n')[: lines]: - print(line) - else: - print(output) - - - ''' - Print the pony with a speech or though bubble and a self quote - ''' - def quote(self, args): - pairs = self.__quotes() - if len(args.opts['-q']) > 0: - ponyset = {} - for pony in args.opts['-q']: - if (len(pony) > 5) and (pony[-5:] == '.pony'): - ponyname = pony[:-5] - if '/' in ponyname: - ponyname = ponyname[ponyname.rindex('/') + 1:] - ponyset[ponyname] = pony - else: - ponyset[pony] = pony - alts = [] - for pair in pairs: - if pair[0] in ponyset: - alts.append((ponyset[pair[0]], pair[1])) - pairs = alts - - if not len(pairs) == 0: - pair = pairs[random.randrange(0, len(pairs))] - qfile = None - try: - qfile = open(pair[1], 'rb') - args.message = qfile.read().decode('utf8', 'replace').strip() - finally: - if qfile is not None: - qfile.close() - args.opts['-f'] = [pair[0]] - elif len(args.opts['-q']) == 0: - sys.stderr.write('Princess Celestia! All the ponies are mute!\n') - exit(1) - else: - args.opts['-f'] = [args.opts['-q'][random.randrange(0, len(args.opts['-q']))]] - args.message = 'Zecora! Help me, I am mute!' - - self.print_pony(args) - - - ''' - Indentifies whether KMS support is utilised - ''' - @staticmethod - def isUsingKMS(): - if not linuxvt: - return False - - env_kms = os.environ['PONYSAY_KMS_PALETTE'] if 'PONYSAY_KMS_PALETTE' in os.environ else None - if env_kms is None: env_kms = '' - - env_kms_cmd = os.environ['PONYSAY_KMS_PALETTE_CMD'] if 'PONYSAY_KMS_PALETTE_CMD' in os.environ else None - if (env_kms_cmd is not None) and (not env_kms_cmd == ''): - env_kms = Popen(shlex.split(env_kms_cmd), stdout=PIPE, stdin=sys.stderr).communicate()[0].decode('utf8', 'replace') - if env_kms[-1] == '\n': - env_kms = env_kms[:-1] - - return env_kms != '' - - - ''' - Returns the file name of the input pony converted to a KMS pony, or if KMS is not used, the input pony itself - ''' - def __kms(self, pony): - if not linuxvt: - return pony - - KMS_VERSION = '1' - - env_kms = os.environ['PONYSAY_KMS_PALETTE'] if 'PONYSAY_KMS_PALETTE' in os.environ else None - if env_kms is None: env_kms = '' - - env_kms_cmd = os.environ['PONYSAY_KMS_PALETTE_CMD'] if 'PONYSAY_KMS_PALETTE_CMD' in os.environ else None - if (env_kms_cmd is not None) and (not env_kms_cmd == ''): - env_kms = Popen(shlex.split(env_kms_cmd), stdout=PIPE, stdin=sys.stderr).communicate()[0].decode('utf8', 'replace') - if env_kms[-1] == '\n': - env_kms = env_kms[:-1] - - if env_kms == '': - return pony - - palette = env_kms - palettefile = env_kms.replace('\033]P', '') - - cachedir = '/var/cache/ponysay' - if not os.path.isdir(cachedir): - cachedir = HOME + '/.cache/ponysay' - if not os.path.isdir(cachedir): - os.makedirs(cachedir) - - newversion = False - if not os.path.isfile(cachedir + '/.version'): - newversion = True - else: - cachev = None - try: - cachev = open(cachedir + '/.version', 'rb') - if cachev.read().decode('utf8', 'replace').replace('\n', '') == KMS_VERSION: - newversion = True - finally: - if cachev is not None: - cachev.close() - if newversion: - for cached in os.listdir(cachedir): - cached = cachedir + '/' + cached - if os.path.isdir(cached) and not os.path.islink(cached): - shutil.rmtree(cached, False) - else: - os.remove(cached) - cachev = None - try: - cachev = open(cachedir + '/.version', 'w+') - cachev.write(KMS_VERSION) - finally: - if cachev is not None: - cachev.close() - - kmsponies = cachedir + '/kmsponies/' + palettefile - kmspony = (kmsponies + pony).replace('//', '/') - - if not os.path.isfile(kmspony): - protokmsponies = cachedir + '/protokmsponies/' - protokmspony = (protokmsponies + pony).replace('//', '/') - - protokmsponydir = protokmspony[:protokmspony.rindex('/')] - kmsponydir = kmspony[: kmspony.rindex('/')] - - _protokmspony = '\'' + protokmspony.replace('\'', '\'\\\'\'') + '\'' - _kmspony = '\'' + kmspony.replace('\'', '\'\\\'\'') + '\'' - _pony = '\'' + pony.replace('\'', '\'\\\'\'') + '\'' - - if not os.path.isfile(protokmspony): - if not os.path.isdir(protokmsponydir): - os.makedirs(protokmsponydir) - if not os.system('ponysay2ttyponysay < ' + _pony + ' > ' + _protokmspony) == 0: - sys.stderr.write('Unable to run ponysay2ttyponysay successfully, you need util-say for KMS support\n') - exit(1) - - if not os.path.isdir(kmsponydir): - os.makedirs(kmsponydir) - if not os.system('tty2colourfultty -p ' + palette + ' < ' + _protokmspony + ' > ' + _kmspony) == 0: - sys.stderr.write('Unable to run tty2colourfultty successfully, you need util-say for KMS support\n') - exit(1) - - return kmspony - - - -ARGUMENTLESS = 0 -ARGUMENTED = 1 -VARIADIC = 2 -''' -Simple argument parser -''' -class ArgParser(): - ''' - Constructor. - The short description is printed on same line as the program name - ''' - def __init__(self, program, description, usage, longdescription = None): - self.__program = program - self.__description = description - self.__usage = usage - self.__longdescription = longdescription - self.__arguments = [] - self.opts = {} - self.optmap = {} - - - ''' - Add option that takes no arguments - ''' - def add_argumentless(self, alternatives, help = None): - ARGUMENTLESS - self.__arguments.append((ARGUMENTLESS, alternatives, None, help)) - stdalt = alternatives[0] - self.opts[stdalt] = None - for alt in alternatives: - self.optmap[alt] = (stdalt, ARGUMENTLESS) - - ''' - Add option that takes one argument - ''' - def add_argumented(self, alternatives, arg, help = None): - self.__arguments.append((ARGUMENTED, alternatives, arg, help)) - stdalt = alternatives[0] - self.opts[stdalt] = None - for alt in alternatives: - self.optmap[alt] = (stdalt, ARGUMENTED) - - ''' - Add option that takes all following argument - ''' - def add_variadic(self, alternatives, arg, help = None): - self.__arguments.append((VARIADIC, alternatives, arg, help)) - stdalt = alternatives[0] - self.opts[stdalt] = None - for alt in alternatives: - self.optmap[alt] = (stdalt, VARIADIC) - - - ''' - Parse arguments - ''' - def parse(self, argv = sys.argv): - self.argcount = len(argv) - 1 - self.files = [] - - argqueue = [] - optqueue = [] - deque = [] - for arg in argv[1:]: - deque.append(arg) - - dashed = False - tmpdashed = False - get = 0 - dontget = 0 - - def unrecognised(arg): - sys.stderr.write('%s: warning: unrecognised option %s\n' % (self.__program, arg)) - - while len(deque) != 0: - arg = deque[0] - deque = deque[1:] - if (get > 0) and (dontget == 0): - get -= 1 - argqueue.append(arg) - elif tmpdashed: - self.files.append(arg) - tmpdashed = False - elif dashed: self.files.append(arg) - elif arg == '++': tmpdashed = True - elif arg == '--': dashed = True - elif (len(arg) > 1) and (arg[0] in ('-', '+')): - if (len(arg) > 2) and (arg[:2] in ('--', '++')): - if dontget > 0: - dontget -= 1 - elif (arg in self.optmap) and (self.optmap[arg][1] == ARGUMENTLESS): - optqueue.append(arg) - argqueue.append(None) - elif '=' in arg: - arg_opt = arg[:arg.index('=')] - if (arg_opt in self.optmap) and (self.optmap[arg_opt][1] >= ARGUMENTED): - optqueue.append(arg_opt) - argqueue.append(arg[arg.index('=') + 1:]) - if self.optmap[arg_opt][1] == VARIADIC: - dashed = True - else: - unrecognised(arg) - elif (arg in self.optmap) and (self.optmap[arg][1] == ARGUMENTED): - optqueue.append(arg) - get += 1 - elif (arg in self.optmap) and (self.optmap[arg][1] == VARIADIC): - optqueue.append(arg) - argqueue.append(None) - dashed = True - else: - unrecognised(arg) - else: - sign = arg[0] - i = 1 - n = len(arg) - while i < n: - narg = sign + arg[i] - i += 1 - if (narg in self.optmap): - if self.optmap[narg][1] == ARGUMENTLESS: - optqueue.append(narg) - argqueue.append(None) - elif self.optmap[narg][1] == ARGUMENTED: - optqueue.append(narg) - nargarg = arg[i:] - if len(nargarg) == 0: - get += 1 - else: - argqueue.append(nargarg) - break - elif self.optmap[narg][1] == VARIADIC: - optqueue.append(narg) - nargarg = arg[i:] - argqueue.append(nargarg if len(nargarg) > 0 else None) - dashed = True - break - else: - unrecognised(arg) - else: - self.files.append(arg) - - i = 0 - n = len(optqueue) - while i < n: - opt = optqueue[i] - arg = argqueue[i] - i += 1 - opt = self.optmap[opt][0] - if (opt not in self.opts) or (self.opts[opt] is None): - self.opts[opt] = [] - self.opts[opt].append(arg) - - for arg in self.__arguments: - if (arg[0] == VARIADIC): - varopt = self.opts[arg[1][0]] - if varopt is not None: - additional = ','.join(self.files).split(',') if len(self.files) > 0 else [] - if varopt[0] is None: - self.opts[arg[1][0]] = additional - else: - self.opts[arg[1][0]] = varopt[0].split(',') + additional - self.files = [] - break - - self.message = ' '.join(self.files) if len(self.files) > 0 else None - - - ''' - Prints a colourful help message - ''' - def help(self): - print('\033[1m%s\033[21m %s %s' % (self.__program, '-' if linuxvt else '—', self.__description)) - print() - if self.__longdescription is not None: - print(self.__longdescription) - print() - - print('\033[1mUSAGE:\033[21m', end='') - first = True - for line in self.__usage.split('\n'): - if first: - first = False - else: - print(' or', end='') - print('\t%s' % (line)) - print() - - print('\033[1mSYNOPSIS:\033[21m') - (lines, lens) = ([], []) - for opt in self.__arguments: - opt_type = opt[0] - opt_alts = opt[1] - opt_arg = opt[2] - opt_help = opt[3] - if opt_help is None: - continue - (line, l) = ('', 0) - first = opt_alts[0] - last = opt_alts[-1] - alts = ('', last) if first is last else (first, last) - for opt_alt in alts: - if opt_alt is alts[-1]: - line += '%colour%' + opt_alt - l += len(opt_alt) - if opt_type == ARGUMENTED: line += ' \033[4m%s\033[24m' % (opt_arg); l += len(opt_arg) + 1 - elif opt_type == VARIADIC: line += ' [\033[4m%s\033[24m...]' % (opt_arg); l += len(opt_arg) + 6 - else: - line += ' \033[2m%s\033[22m ' % (opt_alt) - l += len(opt_alt) + 6 - lines.append(line) - lens.append(l) - - col = max(lens) - col += 8 - ((col - 4) & 7) - index = 0 - for opt in self.__arguments: - opt_help = opt[3] - if opt_help is None: - continue - first = True - colour = '36' if (index & 1) == 0 else '34' - print(lines[index].replace('%colour%', '\033[%s;1m' % (colour)), end=' ' * (col - lens[index])) - for line in opt_help.split('\n'): - if first: - first = False - print('%s' % (line), end='\033[21;39m\n') - else: - print('%s\033[%sm%s\033[39m' % (' ' * col, colour, line)) - index += 1 - - print() - - - -''' -Balloon format class -''' -class Balloon(): - ''' - Constructor - ''' - def __init__(self, link, linkmirror, ww, ee, nw, nnw, n, nne, ne, nee, e, see, se, sse, s, ssw, sw, sww, w, nww): - (self.link, self.linkmirror) = (link, linkmirror) - (self.ww, self.ee) = (ww, ee) - (self.nw, self.ne, self.se, self.sw) = (nw, ne, se, sw) - (self.nnw, self.n, self.nne) = (nnw, n, nne) - (self.nee, self.e, self.see) = (nee, e, see) - (self.sse, self.s, self.ssw) = (sse, s, ssw) - (self.sww, self.w, self.nww) = (sww, w, nww) - - _ne = max(ne, key = UCS.dispLen) - _nw = max(nw, key = UCS.dispLen) - _se = max(se, key = UCS.dispLen) - _sw = max(sw, key = UCS.dispLen) - - minE = UCS.dispLen(max([_ne, nee, e, see, _se, ee], key = UCS.dispLen)) - minW = UCS.dispLen(max([_nw, nww, e, sww, _sw, ww], key = UCS.dispLen)) - minN = len(max([ne, nne, n, nnw, nw], key = len)) - minS = len(max([se, sse, s, ssw, sw], key = len)) - - self.minwidth = minE + minE - self.minheight = minN + minS - - - ''' - Generates a balloon with a message - ''' - def get(self, minw, minh, lines, lencalc): - h = self.minheight + len(lines) - w = self.minwidth + lencalc(max(lines, key = lencalc)) - if w < minw: w = minw - if h < minh: h = minh - - if len(lines) > 1: - (ws, es) = ({0 : self.nww, len(lines) - 1 : self.sww}, {0 : self.nee, len(lines) - 1 : self.see}) - for j in range(1, len(lines) - 1): - ws[j] = self.w - es[j] = self.e - else: - (ws, es) = ({0 : self.ww}, {0 : self.ee}) - - rc = [] - - for j in range(0, len(self.n)): - outer = UCS.dispLen(self.nw[j]) + UCS.dispLen(self.ne[j]) - inner = UCS.dispLen(self.nnw[j]) + UCS.dispLen(self.nne[j]) - if outer + inner >= w: - rc.append(self.nw[j] + self.nnw[j] + self.n[j] * (w - outer - inner) + self.nne[j] + self.ne[j]) - else: - rc.append(self.nw[j] + self.n[j] * (w - outer) + self.ne[j]) - - for j in range(0, len(lines)): - rc.append(ws[j] + lines[j] + ' ' * (w - lencalc(lines[j]) - UCS.dispLen(self.w) - UCS.dispLen(self.e)) + es[j]) - - for j in range(0, len(self.s)): - outer = UCS.dispLen(self.sw[j]) + UCS.dispLen(self.se[j]) - inner = UCS.dispLen(self.ssw[j]) + UCS.dispLen(self.sse[j]) - if outer + inner >= w: - rc.append(self.sw[j] + self.ssw[j] + self.s[j] * (w - outer - inner) + self.sse[j] + self.se[j]) - else: - rc.append(self.sw[j] + self.s[j] * (w - outer) + self.se[j]) - - return '\n'.join(rc) - - - -''' -Replacement for cowsay -''' -class Backend(): - ''' - Constructor - Takes message [string], ponyfile [filename string], wrapcolumn [None or an int], width [None or an int] and balloon [Balloon object] - ''' - def __init__(self, message, ponyfile, wrapcolumn, width, balloon): - self.message = message - self.ponyfile = ponyfile - self.wrapcolumn = wrapcolumn - self.width = width - self.balloon = balloon - - self.link = {'\\' : self.balloon.link, '/' : self.balloon.linkmirror} - - self.output = '' - self.pony = None - - - ''' - Process all data - ''' - def parse(self): - self.__expandMessage() - self.__loadFile() - self.__processPony() - self.__truncate() - - - ''' - Converts all tabs in the message to spaces by expanding - ''' - def __expandMessage(self): - lines = self.message.split('\n') - buf = '' - for line in lines: - (i, n, x) = (0, len(line), 0) - while i < n: - c = line[i] - i += 1 - if c == '\033': - colour = self.__getcolour(line, i - 1) - i += len(colour) - 1 - buf += colour - elif c == '\t': - nx = 8 - (x & 7) - buf += ' ' * nx - x += nx - else: - buf += c - if not UCS.isCombining(c): - x += 1 - buf += '\n' - self.message = buf[:-1] - - - ''' - Loads the pony file - ''' - def __loadFile(self): - ponystream = None - try: - ponystream = open(self.ponyfile, 'rb') - self.pony = ponystream.read().decode('utf8', 'replace') - finally: - if ponystream is not None: - ponystream.close() - - - ''' - Truncate output to the width of the screen - ''' - def __truncate(self): - if self.width is None: - return - lines = self.output.split('\n') - self.output = '' - for line in lines: - (i, n, x) = (0, len(line), 0) - while i < n: - c = line[i] - i += 1 - if c == '\033': - colour = self.__getcolour(line, i - 1) - i += len(colour) - 1 - self.output += colour - else: - if x < self.width: - self.output += c - if not UCS.isCombining(c): - x += 1 - self.output += '\n' - self.output = self.output[:-1] - - - ''' - Process the pony file and generate output to self.output - ''' - def __processPony(self): - self.output = '' - - AUTO_PUSH = '\033[01010~' - AUTO_POP = '\033[10101~' - - variables = {'' : '$'} - for key in self.link: - variables[key] = AUTO_PUSH + self.link[key] + AUTO_POP - - indent = 0 - dollar = None - balloonLines = None - colourstack = ColourStack(AUTO_PUSH, AUTO_POP) - - (i, n, lineindex, skip, nonskip) = (0, len(self.pony), 0, 0, 0) - while i < n: - c = self.pony[i] - if c == '\t': - n += 8 - (indent & 7) - ed = ' ' * (7 - (indent & 7)) - c = ' ' - self.pony = self.pony[:i] + ed + self.pony[i:] - i += 1 - if c == '$': - if dollar is not None: - if '=' in dollar: - name = dollar[:find('=')] - value = dollar[find('=') + 1:] - variables[name] = value - elif (len(dollar) < 7) or not (dollar[:7] == 'balloon'): - data = variables[dollar].replace('$', '$$') - if data == '$$': # if not handled specially we will get an infinity loop - if (skip == 0) or (nonskip > 0): - if nonskip > 0: - nonskip -= 1 - self.output += '$' - indent += 1 - else: - skip -= 1 - else: - n += len(data) - self.pony = self.pony[:i] + data + self.pony[i:] - else: - (w, h) = (0, 0) - props = dollar[7:] - if len(props) > 0: - if ',' in props: - if props[0] is not ',': - w = int(props[:props.index(',')]) - h = int(props[props.index(',') + 1:]) - else: - w = int(props) - balloon = self.__getballoon(w, h, indent) - balloon = balloon.split('\n') - for b in balloon[0]: - self.output += b + colourstack.feed(b) - if lineindex == 0: - balloonpre = '\n' + (' ' * indent) - for line in balloon[1:]: - self.output += balloonpre; - for b in line: - self.output += b + colourstack.feed(b); - indent = 0 - elif len(balloon) > 1: - balloonLines = balloon - balloonLine = 0 - balloonIndent = indent - indent += self.__len(balloonLines[0]) - balloonLines[0] = None - dollar = None - else: - dollar = '' - elif dollar is not None: - if c == '\033': - c = self.pony[i] - i += 1 - dollar += c - elif c == '\033': - colour = self.__getcolour(self.pony, i - 1) - for b in colour: - self.output += b + colourstack.feed(b); - i += len(colour) - 1 - elif c == '\n': - self.output += c - indent = 0 - (skip, nonskip) = (0, 0) - lineindex += 1 - if balloonLines is not None: - balloonLine += 1 - if balloonLine == len(balloonLines): - balloonLines = None - else: - if (balloonLines is not None) and (balloonLines[balloonLine] is not None) and (balloonIndent == indent): - data = balloonLines[balloonLine] - datalen = self.__len(data) - skip += datalen - nonskip += datalen - data = data.replace('$', '$$') - n += len(data) - self.pony = self.pony[:i] + data + self.pony[i:] - balloonLines[balloonLine] = None - else: - if (skip == 0) or (nonskip > 0): - if nonskip > 0: - nonskip -= 1 - self.output += c + colourstack.feed(c); - if not UCS.isCombining(c): - indent += 1 - else: - skip -= 1 - - if balloonLines is not None: - for line in balloonLines[balloonLine:]: - data = ' ' * (balloonIndent - indent) + line + '\n' - for b in data: - self.output += b + colourstack.feed(b); - indent = 0 - - self.output = self.output.replace(AUTO_PUSH, '').replace(AUTO_POP, '') - - - ''' - Gets colour code att the currect offset in a buffer - ''' - def __getcolour(self, input, offset): - (i, n) = (offset, len(input)) - rc = input[i] - i += 1 - if i == n: return rc - c = input[i] - i += 1 - rc += c - - if c == ']': - if i == n: return rc - c = input[i] - i += 1 - rc += c - if c == 'P': - di = 0 - while (di < 7) and (i < n): - c = input[i] - i += 1 - di += 1 - rc += c - elif c == '[': - while i < n: - c = input[i] - i += 1 - rc += c - if (c == '~') or (('a' <= c) and (c <= 'z')) or (('A' <= c) and (c <= 'Z')): - break - - return rc - - - ''' - Calculates the number of visible characters in a text - ''' - def __len(self, input): - (rc, i, n) = (0, 0, len(input)) - while i < n: - c = input[i] - if c == '\033': - i += len(self.__getcolour(input, i)) - else: - i += 1 - if not UCS.isCombining(c): - rc += 1 - return rc - - - ''' - Generates a balloon with the message - ''' - def __getballoon(self, width, height, left): - wrap = None - if self.wrapcolumn is not None: - wrap = self.wrapcolumn - left - if wrap < 8: - wrap = 8 - - msg = self.message - if wrap is not None: - msg = self.__wrapMessage(msg, wrap) - - if '\033' in msg: - AUTO_PUSH = '\033[01010~' - AUTO_POP = '\033[10101~' - cstack = ColourStack(AUTO_PUSH, AUTO_POP) - buf = '' - for c in msg: - if c == '\n': - for cc in ('%s\n%s' % (AUTO_PUSH, AUTO_POP)): - buf += cc - buf += cstack.feed(cc) - else: - buf += c - buf += cstack.feed(c) - msg = buf - - return self.balloon.get(width, height, msg.split('\n'), self.__len) - - - ''' - Wraps the message - ''' - def __wrapMessage(self, message, wrap): - lines = message.split('\n') - buf = '' - for line in lines: - b = [None] * len(line) - map = {} - (bi, cols, w) = (0, 0, wrap) - (indent, indentc) = (-1, 0) - - (i, n) = (0, len(line)) - while i <= n: - d = None - if i != n: - d = line[i] - i += 1 - if d == '\033': # TODO this should use self.__getcolour() - b[bi] = d - bi += 1 - b[bi] = line[i] - d = line[i] - bi += 1 - i += 1 - if d == '[': - while True: - b[bi] = line[i] - d = line[i] - bi += 1 - i += 1 - if (('a' <= d) and (d <= 'z')) or (('A' <= d) and (d <= 'Z')) or (d == '~'): - break - elif d == ']': - b[bi] = line[i] - d = line[i] - bi += 1 - i += 1 - if d == 'P': - for j in range(0, 7): - b[bi] = line[i] - bi += 1 - i += 1 - elif (d is not None) and (d != ' '): - if indent == -1: - indent = i - 1 - for j in range(0, indent): - if line[j] == ' ': - indentc += 1 - b[bi] = d - bi += 1 - if not UCS.isCombining(d): - cols += 1 - map[cols] = bi - else: - mm = 0 - while (w > 8) and (cols > w + 3): - mm += w - 1 - m = map[mm] - for bb in b[:m]: - buf += bb - buf += '-\n' - cols -= w - 1 - m += w -1 - bi -= m - bb = b[m:] - for j in range(0, bi): - b[j] = bb[j] - w = wrap - if indent != -1: - buf += line[:indent] - w -= indentc - if cols > w: - buf += '\n' - w = wrap - if indent != -1: - buf += line[:indent] - w -= indentc - for bb in b[:bi]: - buf += bb - w -= cols - cols = 0 - bi = 0 - if d == -1: - i += 1 - else: - if w > 0: - buf += ' ' - w -= 1 - else: - buf += '\n' - w = wrap - if indent != -1: - buf + line[:indent] - w -= indentc - - buf += '\n' - return '\n'.join(line.rstrip() for line in buf[:-1].split('\n')) - - -''' -ANSI colour stack -''' -class ColourStack(): - ''' - Constructor - ''' - def __init__(self, autopush, autopop): - self.autopush = autopush - self.autopop = autopop - self.lenpush = len(autopush) - self.lenpop = len(autopop) - self.bufproto = ' ' * (self.lenpush if self.lenpush > self.lenpop else self.lenpop) - self.stack = [] - self.push() - self.seq = None - - - def push(self): - self.stack = [[self.bufproto, None, None, [False] * 9]] + self.stack - if len(self.stack) == 1: - return None - return '\033[0m' - - - def pop(self): - old = self.stack[0] - self.stack = self.stack[1:] - rc = '\033[0;' - new = self.stack[0] - if new[1] is not None: rc += new[1] + ';' - if new[2] is not None: rc += new[2] + ';' - for i in range(0, 9): - if new[3][i]: - rc += str(i + 1) + ';' - return rc[:-1] + 'm' - - - def feed(self, char): - if self.seq is not None: - self.seq += char - if (char == '~') or (('a' <= char) and (char <= 'z')) or (('A' <= char) and (char <= 'Z')): - if (self.seq[0] == '[') and (self.seq[-1] == 'm'): - self.seq = self.seq[1:-1].split(';') - (i, n) = (0, len(self.seq)) - while i < n: - part = self.seq[i] - p = 0 if part == '' else int(part) - i += 1 - if p == 0: self.stack[0][1:] = [None, None, [False] * 9] - elif (1 <= p) and (p <= 9): self.stack[0][3][p - 1] = True - elif (21 <= p) and (p <= 29): self.stack[0][3][p - 21] = False - elif p == 39: self.stack[0][1] = None - elif p == 49: self.stack[0][2] = None - elif (30 <= p) and (p <= 37): self.stack[0][1] = part - elif (90 <= p) and (p <= 97): self.stack[0][1] = part - elif (40 <= p) and (p <= 47): self.stack[0][2] = part - elif (100 <= p) and (p <= 107): self.stack[0][2] = part - elif p == 38: - self.stack[0][1] = '%s;%s;%s' % (part, self.seq[i], self.seq[i + 1]) - i += 2 - elif p == 48: - self.stack[0][2] = '%s;%s;%s' % (part, self.seq[i], self.seq[i + 1]) - i += 2 - self.seq = None - elif char == '\033': - self.seq = '' - buf = self.stack[0][0] - buf = buf[1:] + char - rc = '' - if buf[-self.lenpush:] == self.autopush: rc = self.push() - elif buf[-self.lenpop:] == self.autopop: rc = self.pop() - self.stack[0][0] = buf - return rc - - - -''' -UCS utility class -''' -class UCS(): - ''' - Checks whether a character is a combining character - ''' - @staticmethod - def isCombining(char): - o = ord(char) - if (0x0300 <= o) and (o <= 0x036F): return True - if (0x20D0 <= o) and (o <= 0x20FF): return True - if (0x1DC0 <= o) and (o <= 0x1DFF): return True - if (0xFE20 <= o) and (o <= 0xFE2F): return True - return False - - - ''' - Gets the number of combining characters in a string - ''' - @staticmethod - def countCombining(string): - rc = 0 - for char in string: - if UCS.isCombining(char): - rc += 1 - return rc - - - ''' - Gets length of a string not counting combining characters - ''' - @staticmethod - def dispLen(string): - return len(string) - UCS.countCombining(string) - - - -''' -The user's home directory -''' -HOME = os.environ['HOME'] if 'HOME' in os.environ else os.path.expanduser('~') - - -''' -Whether the program is execute in Linux VT (TTY) -''' -linuxvt = ('TERM' in os.environ) and (os.environ['TERM'] == 'linux') - - -''' -Whether the script is executed as ponythink -''' -isthink = (len(__file__) >= 5) and (__file__[-5:] == 'think') -isthink = ((len(__file__) >= 8) and (__file__[-8:] == 'think.py')) or isthink - - -''' -Whether stdin is piped -''' -pipelinein = not sys.stdin.isatty() - -''' -Whether stdout is piped -''' -pipelineout = not sys.stdout.isatty() - -''' -Whether stderr is piped -''' -pipelineerr = not sys.stderr.isatty() - - -''' -Whether KMS is used -''' -usekms = Ponysay.isUsingKMS() - - -''' -The directories where pony files are stored, ttyponies/ are used if the terminal is Linux VT (also known as TTY) and not with KMS -''' -appendset = set() -ponydirs = [] -if linuxvt and not usekms: _ponydirs = [HOME + '/.local/share/ponysay/ttyponies/', '/usr/share/ponysay/ttyponies/'] -else: _ponydirs = [HOME + '/.local/share/ponysay/ponies/', '/usr/share/ponysay/ponies/'] -for ponydir in _ponydirs: - if os.path.isdir(ponydir) and (ponydir not in appendset): - ponydirs.append(ponydir) - appendset.add(ponydir) - - -''' -The directories where pony files are stored, extrattyponies/ are used if the terminal is Linux VT (also known as TTY) and not with KMS -''' -appendsetset = set() -extraponydirs = [] -if linuxvt and not usekms: _extraponydirs = [HOME + '/.local/share/ponysay/extrattyponies/', '/usr/share/ponysay/extrattyponies/'] -else: _extraponydirs = [HOME + '/.local/share/ponysay/extraponies/', '/usr/share/ponysay/extraponies/'] -for extraponydir in _extraponydirs: - if os.path.isdir(extraponydir) and (extraponydir not in appendset): - extraponydirs.append(extraponydir) - appendset.add(extraponydir) - - -''' -The directories where quotes files are stored -''' -appendset = set() -quotedirs = [] -_quotedirs = [HOME + '/.local/share/ponysay/quotes/', '/usr/share/ponysay/quotes/'] -for quotedir in _quotedirs: - if os.path.isdir(quotedir) and (quotedir not in appendset): - quotedirs.append(quotedir) - appendset.add(quotedir) - - -''' -The directories where balloon style files are stored -''' -appendset = set() -balloondirs = [] -_balloondirs = [HOME + '/.local/share/ponysay/balloons/', '/usr/share/ponysay/balloons/'] -for balloondir in _balloondirs: - if os.path.isdir(balloondir) and (balloondir not in appendset): - balloondirs.append(balloondir) - appendset.add(balloondir) - - -''' -ucsmap files -''' -appendset = set() -ucsmaps = [] -_ucsmaps = [HOME + '/.local/share/ponysay/ucsmap', '/usr/share/ponysay/ucsmap'] -for ucsmap in _ucsmaps: - if os.path.isdir(ucsmap) and (ucsmap not in appendset): - ucsmaps.append(ucsmap) - appendset.add(ucsmap) - - - -usage_saythink = '\033[34;1m(ponysay | ponythink)\033[21;39m' -usage_common = '[-c] [-W\033[4mCOLUMN\033[24m] [-b\033[4mSTYLE\033[24m]' -usage_listhelp = '(-l | -L | -B | +l | +L | -v | -h)' -usage_file = '[-f\033[4mPONY\033[24m]* [[--] \033[4mmessage\033[24m]' -usage_xfile = '[-F\033[4mPONY\033[24m]* [[--] \033[4mmessage\033[24m]' -usage_quote = '-q [\033[4mPONY\033[24m*]' - -usage = '%s %s\n%s %s %s\n%s %s %s\n%s %s %s' % (usage_saythink, usage_listhelp, - usage_saythink, usage_common, usage_file, - usage_saythink, usage_common, usage_xfile, - usage_saythink, usage_common, usage_quote) - -usage = usage.replace('\033[', '\0') -for sym in ('[', ']', '(', ')', '|', '...', '*'): - usage = usage.replace(sym, '\033[2m' + sym + '\033[22m') -usage = usage.replace('\0', '\033[') - -''' -Argument parsing -''' -opts = ArgParser(program = 'ponythink' if isthink else 'ponysay', - description = 'cowsay reimplemention for ponies', - usage = usage, - longdescription = -'''Ponysay displays an image of a pony saying some text provided by the user. -If \033[4mmessage\033[24m is not provided, it accepts standard input. For an extensive -documentation run `info ponysay`, or for just a little more help than this -run `man ponysay`. Ponysay has so much more to offer than described here.''') - -opts.add_argumentless(['--quoters']) -opts.add_argumentless(['--onelist']) -opts.add_argumentless(['++onelist']) - -opts.add_argumentless(['-h', '--help'], help = 'Print this help message.') -opts.add_argumentless(['-v', '--version'], help = 'Print the version of the program.') -opts.add_argumentless(['-l', '--list'], help = 'List pony names.') -opts.add_argumentless(['-L', '--symlist', '--altlist'], help = 'List pony names with alternatives.') -opts.add_argumentless(['+l', '++list'], help = 'List non-MLP:FiM pony names.') -opts.add_argumentless(['+L', '++symlist', '++altlist'], help = 'List non-MLP:FiM pony names with alternatives.') -opts.add_argumentless(['-A', '--all'], help = 'List MLP:FIM and non-MLP:FiM pony names.') -opts.add_argumentless(['+A', '++all'], help = 'List MLP:FIM and non-MLP:FiM pony names with alternatives.') -opts.add_argumentless(['-B', '--bubblelist', '--balloonlist'], help = 'List balloon styles.') -opts.add_argumentless(['-c', '--compact'], help = 'Compress messages.') -opts.add_argumented( ['-W', '--wrap'], arg = 'COLUMN', help = 'Specify column where the message should be wrapped.') -opts.add_argumented( ['-b', '--bubble', '--balloon'], arg = 'STYLE', help = 'Select a balloon style.') -opts.add_argumented( ['-f', '--file', '--pony'], arg = 'PONY', help = 'Select a pony.\nEither a file name or a pony name.') -opts.add_argumented( ['-F', '++file', '++pony'], arg = 'PONY', help = 'Select a non-MLP:FiM pony.') -opts.add_variadic( ['-q', '--quote'], arg = 'PONY', help = 'Select a ponies which will quote themself.') - -opts.parse() - - - -''' -Start the program from ponysay.__init__ if this is the executed file -''' -if __name__ == '__main__': - Ponysay(opts) diff --git a/ponysay.py b/ponysay.py new file mode 100755 index 0000000..d1571a7 --- /dev/null +++ b/ponysay.py @@ -0,0 +1,1779 @@ +#!/usr/bin/env python3 +# -*- coding: utf-8 -*- + +''' +ponysay.py - Ponysay, cowsay reimplementation for ponies +Copyright (C) 2012 Erkin Batu Altunbaş et al. + +This program is free software. It comes without any warranty, to +the extent permitted by applicable law. You can redistribute it +and/or modify it under the terms of the Do What The Fuck You Want +To Public License, Version 2, as published by Sam Hocevar. See +http://sam.zoy.org/wtfpl/COPYING for more details. + + +Authors of ponysay.py: + + Erkin Batu Altunbaş: Project leader, helped write the first implementation + Mattias "maandree" Andrée: Major contributor of both implementions + Elis "etu" Axelsson: Major contributor of current implemention and patcher of the first implementation + Sven-Hendrik "svenstaro" Haase: Major contributor of the first implementation + Jan Alexander "heftig" Steffens: Major contributor of the first implementation + Kyah "L-four" Rindlisbacher: Patched the first implementation + +''' + +import os +import shutil +import sys +import random +from subprocess import Popen, PIPE + + +''' +The version of ponysay +''' +VERSION = 'dev' # this line should not be edited, it is fixed by the build system + + + +''' +Hack to enforce UTF-8 in output (in the future, if you see anypony not using utf-8 in programs by default, report them to Princess Celestia so she can banish them to the moon) +''' +def print(text = '', end = '\n'): + sys.stdout.buffer.write((str(text) + end).encode('utf-8')) + + +''' +This is the mane class of ponysay +''' +class Ponysay(): + ''' + Starts the part of the program the arguments indicate + ''' + def __init__(self, args): + if (args.argcount == 0) and not pipelinein: + args.help() + return + + + if args.opts['-h'] is not None: args.help() + elif args.opts['--quoters'] is not None: self.quoters() + elif args.opts['--onelist'] is not None: self.onelist() + elif args.opts['-v'] is not None: self.version() + elif args.opts['-l'] is not None: self.list() + elif args.opts['-L'] is not None: self.linklist() + elif args.opts['-B'] is not None: self.balloonlist() + elif args.opts['++onelist'] is not None: self.__extraponies(); self.onelist() + elif args.opts['+l'] is not None: self.__extraponies(); self.list() + elif args.opts['+L'] is not None: self.__extraponies(); self.linklist() + elif args.opts['-A'] is not None: self.list(); self.__extraponies(); self.list() + elif args.opts['+A'] is not None: self.linklist(); self.__extraponies(); self.linklist() + else: + self.__extraponies(args) + self.__bestpony(args) + self.__ucsremap(args) + if args.opts['-q'] is not None: self.quote(args) + else: self.print_pony(args) + + + ## + ## Methods that run before the mane methods + ## + + ''' + Use extra ponies + ''' + def __extraponies(self, args = None): + if args is None: + ponydirs[:] = extraponydirs + elif args.opts['-F'] is not None: + args.opts['-f'] = args.opts['-F'] + ponydirs[:] = extraponydirs + + + ''' + Use best.pony if nothing else is set + ''' + def __bestpony(self, args): + if (args.opts['-f'] is None) or (args.opts['-q'] is None) or (len(args.opts['-q']) == 0): + for ponydir in ponydirs: + if os.path.isfile(ponydir + 'best.pony') or os.path.islink(ponydir + 'best.pony'): + pony = os.path.realpath(ponydir + 'best.pony') # Canonical path + args.opts['-f' if args.opts['-q'] is None else '-q'] = [pony] + break + + + ''' + Apply pony name remapping to args according to UCS settings + ''' + def __ucsremap(self, args): + env_ucs = os.environ['PONYSAY_UCS_ME'] if 'PONYSAY_UCS_ME' in os.environ else '' + ucs_conf = 0 + if env_ucs in ('yes', 'y', '1'): ucs_conf = 1 + elif env_ucs in ('harder', 'h', '2'): ucs_conf = 2 + + if ucs_conf == 0: + return + + maplines = [] + for ucsmap in ucsmaps: + if os.path.isfile(ucsmap): + mapfile = None + try: + mapfile = open(ucsmap, 'rb') + maplines += [line.replace('\n', '') for line in mapfile.read().decode('utf8', 'replace').split('\n')] + finally: + if mapfile is not None: + mapfile.close() + + map = {} + stripset = ' \t' # must be string, wtf! and way doesn't python's doc say so + for line in maplines: + if (len(line) > 0) and not (line[0] == '#'): + s = line.index('→') + ucs = line[:s] .strip(stripset) + ascii = line[s + 1:].strip(stripset) + map[ucs] = ascii + + for flag in ('-f', '-q'): + if args.opts[flag] is not None: + for i in range(0, len(args.opts[flag])): + if args.opts[flag][i] in map: + args.opts[flag][i] = map[args.opts[flag][i]] + + + ## + ## Auxiliary methods + ## + + ''' + Apply USC:ise pony names according to UCS settings + ''' + def __ucsise(self, ponies, links = None): + env_ucs = os.environ['PONYSAY_UCS_ME'] if 'PONYSAY_UCS_ME' in os.environ else '' + ucs_conf = 0 + if env_ucs in ('yes', 'y', '1'): ucs_conf = 1 + elif env_ucs in ('harder', 'h', '2'): ucs_conf = 2 + + if ucs_conf == 0: + return + + maplines = [] + for ucsmap in ucsmaps: + if os.path.isfile(ucsmap): + mapfile = None + try: + mapfile = open(ucsmap, 'rb') + maplines += [line.replace('\n', '') for line in mapfile.read().decode('utf8', 'replace').split('\n')] + finally: + if mapfile is not None: + mapfile.close() + + map = {} + stripset = ' \t' # must be string, wtf! and way doesn't python's doc say so + for line in maplines: + if (len(line) > 0) and not (line[0] == '#'): + s = line.index('→') + ucs = line[:s] .strip(stripset) + ascii = line[s + 1:].strip(stripset) + map[ascii] = ucs + + if ucs_conf == 1: + for pony in ponies: + if pony in map: + ponies.append(map[pony]) + if links is not None: + links[map[pony]] = pony + else: + for j in range(0, len(ponies)): + if ponies[j] in map: + ponies[j] = map[ponies[j]] + + + ''' + Returns one file with full path, names is filter for names, also accepts filepaths + ''' + def __getponypath(self, names = None): + ponies = {} + + for ponydir in ponydirs: + for ponyfile in os.listdir(ponydir): + pony = ponyfile[:-5] + if pony not in ponies: + ponies[pony] = ponydir + ponyfile + + if not names == None: + for name in names: + if os.path.exists(name): + ponies[name] = name + + if names == None: + names = list(ponies.keys()) + + pony = names[random.randrange(0, len(names))] + if pony not in ponies: + sys.stderr.write('I have never heard of anypony named %s\n' % (pony)); + exit(1) + else: + return ponies[pony] + + + ''' + Returns a set with all ponies that have quotes and are displayable + ''' + def __quoters(self): + quotes = [] + quoteshash = set() + _quotes = [] + for quotedir in quotedirs: + _quotes += [item[:item.index('.')] for item in os.listdir(quotedir)] + for quote in _quotes: + if not quote == '': + if not quote in quoteshash: + quoteshash.add(quote) + quotes.append(quote) + + ponies = set() + for ponydir in ponydirs: + for pony in os.listdir(ponydir): + if not pony[0] == '.': + p = pony[:-5] # remove .pony + for quote in quotes: + if ('+' + p + '+') in ('+' + quote + '+'): + if not p in ponies: + ponies.add(p) + + return ponies + + + ''' + Returns a list with all (pony, quote file) pairs + ''' + def __quotes(self): + quotes = [] + for quotedir in quotedirs: + quotes += [quotedir + item for item in os.listdir(quotedir)] + rc = [] + + for ponydir in ponydirs: + for pony in os.listdir(ponydir): + if not pony[0] == '.': + p = pony[:-5] # remove .pony + for quote in quotes: + q = quote[quote.rindex('/') + 1:] + q = q[:q.rindex('.')] + if ('+' + p + '+') in ('+' + q + '+'): + rc.append((p, quote)) + + return rc + + + ''' + Gets the size of the terminal in (rows, columns) + ''' + def __gettermsize(self): + termsize = Popen(['stty', 'size'], stdout=PIPE, stdin=sys.stderr).communicate()[0] + termsize = termsize.decode('utf8', 'replace')[:-1].split(' ') # [:-1] removes a \n + termsize = [int(item) for item in termsize] + return termsize + + + + ## + ## Listing methods + ## + + ''' + Columnise a list and prints it + ''' + def __columnise(self, ponies): + termwidth = self.__gettermsize()[1] + 2 + ponies.sort(key = lambda pony : pony[0]) + widths = [UCS.dispLen(pony[0]) for pony in ponies] + width = max(widths) + 2 # longest pony file name + space between columns + + cols = termwidth // width + rows = (len(ponies) + cols - 1) // cols + columns = [] + for c in range(0, cols): columns.append([]) + + (y, x) = (0, 0) + for j in range(0, len(ponies)): + cell = ponies[j][1] + ' ' * (width - widths[j]); + columns[x].append(cell) + y += 1 + if y == rows: + x += 1 + y = 0 + + diff = rows * cols - len(ponies) + if diff > 2: + c = cols - 1 + diff -= 1 + while diff > 0: + columns[c] = columns[c - 1][-diff:] + columns[c] + c -= 1 + columns[c] = columns[c][:-diff] + diff -= 1 + pass + + lines = [] + for r in range(0, rows): + lines.append([]) + for c in range(0, cols): + if r < len(columns[c]): + line = lines[r].append(columns[c][r]) + + print('\n'.join([''.join(line)[:-2] for line in lines])); + print() + + + ''' + Lists the available ponies + ''' + def list(self): + quoters = self.__quoters() + + for ponydir in ponydirs: # Loop ponydirs + _ponies = os.listdir(ponydir) + ponies = [] + for pony in _ponies: + if (len(pony) > 5) and (pony[-5:] == '.pony'): + ponies.append(pony[:-5]) + self.__ucsise(ponies) + + if len(ponies) == 0: + continue + print('\033[1mponies located in ' + ponydir + '\033[21m') + + self.__columnise([(pony, '\033[1m' + pony + '\033[21m' if pony in quoters else pony) for pony in ponies]) + + + ''' + Lists the available ponies with alternatives inside brackets + ''' + def linklist(self): + termsize = self.__gettermsize() + quoters = self.__quoters() + + for ponydir in ponydirs: # Loop ponydirs + _ponies = os.listdir(ponydir) + ponies = [] + for pony in _ponies: + if (len(pony) > 5) and (pony[-5:] == '.pony'): + ponies.append(pony[:-5]) + + if len(ponies) == 0: + continue + print('\033[1mponies located in ' + ponydir + '\033[21m') + + pseudolinkmap = {} + self.__ucsise(ponies, pseudolinkmap) + pairs = [] + for pony in ponies: + if pony in pseudolinkmap: + pairs.append((pony, pseudolinkmap[pony] + '.pony')); + else: + pairs.append((pony, os.path.realpath(ponydir + pony + '.pony') if os.path.islink(ponydir + pony + '.pony') else None)) + + ponymap = {} + for pair in pairs: + if (pair[1] is None) or (pair[1] == ''): + if pair[0] not in ponymap: + ponymap[pair[0]] = [] + else: + target = pair[1][:-5] + if '/' in target: + target = target[target.rindex('/') + 1:] + if target in ponymap: + ponymap[target].append(pair[0]) + else: + ponymap[target] = [pair[0]] + + width = 0 + ponies = {} + for pony in ponymap: + w = UCS.dispLen(pony) + item = '\033[1m' + pony + '\033[21m' if (pony in quoters) else pony + syms = ponymap[pony] + syms.sort() + if len(syms) > 0: + w += 2 + len(syms) + item += ' (' + first = True + for sym in syms: + w += UCS.dispLen(sym) + if first: first = False + else: item += ' ' + item += '\033[1m' + sym + '\033[21m' if (sym in quoters) else sym + item += ')' + ponies[(item.replace('\033[1m', '').replace('\033[21m', ''), item)] = w + if width < w: + width = w + + self.__columnise(list(ponies)) + + + ''' + Lists with all ponies that have quotes and are displayable + ''' + def quoters(self): + last = '' + ponies = [] + for pony in self.__quoters(): + ponies.append(pony) + self.__ucsise(ponies) + ponies.sort() + for pony in ponies: + if not pony == last: + last = pony + print(pony) + + + ''' + Lists the available ponies one one column without anything bold + ''' + def onelist(self): + last = '' + _ponies = [] + for ponydir in ponydirs: # Loop ponydirs + _ponies += os.listdir(ponydir) + ponies = [] + for pony in _ponies: + if (len(pony) > 5) and (pony[-5:] == '.pony'): + ponies.append(pony[:-5]) + self.__ucsise(ponies) + ponies.sort() + for pony in ponies: + if not pony == last: + last = pony + print(pony) + + + + ## + ## Balloon methods + ## + + ''' + Prints a list of all balloons + ''' + def balloonlist(self): + termsize = self.__gettermsize() + balloonset = set() + + for balloondir in balloondirs: + for balloon in os.listdir(balloondir): + if isthink and (len(balloon) > 6) and (balloon[-6:] == '.think'): + balloon = balloon[:-6] + elif (not isthink) and (len(balloon) > 4) and (balloon[-4:] == '.say'): + balloon = balloon[:-4] + else: + continue + if balloon not in balloonset: + balloonset.add(balloon) + + self.__columnise([(balloon, balloon) for balloon in list(balloonset)]) + + + ''' + Returns one file with full path, names is filter for style names, also accepts filepaths + ''' + def __getballoonpath(self, names): + if names is None: + return None + balloons = {} + + for balloondir in balloondirs: + for balloon in os.listdir(balloondir): + balloonfile = balloon + if isthink and (len(balloon) > 6) and (balloon[-6:] == '.think'): + balloon = balloon[:-6] + elif (not isthink) and (len(balloon) > 4) and (balloon[-4:] == '.say'): + balloon = balloon[:-4] + else: + continue + if balloon not in balloons: + balloons[balloon] = balloondir + balloonfile + + for name in names: + if os.path.exists(name): + balloons[name] = name + + if names == None: + names = list(balloons.keys()) + + balloon = names[random.randrange(0, len(names))] + if balloon not in balloons: + sys.stderr.write('That balloon style %s does not exist\n' % (balloon)); + exit(1) + else: + return balloons[balloon] + + + ''' + Creates the balloon style object + ''' + def __getballoon(self, balloonfile): + if balloonfile is None: + if isthink: + return Balloon('o', 'o', '( ', ' )', [' _'], ['_'], ['_'], ['_'], ['_ '], ' )', ' )', ' )', ['- '], ['-'], ['-'], ['-'], [' -'], '( ', '( ', '( ') + return Balloon('\\', '/', '< ', ' >', [' _'], ['_'], ['_'], ['_'], ['_ '], ' \\', ' |', ' /', ['- '], ['-'], ['-'], ['-'], [' -'], '\\ ', '| ', '/ ') + + map = {} + for elem in ('\\', '/', 'ww', 'ee', 'nw', 'nnw', 'n', 'nne', 'ne', 'nee', 'e', 'see', 'se', 'sse', 's', 'ssw', 'sw', 'sww', 'w', 'nww'): + map[elem] = [] + + balloonstream = None + try: + balloonstream = open(balloonfile, 'rb') + data = [line.replace('\n', '') for line in balloonstream.read().decode('utf8', 'replace').split('\n')] + finally: + if balloonstream is not None: + balloonstream.close() + + last = None + for line in data: + if len(line) > 0: + if line[0] == ':': + map[last].append(line[1:]) + else: + last = line[:line.index(':')] + value = line[len(last) + 1:] + map[last].append(value) + + return Balloon(map['\\'][0], map['/'][0], map['ww'][0], map['ee'][0], map['nw'], map['nnw'], map['n'], + map['nne'], map['ne'], map['nee'][0], map['e'][0], map['see'][0], map['se'], map['sse'], + map['s'], map['ssw'], map['sw'], map['sww'][0], map['w'][0], map['nww'][0]) + + + + ## + ## Displaying methods + ## + + ''' + Prints the name of the program and the version of the program + ''' + def version(self): + print('%s %s' % ('ponysay', VERSION)) + + + ''' + Print the pony with a speech or though bubble. message, pony and wrap from args are used. + ''' + def print_pony(self, args): + if args.message == None: + msg = ''.join(sys.stdin.readlines()).rstrip() + else: + msg = args.message + + if args.opts['-c'] is not None: ## This algorithm should give some result as cowsay's (according to tests) + buf = '' + last = ' ' + CHARS = '\t \n' + for c in msg: + if (c in CHARS) and (last in CHARS): + if last == '\n': + buf += last + last = c + else: + buf += c + last = c + msg = buf.strip(CHARS) + buf = '' + for c in msg: + if (c != '\n') or (last != '\n'): + buf += c + last = c + msg = buf.replace('\n', '\n\n') + + pony = self.__getponypath(args.opts['-f']) + + if (len(pony) > 4) and (pony[-4:].lower() == '.png'): + pony = '\'' + pony.replace('\'', '\'\\\'\'') + '\'' + pngcmd = ('img2ponysay -p -- ' if linuxvt else 'img2ponysay -- ') + pony + pngpipe = os.pipe() + Popen(pngcmd, stdout=os.fdopen(pngpipe[1], 'w'), shell=True).wait() + pony = '/proc/' + str(os.getpid()) + '/fd/' + str(pngpipe[0]) + + pony = self.__kms(pony) + + if linuxvt: + print('\033[H\033[2J', end='') + + env_width = os.environ['PONYSAY_FULL_WIDTH'] if 'PONYSAY_FULL_WIDTH' in os.environ else None + if env_width is None: env_width = '' + widthtruncation = self.__gettermsize()[1] if env_width not in ('yes', 'y', '1') else None + messagewrap = int(args.opts['-W'][0]) if args.opts['-W'] is not None else None + + balloon = self.__getballoon(self.__getballoonpath(args.opts['-b'])) + + backend = Backend(message = msg, ponyfile = pony, wrapcolumn = messagewrap if messagewrap is not None else 40, width = widthtruncation, balloon = balloon) + backend.parse() + output = backend.output + if (len(output) > 0) and (output[-1] == '\n'): + output = output[:-1] + + + env_bottom = os.environ['PONYSAY_BOTTOM'] if 'PONYSAY_BOTTOM' in os.environ else None + if env_bottom is None: env_bottom = '' + + env_height = os.environ['PONYSAY_TRUNCATE_HEIGHT'] if 'PONYSAY_TRUNCATE_HEIGHT' in os.environ else None + if env_height is None: env_height = '' + + env_lines = os.environ['PONYSAY_SHELL_LINES'] if 'PONYSAY_SHELL_LINES' in os.environ else None + if (env_lines is None) or (env_lines == ''): env_lines = '2' + + lines = self.__gettermsize()[0] - int(env_lines) + + + if linuxvt or (env_height is ('yes', 'y', '1')): + if env_bottom is ('yes', 'y', '1'): + for line in output.split('\n')[: -lines]: + print(line) + else: + for line in output.split('\n')[: lines]: + print(line) + else: + print(output) + + + ''' + Print the pony with a speech or though bubble and a self quote + ''' + def quote(self, args): + pairs = self.__quotes() + if len(args.opts['-q']) > 0: + ponyset = {} + for pony in args.opts['-q']: + if (len(pony) > 5) and (pony[-5:] == '.pony'): + ponyname = pony[:-5] + if '/' in ponyname: + ponyname = ponyname[ponyname.rindex('/') + 1:] + ponyset[ponyname] = pony + else: + ponyset[pony] = pony + alts = [] + for pair in pairs: + if pair[0] in ponyset: + alts.append((ponyset[pair[0]], pair[1])) + pairs = alts + + if not len(pairs) == 0: + pair = pairs[random.randrange(0, len(pairs))] + qfile = None + try: + qfile = open(pair[1], 'rb') + args.message = qfile.read().decode('utf8', 'replace').strip() + finally: + if qfile is not None: + qfile.close() + args.opts['-f'] = [pair[0]] + elif len(args.opts['-q']) == 0: + sys.stderr.write('Princess Celestia! All the ponies are mute!\n') + exit(1) + else: + args.opts['-f'] = [args.opts['-q'][random.randrange(0, len(args.opts['-q']))]] + args.message = 'Zecora! Help me, I am mute!' + + self.print_pony(args) + + + ''' + Indentifies whether KMS support is utilised + ''' + @staticmethod + def isUsingKMS(): + if not linuxvt: + return False + + env_kms = os.environ['PONYSAY_KMS_PALETTE'] if 'PONYSAY_KMS_PALETTE' in os.environ else None + if env_kms is None: env_kms = '' + + env_kms_cmd = os.environ['PONYSAY_KMS_PALETTE_CMD'] if 'PONYSAY_KMS_PALETTE_CMD' in os.environ else None + if (env_kms_cmd is not None) and (not env_kms_cmd == ''): + env_kms = Popen(shlex.split(env_kms_cmd), stdout=PIPE, stdin=sys.stderr).communicate()[0].decode('utf8', 'replace') + if env_kms[-1] == '\n': + env_kms = env_kms[:-1] + + return env_kms != '' + + + ''' + Returns the file name of the input pony converted to a KMS pony, or if KMS is not used, the input pony itself + ''' + def __kms(self, pony): + if not linuxvt: + return pony + + KMS_VERSION = '1' + + env_kms = os.environ['PONYSAY_KMS_PALETTE'] if 'PONYSAY_KMS_PALETTE' in os.environ else None + if env_kms is None: env_kms = '' + + env_kms_cmd = os.environ['PONYSAY_KMS_PALETTE_CMD'] if 'PONYSAY_KMS_PALETTE_CMD' in os.environ else None + if (env_kms_cmd is not None) and (not env_kms_cmd == ''): + env_kms = Popen(shlex.split(env_kms_cmd), stdout=PIPE, stdin=sys.stderr).communicate()[0].decode('utf8', 'replace') + if env_kms[-1] == '\n': + env_kms = env_kms[:-1] + + if env_kms == '': + return pony + + palette = env_kms + palettefile = env_kms.replace('\033]P', '') + + cachedir = '/var/cache/ponysay' + if not os.path.isdir(cachedir): + cachedir = HOME + '/.cache/ponysay' + if not os.path.isdir(cachedir): + os.makedirs(cachedir) + + newversion = False + if not os.path.isfile(cachedir + '/.version'): + newversion = True + else: + cachev = None + try: + cachev = open(cachedir + '/.version', 'rb') + if cachev.read().decode('utf8', 'replace').replace('\n', '') == KMS_VERSION: + newversion = True + finally: + if cachev is not None: + cachev.close() + if newversion: + for cached in os.listdir(cachedir): + cached = cachedir + '/' + cached + if os.path.isdir(cached) and not os.path.islink(cached): + shutil.rmtree(cached, False) + else: + os.remove(cached) + cachev = None + try: + cachev = open(cachedir + '/.version', 'w+') + cachev.write(KMS_VERSION) + finally: + if cachev is not None: + cachev.close() + + kmsponies = cachedir + '/kmsponies/' + palettefile + kmspony = (kmsponies + pony).replace('//', '/') + + if not os.path.isfile(kmspony): + protokmsponies = cachedir + '/protokmsponies/' + protokmspony = (protokmsponies + pony).replace('//', '/') + + protokmsponydir = protokmspony[:protokmspony.rindex('/')] + kmsponydir = kmspony[: kmspony.rindex('/')] + + _protokmspony = '\'' + protokmspony.replace('\'', '\'\\\'\'') + '\'' + _kmspony = '\'' + kmspony.replace('\'', '\'\\\'\'') + '\'' + _pony = '\'' + pony.replace('\'', '\'\\\'\'') + '\'' + + if not os.path.isfile(protokmspony): + if not os.path.isdir(protokmsponydir): + os.makedirs(protokmsponydir) + if not os.system('ponysay2ttyponysay < ' + _pony + ' > ' + _protokmspony) == 0: + sys.stderr.write('Unable to run ponysay2ttyponysay successfully, you need util-say for KMS support\n') + exit(1) + + if not os.path.isdir(kmsponydir): + os.makedirs(kmsponydir) + if not os.system('tty2colourfultty -p ' + palette + ' < ' + _protokmspony + ' > ' + _kmspony) == 0: + sys.stderr.write('Unable to run tty2colourfultty successfully, you need util-say for KMS support\n') + exit(1) + + return kmspony + + + +ARGUMENTLESS = 0 +ARGUMENTED = 1 +VARIADIC = 2 +''' +Simple argument parser +''' +class ArgParser(): + ''' + Constructor. + The short description is printed on same line as the program name + ''' + def __init__(self, program, description, usage, longdescription = None): + self.__program = program + self.__description = description + self.__usage = usage + self.__longdescription = longdescription + self.__arguments = [] + self.opts = {} + self.optmap = {} + + + ''' + Add option that takes no arguments + ''' + def add_argumentless(self, alternatives, help = None): + ARGUMENTLESS + self.__arguments.append((ARGUMENTLESS, alternatives, None, help)) + stdalt = alternatives[0] + self.opts[stdalt] = None + for alt in alternatives: + self.optmap[alt] = (stdalt, ARGUMENTLESS) + + ''' + Add option that takes one argument + ''' + def add_argumented(self, alternatives, arg, help = None): + self.__arguments.append((ARGUMENTED, alternatives, arg, help)) + stdalt = alternatives[0] + self.opts[stdalt] = None + for alt in alternatives: + self.optmap[alt] = (stdalt, ARGUMENTED) + + ''' + Add option that takes all following argument + ''' + def add_variadic(self, alternatives, arg, help = None): + self.__arguments.append((VARIADIC, alternatives, arg, help)) + stdalt = alternatives[0] + self.opts[stdalt] = None + for alt in alternatives: + self.optmap[alt] = (stdalt, VARIADIC) + + + ''' + Parse arguments + ''' + def parse(self, argv = sys.argv): + self.argcount = len(argv) - 1 + self.files = [] + + argqueue = [] + optqueue = [] + deque = [] + for arg in argv[1:]: + deque.append(arg) + + dashed = False + tmpdashed = False + get = 0 + dontget = 0 + + def unrecognised(arg): + sys.stderr.write('%s: warning: unrecognised option %s\n' % (self.__program, arg)) + + while len(deque) != 0: + arg = deque[0] + deque = deque[1:] + if (get > 0) and (dontget == 0): + get -= 1 + argqueue.append(arg) + elif tmpdashed: + self.files.append(arg) + tmpdashed = False + elif dashed: self.files.append(arg) + elif arg == '++': tmpdashed = True + elif arg == '--': dashed = True + elif (len(arg) > 1) and (arg[0] in ('-', '+')): + if (len(arg) > 2) and (arg[:2] in ('--', '++')): + if dontget > 0: + dontget -= 1 + elif (arg in self.optmap) and (self.optmap[arg][1] == ARGUMENTLESS): + optqueue.append(arg) + argqueue.append(None) + elif '=' in arg: + arg_opt = arg[:arg.index('=')] + if (arg_opt in self.optmap) and (self.optmap[arg_opt][1] >= ARGUMENTED): + optqueue.append(arg_opt) + argqueue.append(arg[arg.index('=') + 1:]) + if self.optmap[arg_opt][1] == VARIADIC: + dashed = True + else: + unrecognised(arg) + elif (arg in self.optmap) and (self.optmap[arg][1] == ARGUMENTED): + optqueue.append(arg) + get += 1 + elif (arg in self.optmap) and (self.optmap[arg][1] == VARIADIC): + optqueue.append(arg) + argqueue.append(None) + dashed = True + else: + unrecognised(arg) + else: + sign = arg[0] + i = 1 + n = len(arg) + while i < n: + narg = sign + arg[i] + i += 1 + if (narg in self.optmap): + if self.optmap[narg][1] == ARGUMENTLESS: + optqueue.append(narg) + argqueue.append(None) + elif self.optmap[narg][1] == ARGUMENTED: + optqueue.append(narg) + nargarg = arg[i:] + if len(nargarg) == 0: + get += 1 + else: + argqueue.append(nargarg) + break + elif self.optmap[narg][1] == VARIADIC: + optqueue.append(narg) + nargarg = arg[i:] + argqueue.append(nargarg if len(nargarg) > 0 else None) + dashed = True + break + else: + unrecognised(arg) + else: + self.files.append(arg) + + i = 0 + n = len(optqueue) + while i < n: + opt = optqueue[i] + arg = argqueue[i] + i += 1 + opt = self.optmap[opt][0] + if (opt not in self.opts) or (self.opts[opt] is None): + self.opts[opt] = [] + self.opts[opt].append(arg) + + for arg in self.__arguments: + if (arg[0] == VARIADIC): + varopt = self.opts[arg[1][0]] + if varopt is not None: + additional = ','.join(self.files).split(',') if len(self.files) > 0 else [] + if varopt[0] is None: + self.opts[arg[1][0]] = additional + else: + self.opts[arg[1][0]] = varopt[0].split(',') + additional + self.files = [] + break + + self.message = ' '.join(self.files) if len(self.files) > 0 else None + + + ''' + Prints a colourful help message + ''' + def help(self): + print('\033[1m%s\033[21m %s %s' % (self.__program, '-' if linuxvt else '—', self.__description)) + print() + if self.__longdescription is not None: + print(self.__longdescription) + print() + + print('\033[1mUSAGE:\033[21m', end='') + first = True + for line in self.__usage.split('\n'): + if first: + first = False + else: + print(' or', end='') + print('\t%s' % (line)) + print() + + print('\033[1mSYNOPSIS:\033[21m') + (lines, lens) = ([], []) + for opt in self.__arguments: + opt_type = opt[0] + opt_alts = opt[1] + opt_arg = opt[2] + opt_help = opt[3] + if opt_help is None: + continue + (line, l) = ('', 0) + first = opt_alts[0] + last = opt_alts[-1] + alts = ('', last) if first is last else (first, last) + for opt_alt in alts: + if opt_alt is alts[-1]: + line += '%colour%' + opt_alt + l += len(opt_alt) + if opt_type == ARGUMENTED: line += ' \033[4m%s\033[24m' % (opt_arg); l += len(opt_arg) + 1 + elif opt_type == VARIADIC: line += ' [\033[4m%s\033[24m...]' % (opt_arg); l += len(opt_arg) + 6 + else: + line += ' \033[2m%s\033[22m ' % (opt_alt) + l += len(opt_alt) + 6 + lines.append(line) + lens.append(l) + + col = max(lens) + col += 8 - ((col - 4) & 7) + index = 0 + for opt in self.__arguments: + opt_help = opt[3] + if opt_help is None: + continue + first = True + colour = '36' if (index & 1) == 0 else '34' + print(lines[index].replace('%colour%', '\033[%s;1m' % (colour)), end=' ' * (col - lens[index])) + for line in opt_help.split('\n'): + if first: + first = False + print('%s' % (line), end='\033[21;39m\n') + else: + print('%s\033[%sm%s\033[39m' % (' ' * col, colour, line)) + index += 1 + + print() + + + +''' +Balloon format class +''' +class Balloon(): + ''' + Constructor + ''' + def __init__(self, link, linkmirror, ww, ee, nw, nnw, n, nne, ne, nee, e, see, se, sse, s, ssw, sw, sww, w, nww): + (self.link, self.linkmirror) = (link, linkmirror) + (self.ww, self.ee) = (ww, ee) + (self.nw, self.ne, self.se, self.sw) = (nw, ne, se, sw) + (self.nnw, self.n, self.nne) = (nnw, n, nne) + (self.nee, self.e, self.see) = (nee, e, see) + (self.sse, self.s, self.ssw) = (sse, s, ssw) + (self.sww, self.w, self.nww) = (sww, w, nww) + + _ne = max(ne, key = UCS.dispLen) + _nw = max(nw, key = UCS.dispLen) + _se = max(se, key = UCS.dispLen) + _sw = max(sw, key = UCS.dispLen) + + minE = UCS.dispLen(max([_ne, nee, e, see, _se, ee], key = UCS.dispLen)) + minW = UCS.dispLen(max([_nw, nww, e, sww, _sw, ww], key = UCS.dispLen)) + minN = len(max([ne, nne, n, nnw, nw], key = len)) + minS = len(max([se, sse, s, ssw, sw], key = len)) + + self.minwidth = minE + minE + self.minheight = minN + minS + + + ''' + Generates a balloon with a message + ''' + def get(self, minw, minh, lines, lencalc): + h = self.minheight + len(lines) + w = self.minwidth + lencalc(max(lines, key = lencalc)) + if w < minw: w = minw + if h < minh: h = minh + + if len(lines) > 1: + (ws, es) = ({0 : self.nww, len(lines) - 1 : self.sww}, {0 : self.nee, len(lines) - 1 : self.see}) + for j in range(1, len(lines) - 1): + ws[j] = self.w + es[j] = self.e + else: + (ws, es) = ({0 : self.ww}, {0 : self.ee}) + + rc = [] + + for j in range(0, len(self.n)): + outer = UCS.dispLen(self.nw[j]) + UCS.dispLen(self.ne[j]) + inner = UCS.dispLen(self.nnw[j]) + UCS.dispLen(self.nne[j]) + if outer + inner >= w: + rc.append(self.nw[j] + self.nnw[j] + self.n[j] * (w - outer - inner) + self.nne[j] + self.ne[j]) + else: + rc.append(self.nw[j] + self.n[j] * (w - outer) + self.ne[j]) + + for j in range(0, len(lines)): + rc.append(ws[j] + lines[j] + ' ' * (w - lencalc(lines[j]) - UCS.dispLen(self.w) - UCS.dispLen(self.e)) + es[j]) + + for j in range(0, len(self.s)): + outer = UCS.dispLen(self.sw[j]) + UCS.dispLen(self.se[j]) + inner = UCS.dispLen(self.ssw[j]) + UCS.dispLen(self.sse[j]) + if outer + inner >= w: + rc.append(self.sw[j] + self.ssw[j] + self.s[j] * (w - outer - inner) + self.sse[j] + self.se[j]) + else: + rc.append(self.sw[j] + self.s[j] * (w - outer) + self.se[j]) + + return '\n'.join(rc) + + + +''' +Replacement for cowsay +''' +class Backend(): + ''' + Constructor + Takes message [string], ponyfile [filename string], wrapcolumn [None or an int], width [None or an int] and balloon [Balloon object] + ''' + def __init__(self, message, ponyfile, wrapcolumn, width, balloon): + self.message = message + self.ponyfile = ponyfile + self.wrapcolumn = wrapcolumn + self.width = width + self.balloon = balloon + + self.link = {'\\' : self.balloon.link, '/' : self.balloon.linkmirror} + + self.output = '' + self.pony = None + + + ''' + Process all data + ''' + def parse(self): + self.__expandMessage() + self.__loadFile() + self.__processPony() + self.__truncate() + + + ''' + Converts all tabs in the message to spaces by expanding + ''' + def __expandMessage(self): + lines = self.message.split('\n') + buf = '' + for line in lines: + (i, n, x) = (0, len(line), 0) + while i < n: + c = line[i] + i += 1 + if c == '\033': + colour = self.__getcolour(line, i - 1) + i += len(colour) - 1 + buf += colour + elif c == '\t': + nx = 8 - (x & 7) + buf += ' ' * nx + x += nx + else: + buf += c + if not UCS.isCombining(c): + x += 1 + buf += '\n' + self.message = buf[:-1] + + + ''' + Loads the pony file + ''' + def __loadFile(self): + ponystream = None + try: + ponystream = open(self.ponyfile, 'rb') + self.pony = ponystream.read().decode('utf8', 'replace') + finally: + if ponystream is not None: + ponystream.close() + + + ''' + Truncate output to the width of the screen + ''' + def __truncate(self): + if self.width is None: + return + lines = self.output.split('\n') + self.output = '' + for line in lines: + (i, n, x) = (0, len(line), 0) + while i < n: + c = line[i] + i += 1 + if c == '\033': + colour = self.__getcolour(line, i - 1) + i += len(colour) - 1 + self.output += colour + else: + if x < self.width: + self.output += c + if not UCS.isCombining(c): + x += 1 + self.output += '\n' + self.output = self.output[:-1] + + + ''' + Process the pony file and generate output to self.output + ''' + def __processPony(self): + self.output = '' + + AUTO_PUSH = '\033[01010~' + AUTO_POP = '\033[10101~' + + variables = {'' : '$'} + for key in self.link: + variables[key] = AUTO_PUSH + self.link[key] + AUTO_POP + + indent = 0 + dollar = None + balloonLines = None + colourstack = ColourStack(AUTO_PUSH, AUTO_POP) + + (i, n, lineindex, skip, nonskip) = (0, len(self.pony), 0, 0, 0) + while i < n: + c = self.pony[i] + if c == '\t': + n += 8 - (indent & 7) + ed = ' ' * (7 - (indent & 7)) + c = ' ' + self.pony = self.pony[:i] + ed + self.pony[i:] + i += 1 + if c == '$': + if dollar is not None: + if '=' in dollar: + name = dollar[:find('=')] + value = dollar[find('=') + 1:] + variables[name] = value + elif (len(dollar) < 7) or not (dollar[:7] == 'balloon'): + data = variables[dollar].replace('$', '$$') + if data == '$$': # if not handled specially we will get an infinity loop + if (skip == 0) or (nonskip > 0): + if nonskip > 0: + nonskip -= 1 + self.output += '$' + indent += 1 + else: + skip -= 1 + else: + n += len(data) + self.pony = self.pony[:i] + data + self.pony[i:] + else: + (w, h) = (0, 0) + props = dollar[7:] + if len(props) > 0: + if ',' in props: + if props[0] is not ',': + w = int(props[:props.index(',')]) + h = int(props[props.index(',') + 1:]) + else: + w = int(props) + balloon = self.__getballoon(w, h, indent) + balloon = balloon.split('\n') + for b in balloon[0]: + self.output += b + colourstack.feed(b) + if lineindex == 0: + balloonpre = '\n' + (' ' * indent) + for line in balloon[1:]: + self.output += balloonpre; + for b in line: + self.output += b + colourstack.feed(b); + indent = 0 + elif len(balloon) > 1: + balloonLines = balloon + balloonLine = 0 + balloonIndent = indent + indent += self.__len(balloonLines[0]) + balloonLines[0] = None + dollar = None + else: + dollar = '' + elif dollar is not None: + if c == '\033': + c = self.pony[i] + i += 1 + dollar += c + elif c == '\033': + colour = self.__getcolour(self.pony, i - 1) + for b in colour: + self.output += b + colourstack.feed(b); + i += len(colour) - 1 + elif c == '\n': + self.output += c + indent = 0 + (skip, nonskip) = (0, 0) + lineindex += 1 + if balloonLines is not None: + balloonLine += 1 + if balloonLine == len(balloonLines): + balloonLines = None + else: + if (balloonLines is not None) and (balloonLines[balloonLine] is not None) and (balloonIndent == indent): + data = balloonLines[balloonLine] + datalen = self.__len(data) + skip += datalen + nonskip += datalen + data = data.replace('$', '$$') + n += len(data) + self.pony = self.pony[:i] + data + self.pony[i:] + balloonLines[balloonLine] = None + else: + if (skip == 0) or (nonskip > 0): + if nonskip > 0: + nonskip -= 1 + self.output += c + colourstack.feed(c); + if not UCS.isCombining(c): + indent += 1 + else: + skip -= 1 + + if balloonLines is not None: + for line in balloonLines[balloonLine:]: + data = ' ' * (balloonIndent - indent) + line + '\n' + for b in data: + self.output += b + colourstack.feed(b); + indent = 0 + + self.output = self.output.replace(AUTO_PUSH, '').replace(AUTO_POP, '') + + + ''' + Gets colour code att the currect offset in a buffer + ''' + def __getcolour(self, input, offset): + (i, n) = (offset, len(input)) + rc = input[i] + i += 1 + if i == n: return rc + c = input[i] + i += 1 + rc += c + + if c == ']': + if i == n: return rc + c = input[i] + i += 1 + rc += c + if c == 'P': + di = 0 + while (di < 7) and (i < n): + c = input[i] + i += 1 + di += 1 + rc += c + elif c == '[': + while i < n: + c = input[i] + i += 1 + rc += c + if (c == '~') or (('a' <= c) and (c <= 'z')) or (('A' <= c) and (c <= 'Z')): + break + + return rc + + + ''' + Calculates the number of visible characters in a text + ''' + def __len(self, input): + (rc, i, n) = (0, 0, len(input)) + while i < n: + c = input[i] + if c == '\033': + i += len(self.__getcolour(input, i)) + else: + i += 1 + if not UCS.isCombining(c): + rc += 1 + return rc + + + ''' + Generates a balloon with the message + ''' + def __getballoon(self, width, height, left): + wrap = None + if self.wrapcolumn is not None: + wrap = self.wrapcolumn - left + if wrap < 8: + wrap = 8 + + msg = self.message + if wrap is not None: + msg = self.__wrapMessage(msg, wrap) + + if '\033' in msg: + AUTO_PUSH = '\033[01010~' + AUTO_POP = '\033[10101~' + cstack = ColourStack(AUTO_PUSH, AUTO_POP) + buf = '' + for c in msg: + if c == '\n': + for cc in ('%s\n%s' % (AUTO_PUSH, AUTO_POP)): + buf += cc + buf += cstack.feed(cc) + else: + buf += c + buf += cstack.feed(c) + msg = buf + + return self.balloon.get(width, height, msg.split('\n'), self.__len) + + + ''' + Wraps the message + ''' + def __wrapMessage(self, message, wrap): + lines = message.split('\n') + buf = '' + for line in lines: + b = [None] * len(line) + map = {} + (bi, cols, w) = (0, 0, wrap) + (indent, indentc) = (-1, 0) + + (i, n) = (0, len(line)) + while i <= n: + d = None + if i != n: + d = line[i] + i += 1 + if d == '\033': # TODO this should use self.__getcolour() + b[bi] = d + bi += 1 + b[bi] = line[i] + d = line[i] + bi += 1 + i += 1 + if d == '[': + while True: + b[bi] = line[i] + d = line[i] + bi += 1 + i += 1 + if (('a' <= d) and (d <= 'z')) or (('A' <= d) and (d <= 'Z')) or (d == '~'): + break + elif d == ']': + b[bi] = line[i] + d = line[i] + bi += 1 + i += 1 + if d == 'P': + for j in range(0, 7): + b[bi] = line[i] + bi += 1 + i += 1 + elif (d is not None) and (d != ' '): + if indent == -1: + indent = i - 1 + for j in range(0, indent): + if line[j] == ' ': + indentc += 1 + b[bi] = d + bi += 1 + if not UCS.isCombining(d): + cols += 1 + map[cols] = bi + else: + mm = 0 + while (w > 8) and (cols > w + 3): + mm += w - 1 + m = map[mm] + for bb in b[:m]: + buf += bb + buf += '-\n' + cols -= w - 1 + m += w -1 + bi -= m + bb = b[m:] + for j in range(0, bi): + b[j] = bb[j] + w = wrap + if indent != -1: + buf += line[:indent] + w -= indentc + if cols > w: + buf += '\n' + w = wrap + if indent != -1: + buf += line[:indent] + w -= indentc + for bb in b[:bi]: + buf += bb + w -= cols + cols = 0 + bi = 0 + if d == -1: + i += 1 + else: + if w > 0: + buf += ' ' + w -= 1 + else: + buf += '\n' + w = wrap + if indent != -1: + buf + line[:indent] + w -= indentc + + buf += '\n' + return '\n'.join(line.rstrip() for line in buf[:-1].split('\n')) + + +''' +ANSI colour stack +''' +class ColourStack(): + ''' + Constructor + ''' + def __init__(self, autopush, autopop): + self.autopush = autopush + self.autopop = autopop + self.lenpush = len(autopush) + self.lenpop = len(autopop) + self.bufproto = ' ' * (self.lenpush if self.lenpush > self.lenpop else self.lenpop) + self.stack = [] + self.push() + self.seq = None + + + def push(self): + self.stack = [[self.bufproto, None, None, [False] * 9]] + self.stack + if len(self.stack) == 1: + return None + return '\033[0m' + + + def pop(self): + old = self.stack[0] + self.stack = self.stack[1:] + rc = '\033[0;' + new = self.stack[0] + if new[1] is not None: rc += new[1] + ';' + if new[2] is not None: rc += new[2] + ';' + for i in range(0, 9): + if new[3][i]: + rc += str(i + 1) + ';' + return rc[:-1] + 'm' + + + def feed(self, char): + if self.seq is not None: + self.seq += char + if (char == '~') or (('a' <= char) and (char <= 'z')) or (('A' <= char) and (char <= 'Z')): + if (self.seq[0] == '[') and (self.seq[-1] == 'm'): + self.seq = self.seq[1:-1].split(';') + (i, n) = (0, len(self.seq)) + while i < n: + part = self.seq[i] + p = 0 if part == '' else int(part) + i += 1 + if p == 0: self.stack[0][1:] = [None, None, [False] * 9] + elif (1 <= p) and (p <= 9): self.stack[0][3][p - 1] = True + elif (21 <= p) and (p <= 29): self.stack[0][3][p - 21] = False + elif p == 39: self.stack[0][1] = None + elif p == 49: self.stack[0][2] = None + elif (30 <= p) and (p <= 37): self.stack[0][1] = part + elif (90 <= p) and (p <= 97): self.stack[0][1] = part + elif (40 <= p) and (p <= 47): self.stack[0][2] = part + elif (100 <= p) and (p <= 107): self.stack[0][2] = part + elif p == 38: + self.stack[0][1] = '%s;%s;%s' % (part, self.seq[i], self.seq[i + 1]) + i += 2 + elif p == 48: + self.stack[0][2] = '%s;%s;%s' % (part, self.seq[i], self.seq[i + 1]) + i += 2 + self.seq = None + elif char == '\033': + self.seq = '' + buf = self.stack[0][0] + buf = buf[1:] + char + rc = '' + if buf[-self.lenpush:] == self.autopush: rc = self.push() + elif buf[-self.lenpop:] == self.autopop: rc = self.pop() + self.stack[0][0] = buf + return rc + + + +''' +UCS utility class +''' +class UCS(): + ''' + Checks whether a character is a combining character + ''' + @staticmethod + def isCombining(char): + o = ord(char) + if (0x0300 <= o) and (o <= 0x036F): return True + if (0x20D0 <= o) and (o <= 0x20FF): return True + if (0x1DC0 <= o) and (o <= 0x1DFF): return True + if (0xFE20 <= o) and (o <= 0xFE2F): return True + return False + + + ''' + Gets the number of combining characters in a string + ''' + @staticmethod + def countCombining(string): + rc = 0 + for char in string: + if UCS.isCombining(char): + rc += 1 + return rc + + + ''' + Gets length of a string not counting combining characters + ''' + @staticmethod + def dispLen(string): + return len(string) - UCS.countCombining(string) + + + +''' +The user's home directory +''' +HOME = os.environ['HOME'] if 'HOME' in os.environ else os.path.expanduser('~') + + +''' +Whether the program is execute in Linux VT (TTY) +''' +linuxvt = ('TERM' in os.environ) and (os.environ['TERM'] == 'linux') + + +''' +Whether the script is executed as ponythink +''' +isthink = (len(__file__) >= 5) and (__file__[-5:] == 'think') +isthink = ((len(__file__) >= 8) and (__file__[-8:] == 'think.py')) or isthink + + +''' +Whether stdin is piped +''' +pipelinein = not sys.stdin.isatty() + +''' +Whether stdout is piped +''' +pipelineout = not sys.stdout.isatty() + +''' +Whether stderr is piped +''' +pipelineerr = not sys.stderr.isatty() + + +''' +Whether KMS is used +''' +usekms = Ponysay.isUsingKMS() + + +''' +The directories where pony files are stored, ttyponies/ are used if the terminal is Linux VT (also known as TTY) and not with KMS +''' +appendset = set() +ponydirs = [] +if linuxvt and not usekms: _ponydirs = [HOME + '/.local/share/ponysay/ttyponies/', '/usr/share/ponysay/ttyponies/'] +else: _ponydirs = [HOME + '/.local/share/ponysay/ponies/', '/usr/share/ponysay/ponies/'] +for ponydir in _ponydirs: + if os.path.isdir(ponydir) and (ponydir not in appendset): + ponydirs.append(ponydir) + appendset.add(ponydir) + + +''' +The directories where pony files are stored, extrattyponies/ are used if the terminal is Linux VT (also known as TTY) and not with KMS +''' +appendsetset = set() +extraponydirs = [] +if linuxvt and not usekms: _extraponydirs = [HOME + '/.local/share/ponysay/extrattyponies/', '/usr/share/ponysay/extrattyponies/'] +else: _extraponydirs = [HOME + '/.local/share/ponysay/extraponies/', '/usr/share/ponysay/extraponies/'] +for extraponydir in _extraponydirs: + if os.path.isdir(extraponydir) and (extraponydir not in appendset): + extraponydirs.append(extraponydir) + appendset.add(extraponydir) + + +''' +The directories where quotes files are stored +''' +appendset = set() +quotedirs = [] +_quotedirs = [HOME + '/.local/share/ponysay/quotes/', '/usr/share/ponysay/quotes/'] +for quotedir in _quotedirs: + if os.path.isdir(quotedir) and (quotedir not in appendset): + quotedirs.append(quotedir) + appendset.add(quotedir) + + +''' +The directories where balloon style files are stored +''' +appendset = set() +balloondirs = [] +_balloondirs = [HOME + '/.local/share/ponysay/balloons/', '/usr/share/ponysay/balloons/'] +for balloondir in _balloondirs: + if os.path.isdir(balloondir) and (balloondir not in appendset): + balloondirs.append(balloondir) + appendset.add(balloondir) + + +''' +ucsmap files +''' +appendset = set() +ucsmaps = [] +_ucsmaps = [HOME + '/.local/share/ponysay/ucsmap', '/usr/share/ponysay/ucsmap'] +for ucsmap in _ucsmaps: + if os.path.isdir(ucsmap) and (ucsmap not in appendset): + ucsmaps.append(ucsmap) + appendset.add(ucsmap) + + + +usage_saythink = '\033[34;1m(ponysay | ponythink)\033[21;39m' +usage_common = '[-c] [-W\033[4mCOLUMN\033[24m] [-b\033[4mSTYLE\033[24m]' +usage_listhelp = '(-l | -L | -B | +l | +L | -v | -h)' +usage_file = '[-f\033[4mPONY\033[24m]* [[--] \033[4mmessage\033[24m]' +usage_xfile = '[-F\033[4mPONY\033[24m]* [[--] \033[4mmessage\033[24m]' +usage_quote = '-q [\033[4mPONY\033[24m*]' + +usage = '%s %s\n%s %s %s\n%s %s %s\n%s %s %s' % (usage_saythink, usage_listhelp, + usage_saythink, usage_common, usage_file, + usage_saythink, usage_common, usage_xfile, + usage_saythink, usage_common, usage_quote) + +usage = usage.replace('\033[', '\0') +for sym in ('[', ']', '(', ')', '|', '...', '*'): + usage = usage.replace(sym, '\033[2m' + sym + '\033[22m') +usage = usage.replace('\0', '\033[') + +''' +Argument parsing +''' +opts = ArgParser(program = 'ponythink' if isthink else 'ponysay', + description = 'cowsay reimplemention for ponies', + usage = usage, + longdescription = +'''Ponysay displays an image of a pony saying some text provided by the user. +If \033[4mmessage\033[24m is not provided, it accepts standard input. For an extensive +documentation run `info ponysay`, or for just a little more help than this +run `man ponysay`. Ponysay has so much more to offer than described here.''') + +opts.add_argumentless(['--quoters']) +opts.add_argumentless(['--onelist']) +opts.add_argumentless(['++onelist']) + +opts.add_argumentless(['-h', '--help'], help = 'Print this help message.') +opts.add_argumentless(['-v', '--version'], help = 'Print the version of the program.') +opts.add_argumentless(['-l', '--list'], help = 'List pony names.') +opts.add_argumentless(['-L', '--symlist', '--altlist'], help = 'List pony names with alternatives.') +opts.add_argumentless(['+l', '++list'], help = 'List non-MLP:FiM pony names.') +opts.add_argumentless(['+L', '++symlist', '++altlist'], help = 'List non-MLP:FiM pony names with alternatives.') +opts.add_argumentless(['-A', '--all'], help = 'List MLP:FIM and non-MLP:FiM pony names.') +opts.add_argumentless(['+A', '++all'], help = 'List MLP:FIM and non-MLP:FiM pony names with alternatives.') +opts.add_argumentless(['-B', '--bubblelist', '--balloonlist'], help = 'List balloon styles.') +opts.add_argumentless(['-c', '--compact'], help = 'Compress messages.') +opts.add_argumented( ['-W', '--wrap'], arg = 'COLUMN', help = 'Specify column where the message should be wrapped.') +opts.add_argumented( ['-b', '--bubble', '--balloon'], arg = 'STYLE', help = 'Select a balloon style.') +opts.add_argumented( ['-f', '--file', '--pony'], arg = 'PONY', help = 'Select a pony.\nEither a file name or a pony name.') +opts.add_argumented( ['-F', '++file', '++pony'], arg = 'PONY', help = 'Select a non-MLP:FiM pony.') +opts.add_variadic( ['-q', '--quote'], arg = 'PONY', help = 'Select a ponies which will quote themself.') + +opts.parse() + + + +''' +Start the program from ponysay.__init__ if this is the executed file +''' +if __name__ == '__main__': + Ponysay(opts) diff --git a/setup.py b/setup.py index db5b0d1..24bd570 100755 --- a/setup.py +++ b/setup.py @@ -371,7 +371,7 @@ class Setup(): (fileout, filein) = (None, None) try: fileout = open('ponysay.install', 'wb+') - filein = open('ponysay', 'rb') + filein = open('ponysay.py', 'rb') data = filein.read().decode('utf-8', 'replace') if '#!/usr/bin/env python3' in data: -- cgit