lc_ldap/filter2.py

67 lines
1.4 KiB
Python

#!/usr/bin/env python
"""Plus rapide"""
import pyparsing
def simplify(l):
if not isinstance(l, list):
return l
if len(l) == 1:
return simplify(l[0])
else:
return [simplify(i) for i in l]
def toprefix(l):
if not isinstance(l, list):
return l
op=l[1]
args=[toprefix(i) for i in l if i!=op]
return [op]+args
def prioritize(l):
if not isinstance(l, list):
return l
l=simplify(l)
for c in ['!=', '=', '&', '|']:
i=0
while i<len(l):
if l[i] == c:
tmp=[prioritize(l[i-1]), l[i], prioritize(l[i+1])]
l[i-1]=tmp
del(l[i+1])
del(l[i])
i-=1
i+=1
return simplify(l)
def collapse(l):
"""retire les parentheses inutiles"""
if not isinstance(l, list):
return l
if not isinstance(l[1], list):
return [l[0], collapse(l[1]), collapse(l[2])]
if l[0] == l[1][0]:
return [l[0], collapse(l[1][1]), collapse(l[1][2]), collapse(l[2])]
else:
return [l[0], collapse(l[1]), collapse(l[2])]
def toldapfilter(l):
if not isinstance(l, list):
return l
op=l[0]
if op == "=":
return "%s=%s" % (l[1], l[2])
elif op == "!=":
return "!(%s=%s)" % (l[1], l[2])
return op + ''.join(['(%s)' % toldapfilter(i) for i in l[1:]])
txt = "".join(c for c in pyparsing.printables if c not in '()&|!=')
expr = pyparsing.nestedExpr("(", ")", pyparsing.Word(txt) | pyparsing.oneOf("& | != ="))
def human_to_ldap(data):
data='(%s)' % data
l=expr.parseString(data).asList()
return "(%s)" % toldapfilter(collapse(toprefix(prioritize(l))))