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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
|
#!/usr/bin/env python3
import os, sys, random
from os.path import dirname, realpath, exists
from pkg_resources import resource_string, resource_listdir, resource_exists
import argparse, textwrap
try:
import re2 as re
except:
import re
# (oneline, multiline, bottom, top, linkl, linkr)
# {one,multi}line := (left, right)
# {left,right} := (top, middle, bottom)
balloonstyles= {'cowsay': (((' ', '', '< '), (' ', '', '> ')), ((' /', '|', '\\ '), (' \\', '|', '/ ')), '-', '_', '\\', '/'),
'cowsay.think': (((' ', '', '( '), (' ', '', ') ')), ((' (', '(', '( '), (' )', ')', ') ')), '-', '_', 'o', 'o'),
'ascii': (((' /|', '', '\\ '), (' \\|', '', '/ ')), ((' /|', '|', '|\\'), (' \\|', '|', '|/')), '_', '_', '\\', '/'),
'ascii.think': (((' ((', '', '( '), (' ))', '', ') ')), ((' ((', '(', '(('), (' ))', ')', '))')), '_', '_', 'o', 'o'),
'unicode': ((('┌││', '', '│└ '), ('┐││', '', '│┘ ')), (('┌││', '│', '││└'), ('┐││', '│', '││┘')), '─', '─', '╲', '╱'),
'round': ((('╭││', '', '│╰ '), ('╮││', '', '│╯ ')), (('╭││', '│', '││╰'), ('╮││', '│', '││╯')), '─', '─', '╲', '╱'),
'linux-vt': ((('┌││', '', '│└ '), ('┐││', '', '│┘ ')), (('┌││', '│', '││└'), ('┐││', '│', '││┘')), '─', '─', '\\', '/')}
def list_ponies(markQuotes=False):
quotes = lambda n: ' (quotes)' if markQuotes and exists(ponypath+'/'+n+'.quotes') else ''
return [ f[:-5]+quotes(f[:-5]) for f in resource_listdir(__name__, 'ponies') if not f.endswith('.quotes') ]
def list_ponies_with_quotes():
return [ f[:-7] for f in resource_listdir(__name__, 'ponies') if f.endswith('.quotes') ]
def load_pony(name):
return str(resource_string(__name__, 'ponies/'+name+'.pony'), 'utf-8').split('\n')
def random_quote(name):
quotepath='ponies/'+name+'.quotes'
if resource_exists(__name__, quotepath):
return random.choice(str(resource_string(__name__, quotepath), 'utf-8').split('\n\n'))
else:
return None
def render_balloon(text, balloonstyle, minwidth=0, maxwidth=40, pad=str.center):
if text is None:
return []
(oneline, multiline, bottom, top, linkl, linkr) = balloonstyle
lines = [ ' '+wrapline+' ' for textline in text.center(minwidth).split('\n') for wrapline in textwrap.wrap(textline, maxwidth) ]
width = max([ len(line) for line in lines ]+[minwidth])
side = lambda top, middle, bottom: top + middle*(len(lines)-2) + bottom
leftside, rightside = oneline if len(lines) == 1 else multiline
topextra, bottomextra = len(leftside[0])-2, len(leftside[2])-2
leftside, rightside = side(*leftside), side(*rightside)
lines = [top*width] + [' '*width]*topextra + [ pad(line, width) for line in lines ] + [' '*width]*bottomextra + [bottom*width]
return [ l+m+r for l,m,r in zip(leftside, lines, rightside) ]
def render_pony(name, text, balloonstyle, width=80, center=False, centertext=False):
pony = load_pony(name)
balloon = link_l = link_r = ''
if text:
[link_l, link_r] = balloonstyle[-2:]
for i,line in enumerate(pony):
match = re.search('\$balloon([0-9]*)\$', line)
if match:
minwidth = int(match.group(1) or '0')
pony[i:i+1] = render_balloon(text, balloonstyle, minwidth=minwidth, maxwidth=int(width/2), pad=str.center if centertext else str.ljust)
break
try:
first = pony.index('$$$')
second = pony[first+1:].index('$$$')
pony[first:] = pony[first+1+second+1:]
except:
pass
pony = [ line.replace('$\\$', link_l).replace('$/$', link_r) for line in pony ]
indent = ''
if center:
ponywidth = max([ len(re.sub(r'\x1B\[[0-9;]+m|\$.*\$', '', line)) for line in pony ])
indent = ' '*int((width-ponywidth)/2)
wre = re.compile('((\x1B\[[0-9;]+m)*.){0,%s}' % width)
reset = '[39;49m\n'
return indent+(reset+indent).join([ wre.search(line).group() for line in pony ])+reset
def main():
termwidth = 80
try:
termwidth = os.get_terminal_size()[0]
except:
pass
parser = argparse.ArgumentParser(description='Cowsay with ponies')
parser.add_argument('-p', '--pony', type=str, default='random', help='The name of the pony to be used. Use "-p list" to list all ponies, "-p random" (default) to use a random pony.')
parser.add_argument('-q', '--quote', action='store_true', help='Use a random quote of the pony being displayed as text')
parser.add_argument('-c', '--center', action='store_true', help='Use a random quote of the pony being displayed as text')
parser.add_argument('-C', '--center-text', action='store_true', help='Center the text in the bubble')
parser.add_argument('-w', '--width', type=int, default=termwidth, help='Terminal width. Use 0 for unlimited width. Default: autodetect')
parser.add_argument('-b', '--balloon', type=str, default='cowsay', help='Balloon style to use. Use "-b list" to list available styles.')
parser.add_argument('text', type=str, nargs='*', help='The text to be placed in the speech bubble')
args = parser.parse_args()
think = sys.argv[0].endswith('think')
if args.pony == "list":
print('\n'.join(sorted(list_ponies() if not args.quote else list_ponies_with_quotes())))
sys.exit()
if args.balloon == 'list':
print('\n'.join([ s.replace('.think', '') for s in balloonstyles.keys() if s.endswith('.think') == think ]))
sys.exit()
pony = args.pony
if pony == "random":
pony = random.choice(list_ponies() if not args.quote else list_ponies_with_quotes())
text = ' '.join(args.text) or None
if text == '-':
text = '\n'.join(sys.stdin.readlines())
if args.quote:
text = random_quote(pony)
balloonstyle = None
if think:
balloonstyle = balloonstyles[args.balloon+'.think']
else:
balloonstyle = balloonstyles[args.balloon]
print(render_pony(pony, text,
balloonstyle=balloonstyle,
width=args.width or sys.maxint,
center=args.center,
centertext=args.center_text))
def qotd_server():
from socketserver import ThreadingMixIn, TCPServer, BaseRequestHandler
# Quote-Of-The-Day protocol implementation using ponysay backend
# See RFC865 ( https://tools.ietf.org/html/rfc865 ) for details.
# To prevent traffic amplification attacks we are only providing a TCP service.
class ThreadingTCPServer(ThreadingMixIn, TCPServer): pass
ponylist = list_ponies_with_quotes()
class QOTDHandler(BaseRequestHandler):
def handle(self):
pony = random.choice(ponylist)
s = render_pony(pony, random_quote(pony),
balloonstyle=balloonstyles['cowsay'],
center=True,
centertext=False,
width=120)
self.request.sendall(bytes(s, "UTF-8"))
HOST, PORT = "", 8017
server = ThreadingTCPServer((HOST, PORT), QOTDHandler)
server.serve_forever()
def termcenter():
parser = argparse.ArgumentParser(description='Center stuff on terminals')
parser.add_argument('string', nargs='*', type=str)
args = parser.parse_args()
for e in [sys.stdin] + args.string:
lines = [e] if isinstance(e, str) else e.readlines()
if lines:
width = max(map(len, map(lambda s: re.sub(r'\x1B\[[0-9;]+m|\$.*\$', '', s), lines)))
pad = int((os.get_terminal_size()[0]- width)/2)
for line in lines:
print(' '*pad + re.sub(r'\$.*\$|\n', '', line))
if __name__ == '__main__':
main()
|
