PonyMirrors/ponysay
1
0
Fork 0
mirror of https://github.com/erkin/ponysay.git synced 2025-02-19 19:04:23 +01:00
Code Issues Projects Releases Packages Wiki Activity

ponybrowser: get ponies in directory

Browse source
This commit is contained in:
Mattias Andrée 2012-11-05 18:30:59 +01:00
parent 50c44c8099
commit f38d3df3fd
2 changed files with 85 additions and 49 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
ponysay-tool.py
View file

@ -231,7 +231,25 @@ class PonysayTool():
break break
(termh, termw) = termsize (termh, termw) = termsize
ponies = [] # TODO fill ponies = set()
for ponyfile in os.listdir(ponydir):
if endswith(ponyfile, '.pony'):
ponyfile = ponyfile[:-5]
if ponyfile not in ponies:
ponies.add(ponyfile)
if restriction is not None:
oldponies = ponies
logic = Ponysay.makeRestrictionLogic(restriction)
ponies = set()
for pony in Ponysay.restrictedPonies(ponydir, logic):
if (pony not in ponies) and (pony in oldponies):
ponies.add(pony)
oldponies = ponies
ponies = list(ponies)
if len(ponies) == 0:
print('\033[1;31m%s\033[21m;39m' % 'No ponies... press Enter to exit.')
input()
panelw = Backend.len(max(ponies, key = Backend.len)) panelw = Backend.len(max(ponies, key = Backend.len))
panely = 0 panely = 0

114
ponysay.py
View file

@ -566,54 +566,10 @@ class Ponysay():
if (names is None) or (len(names) == 0): if (names is None) or (len(names) == 0):
oldponies = ponies oldponies = ponies
if self.restriction is not None: if self.restriction is not None:
table = [(get_test(cell[:cell.index('=')], logic = Ponysay.makeRestrictionLogic(self.restriction)
cell[cell.index('=') + 1:]
)
for cell in clause.lower().split('+'))
for clause in self.restriction
]
def get_test(cell):
strict = cell[0][-1] != '?'
key = cell[0][:-2 if strict else -1]
invert = cell[1][0] == '!'
value = cell[1][1 if invert else 0:]
class SITest:
def __init__(self, cellkey, cellvalue):
(self.cellkey, self.callvalue) = (key, value)
def __call__(self, has):
return False if key not in has else (value not in has[key])
class STest:
def __init__(self, cellkey, cellvalue):
(self.cellkey, self.callvalue) = (key, value)
def __call__(self, has):
return False if key not in has else (value in has[key])
class ITest:
def __init__(self, cellkey, cellvalue):
(self.cellkey, self.callvalue) = (key, value)
def __call__(self, has):
return True if key not in has else (value not in has[key])
class NTest:
def __init__(self, cellkey, cellvalue):
(self.cellkey, self.callvalue) = (key, value)
def __call__(self, has):
return True if key not in has else (value in has[key])
if strict and invert: return SITest(key, value)
if strict: return STest(key, value)
if invert: return ITest(key, value)
return NTest(key, value)
def logic(cells):
for alternative in table:
ok = True
for cell in alternative:
if not cell(cells):
ok = False
break
if ok:
return True
return False
ponies = {} ponies = {}
for ponydir in self.ponydirs: for ponydir in self.ponydirs:
for pony in self.restrictedPonies(ponydir, logic): for pony in Ponysay.restrictedPonies(ponydir, logic):
if (pony not in passed) and (pony in oldponies): if (pony not in passed) and (pony in oldponies):
ponyfile = ponydir + pony + '.pony' ponyfile = ponydir + pony + '.pony'
if oldponies[pony] == ponyfile: if oldponies[pony] == ponyfile:
@ -646,7 +602,7 @@ class Ponysay():
# ponies[pony] = ponydir + ponyfile # ponies[pony] = ponydir + ponyfile
names = list((oldponies if len(ponies) == 0 else ponies).keys()) names = list((oldponies if len(ponies) == 0 else ponies).keys())
## Select a random pony of the choosen onles ## Select a random pony of the choosen ones
pony = names[random.randrange(0, len(names))] pony = names[random.randrange(0, len(names))]
if pony not in ponies: if pony not in ponies:
if not alt: if not alt:
@ -662,6 +618,67 @@ class Ponysay():
return ponies[pony] return ponies[pony]
'''
Make restriction test logic function
@param restriction:list<string> Metadata based restrictions
@return :dict<str, str>→bool Test function
'''
@staticmethod
def makeRestrictionLogic(restriction):
table = [(get_test(cell[:cell.index('=')],
cell[cell.index('=') + 1:]
)
for cell in clause.lower().split('+'))
for clause in restriction
]
def get_test(cell):
strict = cell[0][-1] != '?'
key = cell[0][:-2 if strict else -1]
invert = cell[1][0] == '!'
value = cell[1][1 if invert else 0:]
class SITest:
def __init__(self, cellkey, cellvalue):
(self.cellkey, self.callvalue) = (key, value)
def __call__(self, has):
return False if key not in has else (value not in has[key])
class STest:
def __init__(self, cellkey, cellvalue):
(self.cellkey, self.callvalue) = (key, value)
def __call__(self, has):
return False if key not in has else (value in has[key])
class ITest:
def __init__(self, cellkey, cellvalue):
(self.cellkey, self.callvalue) = (key, value)
def __call__(self, has):
return True if key not in has else (value not in has[key])
class NTest:
def __init__(self, cellkey, cellvalue):
(self.cellkey, self.callvalue) = (key, value)
def __call__(self, has):
return True if key not in has else (value in has[key])
if strict and invert: return SITest(key, value)
if strict: return STest(key, value)
if invert: return ITest(key, value)
return NTest(key, value)
def logic(cells):
for alternative in table:
ok = True
for cell in alternative:
if not cell(cells):
ok = False
break
if ok:
return True
return False
return logic
''' '''
Get ponies that pass restriction Get ponies that pass restriction
@ -669,7 +686,8 @@ class Ponysay():
@param logic:dict<str, str>→bool Restriction test functor @param logic:dict<str, str>→bool Restriction test functor
@return :list<str> Passed ponies @return :list<str> Passed ponies
''' '''
def restrictedPonies(self, ponydir, logic): @staticmethod
def restrictedPonies(ponydir, logic):
import cPickle import cPickle
passed = [] passed = []
if os.path.exists(ponydir + 'metadata'): if os.path.exists(ponydir + 'metadata'):