mirror of
https://github.com/erkin/ponysay.git
synced 2025-02-07 13:36:43 +01:00
pre-class Ponysay code is ran inside if __name__ == "__main__"
This commit is contained in:
Mattias Andrée
parent
c1ed662fcc
commit
388a6ed142
1 changed files with 189 additions and 189 deletions
378
ponysay.py
378
ponysay.py
|
|
@ -2273,200 +2273,200 @@ class SpelloCorrecter(): # Naïvely and quickly proted and adapted from optimise
|
||||||
|
|
||||||
|
|
||||||
'''
|
'''
|
||||||
The user's home directory
|
Start the program from ponysay.__init__ if this is the executed file
|
||||||
'''
|
'''
|
||||||
HOME = os.environ['HOME'] if 'HOME' in os.environ else os.path.expanduser('~')
|
if __name__ == '__main__':
|
||||||
|
'''
|
||||||
|
The user's home directory
|
||||||
|
'''
|
||||||
|
HOME = os.environ['HOME'] if 'HOME' in os.environ else os.path.expanduser('~')
|
||||||
|
|
||||||
|
|
||||||
|
'''
|
||||||
|
Whether the program is execute in Linux VT (TTY)
|
||||||
|
'''
|
||||||
|
linuxvt = ('TERM' in os.environ) and (os.environ['TERM'] == 'linux')
|
||||||
|
|
||||||
|
|
||||||
|
'''
|
||||||
|
Whether the script is executed as ponythink
|
||||||
|
'''
|
||||||
|
isthink = (len(__file__) >= len('think')) and (__file__.endswith('think'))
|
||||||
|
isthink = ((len(__file__) >= len('think.py')) and (__file__.endswith('think.py'))) or isthink
|
||||||
|
|
||||||
|
|
||||||
|
'''
|
||||||
|
Whether stdin is piped
|
||||||
|
'''
|
||||||
|
pipelinein = not sys.stdin.isatty()
|
||||||
|
|
||||||
|
'''
|
||||||
|
Whether stdout is piped
|
||||||
|
'''
|
||||||
|
pipelineout = not sys.stdout.isatty()
|
||||||
|
|
||||||
|
'''
|
||||||
|
Whether stderr is piped
|
||||||
|
'''
|
||||||
|
pipelineerr = not sys.stderr.isatty()
|
||||||
|
|
||||||
|
|
||||||
|
'''
|
||||||
|
Whether KMS is used
|
||||||
|
'''
|
||||||
|
usekms = Ponysay.isUsingKMS()
|
||||||
|
|
||||||
|
|
||||||
'''
|
'''
|
||||||
Whether the program is execute in Linux VT (TTY)
|
Mode string that modifies or adds $ variables in the pony image
|
||||||
'''
|
'''
|
||||||
linuxvt = ('TERM' in os.environ) and (os.environ['TERM'] == 'linux')
|
mode = ''
|
||||||
|
|
||||||
|
|
||||||
'''
|
'''
|
||||||
Whether the script is executed as ponythink
|
The directories where pony files are stored, ttyponies/ are used if the terminal is Linux VT (also known as TTY) and not with KMS
|
||||||
'''
|
'''
|
||||||
isthink = (len(__file__) >= len('think')) and (__file__.endswith('think'))
|
appendset = set()
|
||||||
isthink = ((len(__file__) >= len('think.py')) and (__file__.endswith('think.py'))) or isthink
|
xponydirs = []
|
||||||
|
_ponydirs = [HOME + '/.local/share/ponysay/ponies/', '/usr/share/ponysay/ponies/']
|
||||||
|
for ponydir in _ponydirs:
|
||||||
'''
|
if os.path.isdir(ponydir) and (ponydir not in appendset):
|
||||||
Whether stdin is piped
|
xponydirs.append(ponydir)
|
||||||
'''
|
appendset.add(ponydir)
|
||||||
pipelinein = not sys.stdin.isatty()
|
appendset = set()
|
||||||
|
vtponydirs = []
|
||||||
'''
|
_ponydirs = [HOME + '/.local/share/ponysay/ttyponies/', '/usr/share/ponysay/ttyponies/']
|
||||||
Whether stdout is piped
|
for ponydir in _ponydirs:
|
||||||
'''
|
if os.path.isdir(ponydir) and (ponydir not in appendset):
|
||||||
pipelineout = not sys.stdout.isatty()
|
vtponydirs.append(ponydir)
|
||||||
|
appendset.add(ponydir)
|
||||||
'''
|
|
||||||
Whether stderr is piped
|
|
||||||
'''
|
'''
|
||||||
pipelineerr = not sys.stderr.isatty()
|
The directories where pony files are stored, extrattyponies/ are used if the terminal is Linux VT (also known as TTY) and not with KMS
|
||||||
|
'''
|
||||||
|
appendset = set()
|
||||||
'''
|
extraxponydirs = []
|
||||||
Whether KMS is used
|
_extraponydirs = [HOME + '/.local/share/ponysay/extraponies/', '/usr/share/ponysay/extraponies/']
|
||||||
'''
|
for extraponydir in _extraponydirs:
|
||||||
usekms = Ponysay.isUsingKMS()
|
if os.path.isdir(extraponydir) and (extraponydir not in appendset):
|
||||||
|
extraxponydirs.append(extraponydir)
|
||||||
|
appendset.add(extraponydir)
|
||||||
'''
|
appendset = set()
|
||||||
Mode string that modifies or adds $ variables in the pony image
|
extravtponydirs = []
|
||||||
'''
|
_extraponydirs = [HOME + '/.local/share/ponysay/extrattyponies/', '/usr/share/ponysay/extrattyponies/']
|
||||||
mode = ''
|
for extraponydir in _extraponydirs:
|
||||||
|
if os.path.isdir(extraponydir) and (extraponydir not in appendset):
|
||||||
|
extravtponydirs.append(extraponydir)
|
||||||
'''
|
appendset.add(extraponydir)
|
||||||
The directories where pony files are stored, ttyponies/ are used if the terminal is Linux VT (also known as TTY) and not with KMS
|
|
||||||
'''
|
|
||||||
appendset = set()
|
'''
|
||||||
xponydirs = []
|
The directories where quotes files are stored
|
||||||
_ponydirs = [HOME + '/.local/share/ponysay/ponies/', '/usr/share/ponysay/ponies/']
|
'''
|
||||||
for ponydir in _ponydirs:
|
appendset = set()
|
||||||
if os.path.isdir(ponydir) and (ponydir not in appendset):
|
quotedirs = []
|
||||||
xponydirs.append(ponydir)
|
_quotedirs = [HOME + '/.local/share/ponysay/quotes/', '/usr/share/ponysay/quotes/']
|
||||||
appendset.add(ponydir)
|
for quotedir in _quotedirs:
|
||||||
appendset = set()
|
if os.path.isdir(quotedir) and (quotedir not in appendset):
|
||||||
vtponydirs = []
|
quotedirs.append(quotedir)
|
||||||
_ponydirs = [HOME + '/.local/share/ponysay/ttyponies/', '/usr/share/ponysay/ttyponies/']
|
appendset.add(quotedir)
|
||||||
for ponydir in _ponydirs:
|
|
||||||
if os.path.isdir(ponydir) and (ponydir not in appendset):
|
|
||||||
vtponydirs.append(ponydir)
|
'''
|
||||||
appendset.add(ponydir)
|
The directories where balloon style files are stored
|
||||||
|
'''
|
||||||
|
appendset = set()
|
||||||
'''
|
balloondirs = []
|
||||||
The directories where pony files are stored, extrattyponies/ are used if the terminal is Linux VT (also known as TTY) and not with KMS
|
_balloondirs = [HOME + '/.local/share/ponysay/balloons/', '/usr/share/ponysay/balloons/']
|
||||||
'''
|
for balloondir in _balloondirs:
|
||||||
appendset = set()
|
if os.path.isdir(balloondir) and (balloondir not in appendset):
|
||||||
extraxponydirs = []
|
balloondirs.append(balloondir)
|
||||||
_extraponydirs = [HOME + '/.local/share/ponysay/extraponies/', '/usr/share/ponysay/extraponies/']
|
appendset.add(balloondir)
|
||||||
for extraponydir in _extraponydirs:
|
|
||||||
if os.path.isdir(extraponydir) and (extraponydir not in appendset):
|
|
||||||
extraxponydirs.append(extraponydir)
|
'''
|
||||||
appendset.add(extraponydir)
|
ucsmap files
|
||||||
appendset = set()
|
'''
|
||||||
extravtponydirs = []
|
appendset = set()
|
||||||
_extraponydirs = [HOME + '/.local/share/ponysay/extrattyponies/', '/usr/share/ponysay/extrattyponies/']
|
ucsmaps = []
|
||||||
for extraponydir in _extraponydirs:
|
_ucsmaps = [HOME + '/.local/share/ponysay/ucsmap', '/usr/share/ponysay/ucsmap']
|
||||||
if os.path.isdir(extraponydir) and (extraponydir not in appendset):
|
for ucsmap in _ucsmaps:
|
||||||
extravtponydirs.append(extraponydir)
|
if os.path.isdir(ucsmap) and (ucsmap not in appendset):
|
||||||
appendset.add(extraponydir)
|
ucsmaps.append(ucsmap)
|
||||||
|
appendset.add(ucsmap)
|
||||||
|
|
||||||
'''
|
|
||||||
The directories where quotes files are stored
|
|
||||||
'''
|
usage_saythink = '\033[34;1m(ponysay | ponythink)\033[21;39m'
|
||||||
appendset = set()
|
usage_common = '[-c] [-W\033[4mCOLUMN\033[24m] [-b\033[4mSTYLE\033[24m]'
|
||||||
quotedirs = []
|
usage_listhelp = '(-l | -L | -B | +l | +L | -v | -h)'
|
||||||
_quotedirs = [HOME + '/.local/share/ponysay/quotes/', '/usr/share/ponysay/quotes/']
|
usage_file = '[-f\033[4mPONY\033[24m]* [[--] \033[4mmessage\033[24m]'
|
||||||
for quotedir in _quotedirs:
|
usage_xfile = '(+f\033[4mPONY\033[24m)* [[--] \033[4mmessage\033[24m]'
|
||||||
if os.path.isdir(quotedir) and (quotedir not in appendset):
|
usage_quote = '(-q \033[4mPONY\033[24m)*'
|
||||||
quotedirs.append(quotedir)
|
|
||||||
appendset.add(quotedir)
|
usage = '%s %s\n%s %s %s\n%s %s %s\n%s %s %s' % (usage_saythink, usage_listhelp,
|
||||||
|
usage_saythink, usage_common, usage_file,
|
||||||
|
usage_saythink, usage_common, usage_xfile,
|
||||||
'''
|
usage_saythink, usage_common, usage_quote)
|
||||||
The directories where balloon style files are stored
|
|
||||||
'''
|
usage = usage.replace('\033[', '\0')
|
||||||
appendset = set()
|
for sym in ('[', ']', '(', ')', '|', '...', '*'):
|
||||||
balloondirs = []
|
usage = usage.replace(sym, '\033[2m' + sym + '\033[22m')
|
||||||
_balloondirs = [HOME + '/.local/share/ponysay/balloons/', '/usr/share/ponysay/balloons/']
|
usage = usage.replace('\0', '\033[')
|
||||||
for balloondir in _balloondirs:
|
|
||||||
if os.path.isdir(balloondir) and (balloondir not in appendset):
|
'''
|
||||||
balloondirs.append(balloondir)
|
Argument parsing
|
||||||
appendset.add(balloondir)
|
'''
|
||||||
|
opts = ArgParser(program = 'ponythink' if isthink else 'ponysay',
|
||||||
|
description = 'cowsay reimplemention for ponies',
|
||||||
'''
|
usage = usage,
|
||||||
ucsmap files
|
longdescription =
|
||||||
'''
|
|
||||||
appendset = set()
|
|
||||||
ucsmaps = []
|
|
||||||
_ucsmaps = [HOME + '/.local/share/ponysay/ucsmap', '/usr/share/ponysay/ucsmap']
|
|
||||||
for ucsmap in _ucsmaps:
|
|
||||||
if os.path.isdir(ucsmap) and (ucsmap not in appendset):
|
|
||||||
ucsmaps.append(ucsmap)
|
|
||||||
appendset.add(ucsmap)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
usage_saythink = '\033[34;1m(ponysay | ponythink)\033[21;39m'
|
|
||||||
usage_common = '[-c] [-W\033[4mCOLUMN\033[24m] [-b\033[4mSTYLE\033[24m]'
|
|
||||||
usage_listhelp = '(-l | -L | -B | +l | +L | -v | -h)'
|
|
||||||
usage_file = '[-f\033[4mPONY\033[24m]* [[--] \033[4mmessage\033[24m]'
|
|
||||||
usage_xfile = '(+f\033[4mPONY\033[24m)* [[--] \033[4mmessage\033[24m]'
|
|
||||||
usage_quote = '(-q \033[4mPONY\033[24m)*'
|
|
||||||
|
|
||||||
usage = '%s %s\n%s %s %s\n%s %s %s\n%s %s %s' % (usage_saythink, usage_listhelp,
|
|
||||||
usage_saythink, usage_common, usage_file,
|
|
||||||
usage_saythink, usage_common, usage_xfile,
|
|
||||||
usage_saythink, usage_common, 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 reimplemention for ponies',
|
|
||||||
usage = usage,
|
|
||||||
longdescription =
|
|
||||||
'''Ponysay displays an image of a pony saying some text provided by the user.
|
'''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
|
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
|
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.''')
|
run `man ponysay`. Ponysay has so much more to offer than described here.''')
|
||||||
|
|
||||||
opts.add_argumentless(['--quoters'])
|
opts.add_argumentless(['--quoters'])
|
||||||
opts.add_argumentless(['--onelist'])
|
opts.add_argumentless(['--onelist'])
|
||||||
opts.add_argumentless(['++onelist'])
|
opts.add_argumentless(['++onelist'])
|
||||||
|
|
||||||
opts.add_argumentless(['-X', '--256-colours', '--256colours', '--x-colours'])
|
opts.add_argumentless(['-X', '--256-colours', '--256colours', '--x-colours'])
|
||||||
opts.add_argumentless(['-V', '--tty-colours', '--ttycolours', '--vt-colours'])
|
opts.add_argumentless(['-V', '--tty-colours', '--ttycolours', '--vt-colours'])
|
||||||
opts.add_argumentless(['-K', '--kms-colours', '--kmscolours'])
|
opts.add_argumentless(['-K', '--kms-colours', '--kmscolours'])
|
||||||
|
|
||||||
opts.add_argumented(['+c', '--colour'], arg = 'COLOUR')
|
opts.add_argumented(['+c', '--colour'], arg = 'COLOUR')
|
||||||
opts.add_argumented(['--colour-bubble', '--colour-balloon'], arg = 'COLOUR')
|
opts.add_argumented(['--colour-bubble', '--colour-balloon'], arg = 'COLOUR')
|
||||||
opts.add_argumented(['--colour-link'], arg = 'COLOUR')
|
opts.add_argumented(['--colour-link'], arg = 'COLOUR')
|
||||||
opts.add_argumented(['--colour-msg', '--colour-message'], arg = 'COLOUR')
|
opts.add_argumented(['--colour-msg', '--colour-message'], arg = 'COLOUR')
|
||||||
opts.add_argumented(['--colour-pony'], arg = 'COLOUR')
|
opts.add_argumented(['--colour-pony'], arg = 'COLOUR')
|
||||||
opts.add_argumented(['--colour-wrap', '--colour-hyphen'], arg = 'COLOUR')
|
opts.add_argumented(['--colour-wrap', '--colour-hyphen'], arg = 'COLOUR')
|
||||||
|
|
||||||
opts.add_argumentless(['-h', '--help'], help = 'Print this help message.')
|
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(['-v', '--version'], help = 'Print the version of the program.')
|
||||||
opts.add_argumentless(['-l', '--list'], help = 'List pony names.')
|
opts.add_argumentless(['-l', '--list'], help = 'List pony names.')
|
||||||
opts.add_argumentless(['-L', '--symlist', '--altlist'], help = 'List pony names with alternatives.')
|
opts.add_argumentless(['-L', '--symlist', '--altlist'], help = 'List pony names with alternatives.')
|
||||||
opts.add_argumentless(['+l', '++list'], help = 'List non-MLP:FiM pony names.')
|
opts.add_argumentless(['+l', '++list'], help = 'List non-MLP:FiM pony names.')
|
||||||
opts.add_argumentless(['+L', '++symlist', '++altlist'], help = 'List non-MLP:FiM pony names with alternatives.')
|
opts.add_argumentless(['+L', '++symlist', '++altlist'], help = 'List non-MLP:FiM pony names with alternatives.')
|
||||||
opts.add_argumentless(['-A', '--all'], help = 'List all pony names.')
|
opts.add_argumentless(['-A', '--all'], help = 'List all pony names.')
|
||||||
opts.add_argumentless(['+A', '++all', '--symall', '--altall'], help = 'List all pony names with alternatives.')
|
opts.add_argumentless(['+A', '++all', '--symall', '--altall'], help = 'List all pony names with alternatives.')
|
||||||
opts.add_argumentless(['-B', '--bubblelist', '--balloonlist'], help = 'List balloon styles.')
|
opts.add_argumentless(['-B', '--bubblelist', '--balloonlist'], help = 'List balloon styles.')
|
||||||
opts.add_argumentless(['-c', '--compact'], help = 'Compress messages.')
|
opts.add_argumentless(['-c', '--compact'], help = 'Compress messages.')
|
||||||
opts.add_argumentless(['-o', '--pony-only', '--ponyonly'], help = 'Print only the pony.')
|
opts.add_argumentless(['-o', '--pony-only', '--ponyonly'], help = 'Print only the pony.')
|
||||||
opts.add_argumented( ['-W', '--wrap'], arg = 'COLUMN', help = 'Specify column where the message should be wrapped.')
|
opts.add_argumented( ['-W', '--wrap'], arg = 'COLUMN', help = 'Specify column where the message should be wrapped.')
|
||||||
opts.add_argumented( ['-b', '--bubble', '--balloon'], arg = 'STYLE', help = 'Select a balloon style.')
|
opts.add_argumented( ['-b', '--bubble', '--balloon'], arg = 'STYLE', help = 'Select a balloon style.')
|
||||||
opts.add_argumented( ['-f', '--file', '--pony'], arg = 'PONY', help = 'Select a pony.\nEither a file name or a pony name.')
|
opts.add_argumented( ['-f', '--file', '--pony'], arg = 'PONY', help = 'Select a pony.\nEither a file name or a pony name.')
|
||||||
opts.add_argumented( ['+f', '++file', '++pony'], arg = 'PONY', help = 'Select a non-MLP:FiM pony.')
|
opts.add_argumented( ['+f', '++file', '++pony'], arg = 'PONY', help = 'Select a non-MLP:FiM pony.')
|
||||||
opts.add_argumented( ['-q', '--quote'], arg = 'PONY', help = 'Select a pony which will quote herself.')
|
opts.add_argumented( ['-q', '--quote'], arg = 'PONY', help = 'Select a pony which will quote herself.')
|
||||||
opts.add_variadic( ['--f', '--files', '--ponies'], arg = 'PONY')
|
opts.add_variadic( ['--f', '--files', '--ponies'], arg = 'PONY')
|
||||||
opts.add_variadic( ['++f', '++files', '++ponies'], arg = 'PONY')
|
opts.add_variadic( ['++f', '++files', '++ponies'], arg = 'PONY')
|
||||||
opts.add_variadic( ['--q', '--quotes'], arg = 'PONY')
|
opts.add_variadic( ['--q', '--quotes'], arg = 'PONY')
|
||||||
|
|
||||||
'''
|
'''
|
||||||
Whether at least one unrecognised option was used
|
Whether at least one unrecognised option was used
|
||||||
'''
|
'''
|
||||||
unrecognised = not opts.parse()
|
unrecognised = not opts.parse()
|
||||||
|
|
||||||
|
|
||||||
|
## Start
|
||||||
'''
|
|
||||||
Start the program from ponysay.__init__ if this is the executed file
|
|
||||||
'''
|
|
||||||
if __name__ == '__main__':
|
|
||||||
Ponysay(opts)
|
Ponysay(opts)
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue