mirror of
https://github.com/erkin/ponysay.git
synced 2024-11-29 23:48:00 +01:00
159 lines
5 KiB
Python
Executable file
159 lines
5 KiB
Python
Executable file
#!/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)
|