Add rule api and test
This commit is contained in:
@@ -52,6 +52,8 @@ from .s12_curves import get_curve_schema, get_curve, set_curve
|
||||
|
||||
from .s13_controls import get_control_schema, get_control, set_control
|
||||
|
||||
from .s14_rules import get_rule_schema, get_rule, set_rule
|
||||
|
||||
from .s16_emitters import get_emitter_schema, get_emitter, set_emitter
|
||||
|
||||
from .s21_times import TIME_STATISTIC_NONE, TIME_STATISTIC_AVERAGED, TIME_STATISTIC_MINIMUM, TIME_STATISTIC_MAXIMUM, TIME_STATISTIC_RANGE
|
||||
|
||||
26
api/s14_rules.py
Normal file
26
api/s14_rules.py
Normal file
@@ -0,0 +1,26 @@
|
||||
from .operation import *
|
||||
|
||||
|
||||
def get_rule_schema(name: str) -> dict[str, dict[str, Any]]:
|
||||
return { 'rule' : {'type': 'str' , 'optional': False , 'readonly': False} }
|
||||
|
||||
|
||||
def get_rule(name: str) -> dict[str, Any]:
|
||||
e = read(name, f"select * from rules")
|
||||
return { 'rule': e['rule'] }
|
||||
|
||||
|
||||
def set_rule_cache(name: str, cs: ChangeSet) -> SqlChangeSet:
|
||||
old = get_rule(name)
|
||||
|
||||
redo_sql = f"update rules set rule = '{cs.operations[0]['rule']}' where rule = '{old['rule']}';"
|
||||
undo_sql = f"update rules set rule = '{old['rule']}' where rule = '{cs.operations[0]['rule']}';"
|
||||
|
||||
redo_cs = g_update_prefix | { 'type': 'rule', 'rule': cs.operations[0]['rule'] }
|
||||
undo_cs = g_update_prefix | { 'type': 'rule', 'rule': old['rule'] }
|
||||
|
||||
return SqlChangeSet(redo_sql, undo_sql, redo_cs, undo_cs)
|
||||
|
||||
|
||||
def set_rule(name: str, cs: ChangeSet) -> ChangeSet:
|
||||
return execute_command(name, set_rule_cache(name, cs))
|
||||
@@ -4,3 +4,5 @@ create table rules
|
||||
(
|
||||
rule text primary key
|
||||
);
|
||||
|
||||
insert into rules (rule) values ('');
|
||||
|
||||
@@ -1264,36 +1264,36 @@ class TestApi:
|
||||
self.leave(p)
|
||||
|
||||
|
||||
def test_control(self):
|
||||
p = 'test_control'
|
||||
def test_rule(self):
|
||||
p = 'test_rule'
|
||||
self.enter(p)
|
||||
|
||||
assert get_control(p)['control'] == ''
|
||||
assert get_rule(p)['rule'] == ''
|
||||
|
||||
set_control(p, ChangeSet({'control': 'x'}))
|
||||
assert get_control(p)['control'] == 'x'
|
||||
set_rule(p, ChangeSet({'rule': 'x'}))
|
||||
assert get_rule(p)['rule'] == 'x'
|
||||
|
||||
self.leave(p)
|
||||
|
||||
|
||||
def test_control_op(self):
|
||||
p = 'test_control_op'
|
||||
def test_rule_op(self):
|
||||
p = 'test_rule_op'
|
||||
self.enter(p)
|
||||
|
||||
cs = set_control(p, ChangeSet({'control': 'x'})).operations[0]
|
||||
cs = set_rule(p, ChangeSet({'rule': 'x'})).operations[0]
|
||||
assert cs['operation'] == API_UPDATE
|
||||
assert cs['type'] == 'control'
|
||||
assert cs['control'] == 'x'
|
||||
assert cs['type'] == 'rule'
|
||||
assert cs['rule'] == 'x'
|
||||
|
||||
cs = execute_undo(p).operations[0]
|
||||
assert cs['operation'] == API_UPDATE
|
||||
assert cs['type'] == 'control'
|
||||
assert cs['control'] == ''
|
||||
assert cs['type'] == 'rule'
|
||||
assert cs['rule'] == ''
|
||||
|
||||
cs = execute_redo(p).operations[0]
|
||||
assert cs['operation'] == API_UPDATE
|
||||
assert cs['type'] == 'control'
|
||||
assert cs['control'] == 'x'
|
||||
assert cs['type'] == 'rule'
|
||||
assert cs['rule'] == 'x'
|
||||
|
||||
self.leave(p)
|
||||
|
||||
|
||||
15
tjnetwork.py
15
tjnetwork.py
@@ -429,6 +429,21 @@ def set_control(name: str, cs: ChangeSet) -> ChangeSet:
|
||||
return api.set_control(name, cs)
|
||||
|
||||
|
||||
############################################################
|
||||
# rule 12.[CONTROLS]
|
||||
############################################################
|
||||
|
||||
def get_rule_schema(name: str) -> dict[str, dict[str, Any]]:
|
||||
return api.get_rule_schema(name)
|
||||
|
||||
def get_rule(name: str) -> dict[str, Any]:
|
||||
return api.get_rule(name)
|
||||
|
||||
# example: set_rule(p, ChangeSet({'rule': 'x'}))
|
||||
def set_rule(name: str, cs: ChangeSet) -> ChangeSet:
|
||||
return api.set_rule(name, cs)
|
||||
|
||||
|
||||
############################################################
|
||||
# emitter 16.[EMITTERS]
|
||||
############################################################
|
||||
|
||||
Reference in New Issue
Block a user