aboutsummaryrefslogtreecommitdiff
path: root/ponysay.py
diff options
context:
space:
mode:
Diffstat (limited to 'ponysay.py')
-rwxr-xr-xponysay.py710
1 files changed, 671 insertions, 39 deletions
diff --git a/ponysay.py b/ponysay.py
index 46a3932..f9a87e6 100755
--- a/ponysay.py
+++ b/ponysay.py
@@ -2,13 +2,19 @@
# -*- coding: utf-8 -*-
'''
-ponysay.py - POC of ponysay in python
-Copyright (C) 2012 Elis "etu" Axelsson, Mattias "maandree" Andrée
+ponysay.py - Ponysay, a cowsay wrapper for ponies
+Copyright (C) 2012 Erkin Batu Altunbaş
+
+Authors: 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 first implementation
+ Sven-Hendrik "svenstaro" Haase: Major contributor first implementation
+ Kyah "L-four" Rindlisbacher: Patched the first implementation
+ Jan Alexander "heftig" Steffens: Major contributor first implementation
License: WTFPL
'''
-import argparse
import os
import sys
import random
@@ -18,7 +24,7 @@ from subprocess import Popen, PIPE
'''
The version of ponysay
'''
-VERSION = "2.0-alpha"
+VERSION = '2.0'
'''
@@ -28,38 +34,128 @@ INSTALLDIR = '/usr'
'''
+The user's home directory
+'''
+HOME = os.environ['HOME']
+
+
+'''
+Whether the program is execute in Linux VT (TTY)
+'''
+linuxvt = os.environ['TERM'] == 'linux'
+
+
+'''
+Whether the script is executed as ponythink
+'''
+isthink = (len(__file__) >= 8) and (__file__[-8:] == 'think.py')
+
+
+'''
+Whether stdin is piped
+'''
+pipelinein = not sys.stdin.isatty()
+
+'''
+Whether stdout is piped
+'''
+pipelineout = False #not sys.stdout.isatty() # currently impossible, we need to get rid of the little shell script first
+
+'''
+Whether stderr is piped
+'''
+pipelineerr = not sys.stderr.isatty()
+
+
+'''
The directories where pony files are stored, ttyponies/ are used if the terminal is Linux VT (also known as TTY)
'''
ponydirs = []
-if os.environ['TERM'] == 'linux': _ponydirs = [INSTALLDIR + '/share/ponysay/ttyponies/', os.environ['HOME'] + '/.local/share/ponysay/ttyponies/']
-else: _ponydirs = [INSTALLDIR + '/share/ponysay/ponies/', os.environ['HOME'] + '/.local/share/ponysay/ponies/' ]
+if linuxvt: _ponydirs = [HOME + '/.local/share/ponysay/ttyponies/', INSTALLDIR + '/share/ponysay/ttyponies/']
+else: _ponydirs = [HOME + '/.local/share/ponysay/ponies/', INSTALLDIR + '/share/ponysay/ponies/' ]
for ponydir in _ponydirs:
if os.path.isdir(ponydir):
ponydirs.append(ponydir)
-parser = argparse.ArgumentParser(description = 'Ponysay, like cowsay with ponies')
-parser.add_argument('-v', '--version', action = 'version', version = '%s %s' % ("ponysay", VERSION))
-parser.add_argument('-l', '--list', action = 'store_true', dest = 'list', help = 'list pony files')
-parser.add_argument('-f', '--pony', action = 'append', dest = 'pony', help = 'select a pony (either a file name or a pony name)')
-parser.add_argument('message', nargs = '?', help = 'message to ponysay')
+'''
+The directories where quotes files are stored
+'''
+quotedirs = []
+_quotedirs = [HOME + '/.local/share/ponysay/quotes/', INSTALLDIR + '/share/ponysay/quotes/']
+for quotedir in _quotedirs:
+ if os.path.isdir(quotedir):
+ quotedirs.append(quotedir)
-args = parser.parse_args()
-class ponysay():
+'''
+This is the mane class of ponysay
+'''
+class Ponysay():
+ '''
+ Starts the part of the program the arguments indicate
+ '''
def __init__(self, args):
- if args.list: self.list()
- else: self.print_pony(args)
+ if (args.argcount == 0) and not pipelinein:
+ args.help()
+ return
+
+ if (args.opts['-l'] is not None) and pipelineout:
+ args.opts['--onelist'] = args.opts['-l']
+ args.opts['-l'] = None
+
+ 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['-q'] is not None: self.quote(args)
+ else: self.print_pony(args)
+ ##
+ ## Auxiliary methods
+ ##
+
'''
- Returns a set with all ponies that have quotes and is displayable
+ Returns one file with full path, names is filter for names, also accepts filepaths.
+ '''
+ def __getponypath(self, names = None):
+ ponies = {}
+
+ if not names == None:
+ for name in names:
+ if os.path.isfile(name):
+ ponies[name] = name
+
+ for ponydir in ponydirs:
+ for ponyfile in os.listdir(ponydir):
+ pony = ponyfile[:-5]
+ if pony not in ponies:
+ ponies[pony] = ponydir + ponyfile
+
+ 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 heared of any pony 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 = [item[:item.index('.')] for item in os.listdir(INSTALLDIR + '/share/ponysay/quotes/')]
+ _quotes = []
+ for quotedir in quotedirs:
+ _quotes += [item[:item.index('.')] for item in os.listdir(INSTALLDIR + '/share/ponysay/quotes/')]
for quote in _quotes:
if not quote == '':
if not quote in quoteshash:
@@ -83,7 +179,9 @@ class ponysay():
Returns a list with all (pony, quote file) pairs
'''
def __quotes(self):
- quotes = os.listdir(INSTALLDIR + '/share/ponysay/quotes/')
+ quotes = []
+ for quotedir in quotedirs:
+ quotes += [quotedir + item for item in os.listdir(quotedir)]
rc = []
for ponydir in ponydirs:
@@ -91,19 +189,33 @@ class ponysay():
if not pony[0] == '.':
p = pony[:-5] # remove .pony
for quote in quotes:
- if ('+' + p + '+') in ('+' + quote + '+'):
- rc.append((p, qoute))
+ 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
+ ##
+
+ '''
Lists the available ponies
'''
def list(self):
- termsize = Popen(['stty', 'size'], stdout=PIPE).communicate()[0].decode('utf8', 'replace')[:-1].split(" ")
- termsize = [int(item) for item in termsize]
-
+ termsize = self.__gettermsize()
quoters = self.__quoters()
for ponydir in ponydirs: # Loop ponydirs
@@ -118,42 +230,562 @@ class ponysay():
x = 0
for pony in ponies:
spacing = ' ' * (width - len(pony))
- print(('\033[1m' + pony + '\033[21m' if (pony in quoters) else pony) + spacing, end="") # Print ponyfilename
+ print(('\033[1m' + pony + '\033[21m' if (pony in quoters) else pony) + spacing, end='') # Print ponyfilename
x += width
if x > (termsize[1] - width): # If too wide, make new line
- print();
+ print()
x = 0
- print("\n");
+ print('\n');
+ '''
+ Lists the available ponies with alternatives inside brackets
+ '''
+ def linklist(self):
+ termsize = self.__gettermsize()
+ quoters = self.__quoters()
+
+ for ponydir in ponydirs: # Loop ponydirs
+ print('\033[1mponyfiles located in ' + ponydir + '\033[21m')
+
+ files = os.listdir(ponydir)
+ files = [item[:-5] for item in files] # remove .pony from file name
+ files.sort()
+ pairs = [(item, os.readlink(ponydir + item + '.pony') if os.path.islink(ponydir + item + '.pony') else '') for item in files]
+
+ ponymap = {}
+ for pair in pairs:
+ if 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 = []
+ widths = []
+ for pony in ponymap:
+ w = len(pony)
+ item = '\033[1m' + pony + '\033[21m' if (pony in quoters) else pony
+ syms = ponymap[pony]
+ if len(syms) > 0:
+ w += 2 + len(syms)
+ item += ' ('
+ first = True
+ for sym in syms:
+ w += len(sym)
+ if not first:
+ item += ' '
+ else:
+ first = False
+ item += '\033[1m' + sym + '\033[21m' if (sym in quoters) else sym
+ item += ')'
+ ponies.append(item)
+ widths.append(w)
+ if width < w:
+ width = w
+
+ width += 2;
+ x = 0
+ index = 0
+ for pony in ponies:
+ spacing = ' ' * (width - widths[index])
+ index += 1
+ print(pony + spacing, end='') # Print ponyfilename
+ x += width
+ if x > (termsize[1] - width): # If too wide, make new line
+ print()
+ x = 0
+
+ print('\n');
+
+
+ '''
+ Lists with all ponies that have quotes and are displayable
+ '''
+ def quoters(self):
+ last = ''
+ ponies = []
+ for pony in self.__quoters():
+ ponies.append(pony)
+ 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 = [item[:-5] for item in ponies] # remove .pony from file name
+ ponies.sort()
+ for pony in ponies:
+ if not pony == last:
+ last = pony
+ print(pony)
+
+
+ ##
+ ## Displaying methods
+ ##
+
+ '''
+ Prints the name of the program and the version of the program
+ '''
+ def version(self):
+ print('%s %s' % ('ponysay', VERSION))
+
+
+ '''
+ Returns (the cowsay command, whether it is a custom program)
+ '''
+ def __getcowsay(self):
+ if isthink:
+ cowthink = os.environ['PONYSAY_COWTHINK'] if 'PONYSAY_COWTHINK' in os.environ else None
+ return ('cowthink', False) if (cowthink is None) or (cowthink == '') else (cowthink, True)
+
+ cowsay = os.environ['PONYSAY_COWSAY'] if 'PONYSAY_COWSAY' in os.environ else None
+ return ('cowsay', False) if (cowsay is None) or (cowsay == '') else (cowsay, True)
+
+
+ '''
+ 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 = sys.stdin.read().strip()
+ msg = ''.join(sys.stdin.readlines()).strip()
else:
msg = args.message
- if args.pony == None:
- ponies = [] # Make array with direct paths to all ponies
- for ponydir in ponydirs:
- for ponyfile in os.listdir(ponydir):
- ponies.append(ponydir + ponyfile)
-
- pony = ponies[random.randrange(0, len(ponies) - 1)] # Select random pony
+ pony = self.__getponypath(args.opts['-f'])
+ (cowsay, customcowsay) = self.__getcowsay()
+
+ 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])
+
+ cmd = [cowsay, '-f', self.__kms(pony)]
+ if args.opts['-W'] is not None:
+ cmd += ['-W', args.opts['-W']]
+ cmd.append(msg)
+
+ if linuxvt:
+ print('\033[H\033[2J', end='')
+
+ proc = Popen(cmd, stdout=PIPE, stdin=sys.stderr)
+ output = proc.communicate()[0].decode('utf8', 'replace')
+ if (len(output) > 0) and (output[-1] == '\n'):
+ output = output[:-1]
+ exit_value = proc.returncode
+
+
+ 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()[1] - int(env_lines)
+
+
+ if not exit_value == 0:
+ sys.stderr.write('Unable to successfully execute' + (' custom ' if customcowsay else ' ') + 'cowsay [' + cowsay + ']\n')
+ else:
+ if linuxvt or (env_height is ('yes', 'y', '1')):
+ if env_bottom is ('yes', 'y', '1'):
+ for line in output[: -lines]:
+ print(line)
+ else:
+ for line in output[: lines]:
+ print(line)
+ else:
+ print(output);
+
+
+ ## TODO not implement, but it will be obsolete if we rewrite cowsay
+ '''
+ (if not customcowsay)
+
+ pcmd='#!/usr/bin/perl\nuse utf8;'
+ ccmd=$(for c in $(echo $PATH":" | sed -e 's/:/\/'"$cmd"' /g'); do if [ -f $c ]; then echo $c; break; fi done)
+
+ if [ ${0} == *ponythink ]; then
+ cat <(echo -e $pcmd) $ccmd > "/tmp/ponythink"
+ perl '/tmp/ponythink' "$@"
+ rm '/tmp/ponythink'
+ else
+ perl <(cat <(echo -e $pcmd) $ccmd) "$@"
+ fi
+ '''
+
+
+ '''
+ 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 = set(args.opts['-q'])
+ alts = []
+ for pair in pairs:
+ if pair[0] in ponyset:
+ alts.append(pair)
+ pairs = alts
+ if not len(pairs) == 0:
+ pair = pairs[random.randrange(0, len(pairs))]
+ qfile = None
+ try:
+ qfile = open(pair[1], 'r')
+ args.message = '\n'.join(qfile.readlines()).strip()
+ finally:
+ if qfile is not None:
+ qfile.close()
+ args.opts['-f'] = [pair[0]]
+ elif len(args.opts['-q']) == 0:
+ sys.stderr.write('All the ponies are mute! Call the Princess!\n')
+ exit(1)
else:
- for ponydir in ponydirs:
- if os.path.isfile(ponydir + args.pony[0]):
- pony = ponydir + args.pony[0]
+ args.opts['-f'] = [args.opts['-q'][random.randrange(0, len(args.opts['-q']))]]
+ args.message = 'I got nuthin\' good to say :('
+
+ self.print_pony(args)
+
+
+ '''
+ 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
+
+ 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]
- os.system('cowsay -f ' + pony + ' "' + msg + '"')
+ if env_kms == '':
+ return pony
+
+ palette = env_kms
+ palettefile = env_kms.replace('\033]P', '')
+
+ kmsponies = '/var/cache/ponysay/kmsponies/' + palettefile
+ kmspony = (kmsponies + pony).replace('//', '/')
+
+ if not os.path.isfile(kmspony):
+ protokmsponies = '/var/cache/ponysay/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 -e -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] == '-') or (arg[0] == '+')):
+ if (len(arg) > 2) and ((arg[:2] == '--') or (arg[:2] == '++')):
+ 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()
+
+ 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')
+ print()
+ 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
+ for opt_alt in opt_alts:
+ if opt_alt is opt_alts[-1]:
+ print('\t' + opt_alt, end='')
+ if opt_type == ARGUMENTED: print(' \033[4m%s\033[24m' % (opt_arg))
+ elif opt_type == VARIADIC: print(' [\033[4m%s\033[24m...]' % (opt_arg))
+ else: print()
+ else:
+ print('\t\033[2m' + opt_alt + '\033[22m')
+ first = True
+ for line in opt_help.split('\n'):
+ if first:
+ first = False
+ print('\t\t\033[32;1m%s\033[21;39m' % (line))
+ else:
+ print('\t\t%s' % (line))
+ print()
+
+ print()
+
+
+
+usage_saythink = '\033[34;1m(ponysay | ponythink)\033[21;39m'
+usage_wrap = '--wrap \033[4mCOLUMN\033[24m'
+usage_listhelp = '(--list | ---altlist | --version | --help)'
+usage_file = '[--pony \033[4mPONY\033[24m]... ([--] \033[4mmessage\033[24m | <<<\033[4mmessage\033[24m)'
+usage_quote = '--quote [\033[4mPONY\033[24m...]'
+
+usage = '%s %s\n%s [%s] %s\n%s [%s] %s' % (usage_saythink, usage_listhelp,
+ usage_saythink, usage_wrap, usage_file,
+ usage_saythink, usage_wrap, 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 wrapper 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(['-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 files.')
+opts.add_argumentless(['-L', '--altlist'], help = 'List pony files with alternatives.')
+opts.add_argumented( ['-W', '--wrap'], arg = "COLUMN", help = 'Specify the column when the message should be wrapped.')
+opts.add_argumented( ['-f', '--pony'], arg = "PONY", help = 'Select a pony.\nEither a file name or a pony name.')
+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(args)
+ Ponysay(opts)