104 lines
4.6 KiB
Python
104 lines
4.6 KiB
Python
from .database import *
|
|
|
|
|
|
def get_scada_device_data_schema(name: str) -> dict[str, dict[str, Any]]:
|
|
return { 'device_id' : {'type': 'str' , 'optional': False , 'readonly': True },
|
|
'data' : {'type': 'list' , 'optional': False , 'readonly': False,
|
|
'element': { 'time' : {'type': 'str' , 'optional': False , 'readonly': False },
|
|
'value' : {'type': 'float' , 'optional': False , 'readonly': False } }}}
|
|
|
|
|
|
def get_scada_device_data(name: str, device_id: str) -> dict[str, Any]:
|
|
sds = read_all(name, f"select * from scada_device_data where device_id = '{device_id}' order by time")
|
|
ds = []
|
|
for r in sds:
|
|
ds.append({ 'time': str(r['time']), 'value': float(r['value']) })
|
|
return { 'device_id': device_id, 'data': ds }
|
|
|
|
|
|
def set_scada_device_data_cmd(name: str, cs: ChangeSet) -> DbChangeSet:
|
|
device_id = cs.operations[0]['device_id']
|
|
|
|
old = get_scada_device_data(name, device_id)
|
|
new = { 'device_id': device_id, 'data': [] }
|
|
|
|
f_device_id = f"'{device_id}'"
|
|
|
|
# TODO: transaction ?
|
|
redo_sql = f"delete from scada_device_data where device_id = {f_device_id};"
|
|
for tv in cs.operations[0]['data']:
|
|
time, value = str(tv['time']), float(tv['value'])
|
|
f_time, f_value = f"'{time}'", value
|
|
redo_sql += f"\ninsert into scada_device_data (device_id, time, value) values ({f_device_id}, {f_time}, {f_value});"
|
|
new['data'].append({ 'time': time, 'value': value })
|
|
|
|
undo_sql = f"delete from scada_device_data where device_id = {f_device_id};"
|
|
for tv in old['data']:
|
|
time, value = str(tv['time']), float(tv['value'])
|
|
f_time, f_value = f"'{time}'", value
|
|
undo_sql += f"\ninsert into scada_device_data (device_id, time, value) values ({f_device_id}, {f_time}, {f_value});"
|
|
|
|
redo_cs = { 'type': 'scada_device_data' } | new
|
|
undo_cs = { 'type': 'scada_device_data' } | old
|
|
|
|
return DbChangeSet(redo_sql, undo_sql, [redo_cs], [undo_cs])
|
|
|
|
|
|
def set_scada_device_data(name: str, cs: ChangeSet) -> ChangeSet:
|
|
result = set_scada_device_data_cmd(name, cs)
|
|
result.redo_cs[0] |= g_update_prefix
|
|
result.undo_cs[0] |= g_update_prefix
|
|
return execute_command(name, result)
|
|
|
|
|
|
def add_scada_device_data_cmd(name: str, cs: ChangeSet) -> DbChangeSet:
|
|
values = cs.operations[0]
|
|
device_id = values['device_id']
|
|
time = values['time']
|
|
value = float(values['value'])
|
|
|
|
redo_sql = f"insert into scada_device_data (device_id, time, value) values ('{device_id}', '{time}', {value});"
|
|
undo_sql = f"delete from scada_device_data where device_id = '{device_id}' and time = '{time}';"
|
|
redo_cs = g_add_prefix | { 'type': 'scada_device_data', 'device_id': device_id, 'time': time, 'value': value }
|
|
undo_cs = g_delete_prefix | { 'type': 'scada_device_data', 'device_id': device_id, 'time': time }
|
|
|
|
return DbChangeSet(redo_sql, undo_sql, [redo_cs], [undo_cs])
|
|
|
|
|
|
def add_scada_device_data(name: str, cs: ChangeSet) -> ChangeSet:
|
|
row = try_read(name, f"select * from scada_device_data where device_id = '{cs.operations[0]['device_id']}' and time = '{cs.operations[0]['time']}'")
|
|
if row != None:
|
|
return ChangeSet()
|
|
return execute_command(name, add_scada_device_data_cmd(name, cs))
|
|
|
|
|
|
def delete_scada_device_data_cmd(name: str, cs: ChangeSet) -> DbChangeSet:
|
|
values = cs.operations[0]
|
|
device_id = values['device_id']
|
|
time = values['time']
|
|
value = float(read(name, f"select * from scada_device_data where device_id = '{device_id}' and time = '{time}'")['value'])
|
|
|
|
redo_sql = f"delete from scada_device_data where device_id = '{device_id}' and time = '{time}';"
|
|
undo_sql = f"insert into scada_device_data (device_id, time, value) values ('{device_id}', '{time}', {value});"
|
|
redo_cs = g_delete_prefix | { 'type': 'scada_device_data', 'device_id': device_id, 'time': time }
|
|
undo_cs = g_add_prefix | { 'type': 'scada_device_data', 'device_id': device_id, 'time': time, 'value': value }
|
|
|
|
return DbChangeSet(redo_sql, undo_sql, [redo_cs], [undo_cs])
|
|
|
|
|
|
def delete_scada_device_data(name: str, cs: ChangeSet) -> ChangeSet:
|
|
row = try_read(name, f"select * from scada_device_data where device_id = '{cs.operations[0]['device_id']}' and time = '{cs.operations[0]['time']}'")
|
|
if row == None:
|
|
return ChangeSet()
|
|
return execute_command(name, delete_scada_device_data_cmd(name, cs))
|
|
|
|
|
|
def clean_scada_device_data_cmd(name: str) -> ChangeSet:
|
|
cs = ChangeSet()
|
|
|
|
rows = read_all(name, 'select distinct device_id from scada_device_data')
|
|
for row in rows:
|
|
cs.update({ 'type': 'scada_device_data', 'device_id': row['device_id'], 'data': [] })
|
|
|
|
return cs
|