diff options
-rwxr-xr-x | ponysay.py | 159 |
1 files changed, 159 insertions, 0 deletions
diff --git a/ponysay.py b/ponysay.py new file mode 100755 index 0000000..46a3932 --- /dev/null +++ b/ponysay.py @@ -0,0 +1,159 @@ +#!/usr/bin/env python3 +# -*- coding: utf-8 -*- + +''' +ponysay.py - POC of ponysay in python +Copyright (C) 2012 Elis "etu" Axelsson, Mattias "maandree" Andrée + +License: WTFPL +''' + +import argparse +import os +import sys +import random +from subprocess import Popen, PIPE + + +''' +The version of ponysay +''' +VERSION = "2.0-alpha" + + +''' +The directory where ponysay is installed, this is modified when building with make +''' +INSTALLDIR = '/usr' + + +''' +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/' ] +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') + +args = parser.parse_args() + + +class ponysay(): + def __init__(self, args): + if args.list: self.list() + else: self.print_pony(args) + + + ''' + Returns a set with all ponies that have quotes and is displayable + ''' + def __quoters(self): + quotes = [] + quoteshash = set() + _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: + 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 = os.listdir(INSTALLDIR + '/share/ponysay/quotes/') + rc = [] + + 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 + '+'): + rc.append((p, qoute)) + + return rc + + + ''' + 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] + + quoters = self.__quoters() + + for ponydir in ponydirs: # Loop ponydirs + print('\033[1mponyfiles located in ' + ponydir + '\033[21m') + + ponies = os.listdir(ponydir) + ponies = [item[:-5] for item in ponies] # remove .pony from file name + ponies.sort() + + width = len(max(ponies, key = len)) + 2 # Get the longest ponyfilename lenght + 2 spaces + + 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 + x += width + if x > (termsize[1] - width): # If too wide, make new line + print(); + x = 0 + + print("\n"); + + + def print_pony(self, args): + if args.message == None: + msg = sys.stdin.read().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 + + else: + for ponydir in ponydirs: + if os.path.isfile(ponydir + args.pony[0]): + pony = ponydir + args.pony[0] + + os.system('cowsay -f ' + pony + ' "' + msg + '"') + + + + +''' +Start the program from ponysay.__init__ if this is the executed file +''' +if __name__ == '__main__': + ponysay(args) |