implemented the rule engine
commit
d0d17a7fb6
|
|
@ -0,0 +1,104 @@
|
|||
|
||||
# Created by https://www.gitignore.io/api/python
|
||||
|
||||
### Python ###
|
||||
# Byte-compiled / optimized / DLL files
|
||||
__pycache__/
|
||||
*.py[cod]
|
||||
*$py.class
|
||||
|
||||
# C extensions
|
||||
*.so
|
||||
|
||||
# Distribution / packaging
|
||||
.Python
|
||||
env/
|
||||
build/
|
||||
develop-eggs/
|
||||
dist/
|
||||
downloads/
|
||||
eggs/
|
||||
.eggs/
|
||||
lib/
|
||||
lib64/
|
||||
parts/
|
||||
sdist/
|
||||
var/
|
||||
wheels/
|
||||
*.egg-info/
|
||||
.installed.cfg
|
||||
*.egg
|
||||
|
||||
# PyInstaller
|
||||
# Usually these files are written by a python script from a template
|
||||
# before PyInstaller builds the exe, so as to inject date/other infos into it.
|
||||
*.manifest
|
||||
*.spec
|
||||
|
||||
# Installer logs
|
||||
pip-log.txt
|
||||
pip-delete-this-directory.txt
|
||||
|
||||
# Unit test / coverage reports
|
||||
htmlcov/
|
||||
.tox/
|
||||
.coverage
|
||||
.coverage.*
|
||||
.cache
|
||||
nosetests.xml
|
||||
coverage.xml
|
||||
*,cover
|
||||
.hypothesis/
|
||||
|
||||
# Translations
|
||||
*.mo
|
||||
*.pot
|
||||
|
||||
# Django stuff:
|
||||
*.log
|
||||
local_settings.py
|
||||
|
||||
# Flask stuff:
|
||||
instance/
|
||||
.webassets-cache
|
||||
|
||||
# Scrapy stuff:
|
||||
.scrapy
|
||||
|
||||
# Sphinx documentation
|
||||
docs/_build/
|
||||
|
||||
# PyBuilder
|
||||
target/
|
||||
|
||||
# Jupyter Notebook
|
||||
.ipynb_checkpoints
|
||||
|
||||
# pyenv
|
||||
.python-version
|
||||
|
||||
# celery beat schedule file
|
||||
celerybeat-schedule
|
||||
|
||||
# SageMath parsed files
|
||||
*.sage.py
|
||||
|
||||
# dotenv
|
||||
.env
|
||||
|
||||
# virtualenv
|
||||
.venv
|
||||
venv/
|
||||
ENV/
|
||||
|
||||
# Spyder project settings
|
||||
.spyderproject
|
||||
.spyproject
|
||||
|
||||
# Rope project settings
|
||||
.ropeproject
|
||||
|
||||
# mkdocs documentation
|
||||
/site
|
||||
|
||||
# End of https://www.gitignore.io/api/python
|
||||
Binary file not shown.
|
|
@ -0,0 +1,111 @@
|
|||
import json
|
||||
import datetime
|
||||
|
||||
class Rule(object):
|
||||
"""docstring for Rule."""
|
||||
def __init__(self, rule_line,lineno):
|
||||
super(Rule, self).__init__()
|
||||
self.pattern,constraint = [x.strip() for x in rule_line.split("->")]
|
||||
space_sep = constraint.split()
|
||||
(self.value_type,op),const = space_sep[:2]," ".join(space_sep[2:])
|
||||
self.const = get_const_for_type(self.value_type,const)
|
||||
self.neg = True if op[0] == "!" else False
|
||||
self.op = op[1:] if self.neg else op
|
||||
self.lineno = lineno
|
||||
|
||||
def __repr__(self):
|
||||
neg_val = "not" if self.neg else ""
|
||||
rule_val = [self.pattern,self.value_type,neg_val,self.op,self.const]
|
||||
return "{}:{} {} {} {}".format(*rule_val)
|
||||
|
||||
def apply_neg(self,x):
|
||||
return not x if self.neg else x
|
||||
|
||||
def apply(self,signal):
|
||||
if signal.signal == self.pattern:
|
||||
if signal.value_type != self.value_type:
|
||||
return True
|
||||
if self.op == ">":
|
||||
return self.apply_neg(signal.value > self.const)
|
||||
if self.op == "<":
|
||||
return self.apply_neg(signal.value < self.const)
|
||||
if self.op == "=":
|
||||
return self.apply_neg(signal.value == self.const)
|
||||
return True
|
||||
|
||||
@classmethod
|
||||
def rule_engine_from(cls,rule_str):
|
||||
return Rule.rule_engine_gen([cls(rule_str,1)])
|
||||
|
||||
@staticmethod
|
||||
def rule_engine_gen(rule_list):
|
||||
def check_rules(signal):
|
||||
for rule in rule_list:
|
||||
if not rule.apply(signal):
|
||||
return (False,rule)
|
||||
return (True,None)
|
||||
return check_rules
|
||||
|
||||
@classmethod
|
||||
def read_rules(cls,filename):
|
||||
rule_list = []
|
||||
with open(filename,"r") as rule_file:
|
||||
lineno = 1
|
||||
for line in rule_file.readlines():
|
||||
rulestr = line.strip()
|
||||
if rulestr != "" and not rulestr.startswith("#"):
|
||||
rule_list.append(cls(rulestr,lineno))
|
||||
lineno+=1
|
||||
return Rule.rule_engine_gen(rule_list)
|
||||
|
||||
date_parse = lambda x : datetime.datetime.strptime(x, "%Y-%m-%d %H:%M:%S")
|
||||
def get_const_for_type(value_type,value_data):
|
||||
if value_type == "Integer":
|
||||
value = int(float(value_data))
|
||||
elif value_type == "String":
|
||||
value = str(value_data)
|
||||
elif value_type == "Datetime":
|
||||
if value_data == "now":
|
||||
value = datetime.datetime.now()
|
||||
else:
|
||||
value = date_parse(value_data)
|
||||
else:
|
||||
value = "invalid"
|
||||
return value
|
||||
|
||||
class SignalItem(object):
|
||||
"""docstring for SignalItem."""
|
||||
def __init__(self, sig_dict):
|
||||
super(SignalItem, self).__init__()
|
||||
self.signal = sig_dict["signal"]
|
||||
self.value_type = sig_dict["value_type"]
|
||||
self.value = get_const_for_type(self.value_type,sig_dict["value"])
|
||||
def __repr__(self):
|
||||
return self.signal+":"+str(self.value)
|
||||
|
||||
def load_sigs(filename):
|
||||
with open(filename,"r") as sig_file:
|
||||
return json.load(sig_file)
|
||||
|
||||
def validate_sigs(signals,rule_engine):
|
||||
invalids = []
|
||||
index = 0
|
||||
for sig in signals:
|
||||
sig_obj = SignalItem(sig)
|
||||
index+=1
|
||||
status,rule = rule_engine(sig_obj)
|
||||
if not status:
|
||||
invalids.append((sig_obj,rule,index))
|
||||
return invalids
|
||||
|
||||
def main():
|
||||
check_rules = Rule.read_rules("./rules.txt")
|
||||
signals = load_sigs("./raw_data.json")
|
||||
invalids = validate_sigs(signals,check_rules)
|
||||
invalid_idx = [i[2] for i in invalids]
|
||||
print invalid_idx
|
||||
for sig_obj,rule,index in invalids:
|
||||
print index,sig_obj," Cause Rule ",rule.lineno
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
|
@ -0,0 +1,41 @@
|
|||
import unittest
|
||||
import quartic_rule
|
||||
|
||||
class QuarticTest(unittest.TestCase):
|
||||
"""docstring for QuarticTest."""
|
||||
|
||||
def setUp(self):
|
||||
self.signals = quartic_rule.load_sigs("./raw_data.json")
|
||||
pass
|
||||
|
||||
def test_rules(self):
|
||||
check_rules = quartic_rule.Rule.read_rules("./rules.txt")
|
||||
invalids = quartic_rule.validate_sigs(self.signals,check_rules)
|
||||
invalid_idx = [i[2] for i in invalids]
|
||||
expected = [16 ,26 ,37 ,77 ,88 ,96 ,155,172,190]
|
||||
self.assertEqual(invalid_idx ,expected)
|
||||
|
||||
|
||||
def test_rules_test(self):
|
||||
check_rules = quartic_rule.Rule.read_rules("./rules_test.txt")
|
||||
invalids = quartic_rule.validate_sigs(self.signals,check_rules)
|
||||
invalid_idx = [i[2] for i in invalids]
|
||||
expected = [1, 9, 45, 53, 85, 109, 110, 119, 122, 142, 152, 199]
|
||||
self.assertEqual(invalid_idx ,expected)
|
||||
|
||||
def test_rule1(self):
|
||||
check_rules = quartic_rule.Rule.rule_engine_from("ATL3 -> Datetime !> now")
|
||||
invalids = quartic_rule.validate_sigs(self.signals,check_rules)
|
||||
invalid_idx = [i[2] for i in invalids]
|
||||
expected = [26, 37, 96, 172, 190]
|
||||
self.assertEqual(invalid_idx ,expected)
|
||||
|
||||
def test_rule2(self):
|
||||
check_rules = quartic_rule.Rule.rule_engine_from("ATL2 -> Datetime != 2017-05-13 06:22:35")
|
||||
invalids = quartic_rule.validate_sigs(self.signals,check_rules)
|
||||
invalid_idx = [i[2] for i in invalids]
|
||||
expected = [9]
|
||||
self.assertEqual(invalid_idx ,expected)
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
File diff suppressed because one or more lines are too long
|
|
@ -0,0 +1,3 @@
|
|||
ATL1 -> Integer !> 240
|
||||
ATL2 -> String != LOW
|
||||
ATL3 -> Datetime !> now
|
||||
|
|
@ -0,0 +1,3 @@
|
|||
ATL1 -> Integer !> 12
|
||||
ATL2 -> String != HIGH
|
||||
ATL2 -> Datetime !> 2017-05-13 06:22:34
|
||||
Loading…
Reference in New Issue