m + a method for getting all quotes

This commit is contained in:
Mattias Andrée 2012-08-18 15:10:49 +02:00
parent feeaa47ad8
commit 73d477ce1b

View file

@ -61,7 +61,7 @@ class ponysay():
quoteshash = set()
_quotes = [item[:item.index('.')] for item in os.listdir(INSTALLDIR + '/share/ponysay/quotes/')]
for quote in _quotes:
if not quote[0] == '.':
if not quote == '':
if not quote in quoteshash:
quoteshash.add(quote)
quotes.append(quote)
@ -78,6 +78,25 @@ class ponysay():
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
'''
@ -107,6 +126,7 @@ class ponysay():
print("\n");
def print_pony(self, args):
if args.message == None:
msg = sys.stdin.read().strip()