/
query.py
executable file
·133 lines (111 loc) · 3.09 KB
/
query.py
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
import re
import time
from var import Var
class Query(object):
def __init__(self, left):
self.left = left
self.priority = {}
self.child = 0
self.resolver = self.createResolver(self.left)
def op_add(self, l, r, varMap):
return (l.op_add(r, varMap))
def op_or(self, l, r, varMap):
return (l.op_or(r, varMap))
def op_xor(self, l, r, varMap):
return (l.op_xor(r, varMap))
def opn_add(self, l, r, varMap):
result = Var(l.getName())
if ( l.getValue() == 1 and r.getValue() == 0 ):
result.setValue( 1 )
elif ( l.getValue() == 0 or r.getValue() == 1 ):
result.setValue( 0 )
elif ( l.getValue() == -1 or r.getValue() == -1 ):
result.setValue( -1 )
else:
result.setValue( 0 )
return (result)
def opn_or(self, l, r, varMap):
result = Var(l.getName())
if ( l.getValue() == 1 or r.getValue() == 0 ):
result.setValue( 1 )
elif ( l.getValue() == 0 or r.getValue() == 1 ):
result.setValue( 0 )
elif ( l.getValue() == -1 and r.getValue() == -1 ):
result.setValue( -1 )
else:
result.setValue( 0 )
return (result)
def opn_xor(self, l, r, varMap):
result = Var(l.getName())
if ( l.getValue() == -1 or r.getValue() == -1 ):
result.setValue( -1 )
elif ( l.getValue() == 1 ^ r.getValue() == 0 ):
result.setValue( 1 )
else:
result.setValue( 0 )
return (result)
def isOperator(self, string):
operator = ['+', '|', '&', '^']
if (string in operator):
return (True)
return (False)
def createResolver( self, result ):
resultold = result
i = 0
regexp = "(\((!?([A-Z]|\?)[+^|]){1,}!?([A-Z]|\?)\))"
while ( result != re.sub(regexp, "?", result)):
result = re.sub(regexp, "?", result)
tmp_final = result
resolver = {result:[]}
result = resultold
while ( result != re.subn(regexp, "?", result, count=1)[0] ):
pos = re.search(regexp, result)
tmp = result[pos.start() + 1:pos.end() - 1]
result = re.subn(regexp, "?", result, count=1)[0]
resolver[tmp_final].append(tmp)
return resolver
def run(self, varMap):
return (self.parse_resolver(varMap, self.resolver).getValue())
def parse_resolver(self, varMap, resolver):
tmp = {}
tmpVarMap = varMap
elem = None
self.child = 0
for (key, value) in resolver.items():
if (elem == None):
elem = key
for (i, query) in enumerate(value):
tmp[str(self.child)] = self.calcul(query, varMap)
self.child += 1
for ( i, var ) in tmp.items():
elem = str.replace(elem, "?", var.getName(), 1)
tmpVarMap[var.getName()] = var
return (self.calcul(elem, tmpVarMap))
def calcul(self, query, varMap):
l = Var(str(self.child))
# print (str(self.child))
l.setValue(1)
opposite = 0
ptr = {
"+": self.op_add,
"|": self.op_or,
"^": self.op_xor,
"!+": self.opn_add,
"!|": self.opn_or,
"!^": self.opn_xor
}
op = "+"
for x in query:
if ( op == None and self.isOperator(x) == True ):
op = x
elif ( self.isOperator(x) == False ):
if ( x != "!" ):
if ( opposite == 1 ):
op = "!" + op
val = ptr[op](l, varMap[x], varMap).getValue()
l.setValue(val)
opposite = 0
op = None
else :
opposite = 1
return (l)