forked from WesleyMPG/infeng
/
Engine.py
59 lines (51 loc) · 1.92 KB
/
Engine.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
import json
from Rule import Rule
from Value import Value
from scanner import Scanner
from parser import Parser
from Evaluator import Evaluator
class Engine(object):
def __init__(self, knowledge_file_path):
self.knowledge_base = []
self.values_table = {}
self.evaluator = Evaluator(self.values_table)
self.load_knowledge(knowledge_file_path)
def __load_file_content(self, file):
file = json.load(file)
desc = file['description']
rules = file['rules']
return desc, rules
def __get_description_info(self, item):
#return value, description
if type(item) == bool:
return item, ''
elif type(item) == str:
return '', item
elif type(item) == list:
if type(item[0]) == bool and type(item[1]) == str:
return item[0], item[1]
else:
raise Exception(f'{item} should be [bool, str].')
else:
raise Exception(f'Invalid description {item}.\
Acepted formats: bool, str, [bool, str].')
def load_descriptions(self, desc):
print('Loading descriptions...')
for key in desc.keys():
v, d = self.__get_description_info(desc[key])
self.values_table[key] = Value(key, value=v, description=d)
def load_rules(self, rules):
print('Loading rules...')
for i in rules:
rule = Rule(i)
self.knowledge_base.append(rule)
# associa uma um valor (A) às regras nas quais ele
# aparece no lado direito de uma regra
self.values_table[rule.right].add_rule(rule)
def load_knowledge(self, file_path):
with open(file_path, 'r') as file:
desc, rules = self.__load_file_content(file)
self.load_descriptions(desc)
self.load_rules(rules)
def evaluate(self, expression):
return self.evaluator.evaluate(expression)