1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
#!/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)
'''
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]
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:
print(pony + (" " * (width - len(pony))), 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)
|
