aboutsummaryrefslogtreecommitdiff
path: root/ponysay
diff options
context:
space:
mode:
Diffstat (limited to 'ponysay')
-rwxr-xr-xponysay1779
1 files changed, 0 insertions, 1779 deletions
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)
generated by cgit (git 2.34.1) at 2024-11-21 18:02:05 +0000