Add rule api and test

This commit is contained in:
WQY\qiong
2022-10-29 19:09:34 +08:00
parent 44475cf005
commit 1109ca4b6d
5 changed files with 59 additions and 14 deletions

View File

@@ -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 .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 .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 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
View 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))

View File

@@ -4,3 +4,5 @@ create table rules
( (
rule text primary key rule text primary key
); );
insert into rules (rule) values ('');

View File

@@ -1264,36 +1264,36 @@ class TestApi:
self.leave(p) self.leave(p)
def test_control(self): def test_rule(self):
p = 'test_control' p = 'test_rule'
self.enter(p) self.enter(p)
assert get_control(p)['control'] == '' assert get_rule(p)['rule'] == ''
set_control(p, ChangeSet({'control': 'x'})) set_rule(p, ChangeSet({'rule': 'x'}))
assert get_control(p)['control'] == 'x' assert get_rule(p)['rule'] == 'x'
self.leave(p) self.leave(p)
def test_control_op(self): def test_rule_op(self):
p = 'test_control_op' p = 'test_rule_op'
self.enter(p) 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['operation'] == API_UPDATE
assert cs['type'] == 'control' assert cs['type'] == 'rule'
assert cs['control'] == 'x' assert cs['rule'] == 'x'
cs = execute_undo(p).operations[0] cs = execute_undo(p).operations[0]
assert cs['operation'] == API_UPDATE assert cs['operation'] == API_UPDATE
assert cs['type'] == 'control' assert cs['type'] == 'rule'
assert cs['control'] == '' assert cs['rule'] == ''
cs = execute_redo(p).operations[0] cs = execute_redo(p).operations[0]
assert cs['operation'] == API_UPDATE assert cs['operation'] == API_UPDATE
assert cs['type'] == 'control' assert cs['type'] == 'rule'
assert cs['control'] == 'x' assert cs['rule'] == 'x'
self.leave(p) self.leave(p)

View File

@@ -429,6 +429,21 @@ def set_control(name: str, cs: ChangeSet) -> ChangeSet:
return api.set_control(name, cs) 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] # emitter 16.[EMITTERS]
############################################################ ############################################################